Home
last modified time | relevance | path

Searched refs:arg_min (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dmath_ops.py234 arg_min = deprecation.deprecated(None, "Use `tf.math.argmin` instead")(arg_min) # pylint: disable=… variable
236 tf_export(v1=["arg_min"])(dispatch.add_dispatch_support(arg_min))
313 gen_math_ops.arg_min.__doc__.replace("dimensions",
360 return gen_math_ops.arg_min(input, axis, name=name, output_type=output_type)
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir241 // CHECK-LABEL: arg_min
242 func @arg_min(%arg0: tensor<6xf64>) -> tensor<i32> {
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Doptimize.mlir1615 %1 = "tfl.arg_min"(%0, %cst) : (tensor<16x1024xf32>, tensor<1xi32>) -> tensor<16xi32>
1618 …// CHECK: %[[ARG_MIN:.*]] = "tfl.arg_min"(%arg0, %[[CST]]) : (tensor<16x1024xf32>, tensor<1xi32>) …
1637 %1 = "tfl.arg_min"(%0, %cst) : (tensor<16x1024xf32>, tensor<1xi32>) -> tensor<16xi32>
1640 …// CHECK: %[[ARG_MIN:.*]] = "tfl.arg_min"(%arg0, %[[CST]]) : (tensor<16x1024xf32>, tensor<1xi32>) …
Dops.mlir1570 …// CHECK: "tfl.arg_min"(%arg0, %arg1) {output_type = 2 : i32} : (tensor<3xi32>, tensor<i32>) -> te…
1571 …%0 = "tfl.arg_min"(%arg0, %arg1) {output_type = 2 : i32} : (tensor<3xi32>, tensor<i32>) -> tensor<…
Dlegalize-tf.mlir207 // CHECK: "tfl.arg_min"(%arg0, %arg1) : (tensor<3xi32>, tensor<i32>) -> tensor<i32>
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt776 name: "arg_min"
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md3220 ### tfl.arg_min
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td710 def TFL_ArgMinOp : TFL_Op<"arg_min", [NoSideEffect]> {