- Sort Score
- Result 10 results
- Languages All
Results 21 - 21 of 21 for 8x8x8xf32 (0.21 sec)
-
tensorflow/compiler/mlir/lite/tests/ops.mlir
func.return %0, %1: tensor<*xf32>, tensor<*xi32> } // ----- // CHECK-LABEL: topk_2 func.func @topk_2(%arg0: tensor<3x4x8xf32>) -> (tensor<3x4x2xf32>, tensor<3x4x2xi32>) { %0 = arith.constant dense<2> : tensor<i32> %1:2 = "tfl.topk_v2"(%arg0, %0) : (tensor<3x4x8xf32>, tensor<i32>) -> (tensor<3x4x2xf32>, tensor<3x4x2xi32>) func.return %1#0, %1#1: tensor<3x4x2xf32>, tensor<3x4x2xi32> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0)