Searched refs:ArgMaxTwoPass (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.h | 86 XlaOp ArgMaxTwoPass(XlaOp input, PrimitiveType output_type, int axis,
|
| D | arithmetic_test.cc | 71 return ArgMaxTwoPass(op, type, axis, tie_low); in TestArgMinMax()
|
| D | arithmetic.cc | 230 XlaOp ArgMaxTwoPass(XlaOp input, PrimitiveType output_type, int axis, in ArgMaxTwoPass() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | index_ops.cc | 78 output = xla::ArgMaxTwoPass(input, index_xla_type, axis); in Compile()
|
| D | categorical_op.cc | 115 argmax = xla::ArgMaxTwoPass(softmax_entries, xla_output_type, in Compile()
|