Searched defs:ArgMinMax (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A 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
|
| H A D | legacy_reference_ops.h | 2150 void ArgMinMax(const T3* axis, const T1* input_data, const Dims<4>& input_dims, in ArgMinMax() function
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ |
| H A D | ArgMinMax.cpp | 47 TEST_SUITE(ArgMinMax) in TEST_SUITE() argument
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | ArgMinMax.cpp | 16 void ArgMinMax(Decoder<float>& in, OUT* out, const TensorInfo& inputTensorInfo, in ArgMinMax() function
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/CL/ |
| H A D | ArgMinMax.cpp | 60 TEST_SUITE(ArgMinMax) in TEST_SUITE() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | arithmetic.cc | 138 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min) { in ArgMinMax() function
|
| /aosp_15_r20/external/ComputeLibrary/ |
| H A D | filelist.json | 159 "ArgMinMax": { object
890 "ArgMinMax": { object
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 4991 void ArgMinMax(const T3* axis, const T1* input_data, const Dims<4>& input_dims, in ArgMinMax() function
|
| H A D | optimized_ops.h | 7982 inline void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() function
|