Searched defs:ArgMinMax (Results 1 – 3 of 3) sorted by relevance
25 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() function
2143 void ArgMinMax(const T3* axis, const T1* input_data, const Dims<4>& input_dims, in ArgMinMax() function
143 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min, in ArgMinMax() function