Searched refs:ArgMinTwoPass (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.h | 96 XlaOp ArgMinTwoPass(XlaOp input, PrimitiveType output_type, int axis,
|
| D | arithmetic_test.cc | 62 return ArgMinTwoPass(op, type, axis, tie_low); in TestArgMinMax()
|
| D | arithmetic.cc | 235 XlaOp ArgMinTwoPass(XlaOp input, PrimitiveType output_type, int axis, in ArgMinTwoPass() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | index_ops.cc | 72 output = xla::ArgMinTwoPass(input, index_xla_type, axis); in Compile()
|