- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GatherV2 (0.14 sec)
-
tensorflow/compiler/mlir/lite/tests/fuse-tftext.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 460.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
func.return %0 : tensor<2x4x128xf32> } //===----------------------------------------------------------------------===// // tf.GatherV2 legalization //===----------------------------------------------------------------------===// // ----- // CHECK-LABEL: @gather_v2 // CHECK-SAME: %[[PARAMS:[a-zA-Z0-9_]+]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
func.return %0 : tensor<8xui8> } // CHECK-LABEL: func @torch_index_select( // CHECK: %[[AXIS:.+]] = "tf.Const"() <{value = dense<0> : tensor<i64>}> : () -> tensor<i64> // CHECK: %[[RES:.+]] = "tf.GatherV2"(%arg0, %arg1, %[[AXIS]]) <{batch_dims = 0 : i64}> // CHECK: return %[[RES]] func.func @torch_index_select(%arg0: tensor<2x1xf32>, %arg1: tensor<2xi32>) -> tensor<2x1xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0)