Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/micro/kernels/
Darg_min_max.cc40 reference_ops::ArgMinMax(input1_shape, input1_data, input2_data, in ArgMinMaxHelper()
43 reference_ops::ArgMinMax(input1_shape, input1_data, input2_data, in ArgMinMaxHelper()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc143 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min, in ArgMinMax() function
222 return ArgMinMax(input, output_type, axis, /*is_min=*/false, stable, tie_low); in ArgMax()
227 return ArgMinMax(input, output_type, axis, /*is_min=*/true, stable, tie_low); in ArgMin()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Darg_min_max.h25 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() function
Dlegacy_reference_ops.h2138 ArgMinMax(DimsToShape(input_dims), input_data, axis, output_shape, in ArgMax()
2143 void ArgMinMax(const T3* axis, const T1* input_data, const Dims<4>& input_dims, in ArgMinMax() function
2145 ArgMinMax(axis, DimsToShape(input_dims), input_data, DimsToShape(output_dims), in ArgMinMax()
Dreference_ops.h1469 ArgMinMax(input1_shape, input1_data, input2_data, output_shape, output_data, in ArgMax()
/external/tensorflow/tensorflow/lite/kernels/
Darg_min_max.cc144 optimized_ops::ArgMinMax( \ in Eval()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h36 using reference_ops::ArgMinMax;
Doptimized_ops.h68 using reference_ops::ArgMinMax;