Searched refs:ArgMinMaxTwoPass (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 175 XlaOp ArgMinMaxTwoPass(XlaOp input, PrimitiveType output_type, int axis, in ArgMinMaxTwoPass() function
232 return ArgMinMaxTwoPass(input, output_type, axis, /*is_min=*/false, tie_low); in ArgMaxTwoPass()
237 return ArgMinMaxTwoPass(input, output_type, axis, /*is_min=*/true, tie_low); in ArgMinTwoPass()
|