- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for 2x5x7x3xf32 (0.25 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
} func.func @gatherWithBatchDims(%arg0 : tensor<2x3x6xf32>, %arg1 : tensor<2x5xi32>) -> tensor<2x5x3x6xf32> { %0 = "tf.Const"() { value = dense<[1]> : tensor<1xi32> } : () -> tensor<1xi32> %1 = "tf.GatherV2"(%arg0, %arg1, %0) {batch_dims = 1 : i64} : (tensor<2x3x6xf32>, tensor<2x5xi32>, tensor<1xi32>) -> tensor<2x5x3x6xf32> func.return %1 : tensor<2x5x3x6xf32> // CHECK-LABEL:gatherWithBatchDims
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0)