Searched defs:ArgMinMax (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | arg_min_max.h | 36 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() function
78 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() function
|
| D | legacy_reference_ops.h | 2148 void ArgMinMax(const T3* axis, const T1* input_data, const Dims<4>& input_dims, in ArgMinMax() function
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | ArgMinMax.cpp | 47 TEST_SUITE(ArgMinMax) in TEST_SUITE() argument
|
| /external/armnn/src/backends/reference/workloads/ |
| D | ArgMinMax.cpp | 16 void ArgMinMax(Decoder<float>& in, OUT* out, const TensorInfo& inputTensorInfo, in ArgMinMax() function
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | ArgMinMax.cpp | 60 TEST_SUITE(ArgMinMax) in TEST_SUITE() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 143 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min, in ArgMinMax() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4989 void ArgMinMax(const T3* axis, const T1* input_data, const Dims<4>& input_dims, in ArgMinMax() function
|
| D | optimized_ops.h | 8032 inline void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() function
|