Searched refs:ArgMinMax (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 128 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min) { in ArgMinMax() function
179 return ArgMinMax(input, output_type, axis, /*is_min=*/false); in ArgMax()
183 return ArgMinMax(input, output_type, axis, /*is_min=*/true); in ArgMin()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | arg_min_max.cc | 124 optimized_ops::ArgMinMax( \ in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 2046 ArgMinMax(DimsToShape(input_dims), input_data, axis, output_shape, in ArgMax()
2051 void ArgMinMax(const T3* axis, const T1* input_data, const Dims<4>& input_dims, in ArgMinMax() function
2053 ArgMinMax(axis, DimsToShape(input_dims), input_data, DimsToShape(output_dims), in ArgMinMax()
|
| D | reference_ops.h | 3865 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() function
3914 ArgMinMax(input1_shape, input1_data, input2_data, output_shape, output_data, in ArgMax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 33 using reference_ops::ArgMinMax;
|
| D | optimized_ops.h | 50 using reference_ops::ArgMinMax;
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 45 using reference_ops::ArgMinMax;
|