- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for TOPK_V2 (0.4 sec)
-
tensorflow/compiler/mlir/lite/tests/optimize.mlir
// CHECK: %[[VALUES:.*]], %[[INDICES:.*]] = "tfl.topk_v2"(%[[ARG]], %[[K]]) : (tensor<3x10xf32>, tensor<i32>) -> (tensor<3x5xf32>, tensor<3x5xi32>) // CHECK: return %[[VALUES]] : tensor<3x5xf32> // CHECK: } func.func @optimizeTopKOnlyValues(%arg: tensor<3x10xf32>) -> tensor<3x5xf32>{ %K = arith.constant dense<10> : tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 284.1K bytes - Viewed (0)