- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ApproxTopK (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/mark_ops_for_outside_compilation.mlir
func.func @tf2xla_fallback_op_approx_top_k(%arg0: tensor<16xf32>) -> (tensor<?xf32>, tensor<?xi32>) { %0:2 = "tf_device.cluster"() ({ // CHECK: tf.ApproxTopK // CHECK-NOT: _xla_outside_compilation %1:2 = "tf.ApproxTopK"(%arg0) {k = 2} : (tensor<16xf32>) -> (tensor<?xf32>, tensor<?xi32>) tf_device.return %1#0, %1#1 : tensor<?xf32>, tensor<?xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 16:22:32 UTC 2024 - 29.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir
%0 = "tf.ReadVariableOp"(%arg0) {device = ""} : (tensor<!tf_type.resource<tensor<10x500xbf16>>>) -> tensor<10x500xbf16> // CHECK: mhlo.compare GT
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 38.6K bytes - Viewed (0)