- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for 2x2x3x2xf32 (0.16 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK: return } func.func @topk_4(%arg0: tensor<1x2x3x4xf32>) -> (tensor<1x2x3x2xf32>, tensor<1x2x3x2xi32>) { %0 = "tf.Const"() { value = dense<2> : tensor<i32> } : () -> tensor<i32> %1:2 = "tf.TopKV2"(%arg0, %0) : (tensor<1x2x3x4xf32>, tensor<i32>) -> (tensor<1x2x3x2xf32>, tensor<1x2x3x2xi32>) func.return %1#0, %1#1: tensor<1x2x3x2xf32>, tensor<1x2x3x2xi32> // CHECK-LABEL: topk_4 // CHECK: "tfl.topk_v2"(%arg0, %cst)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
} // ----- // CHECK-LABEL: testReverseV2 func.func @testReverseV2(%arg0: tensor<1x2x3x4xf32>, %arg1 : tensor<2xi32>) -> tensor<1x2x3x4xf32> { // CHECK: "tfl.reverse_v2"(%arg0, %arg1) %0 = "tfl.reverse_v2"(%arg0, %arg1): (tensor<1x2x3x4xf32>, tensor<2xi32>) -> tensor<1x2x3x4xf32> func.return %0 : tensor<1x2x3x4xf32> } // ----- // test select // CHECK-LABEL: testSelect
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0)