Home
last modified time | relevance | path

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

12

/external/armnn/src/backends/reference/workloads/
DArgMinMax.cpp16 void ArgMinMax(Decoder<float>& in, OUT* out, const TensorInfo& inputTensorInfo, in ArgMinMax() function
49 template void ArgMinMax(Decoder<float>& in, int32_t* out, const TensorInfo& inputTensorInfo,
52 template void ArgMinMax(Decoder<float>& in, int64_t* out, const TensorInfo& inputTensorInfo,
DRefArgMinMaxWorkload.cpp46ArgMinMax(decoder, output, inputTensorInfo, outputTensorInfo, m_Data.m_Parameters.m_Function, in Execute()
50ArgMinMax(decoder, output, inputTensorInfo, outputTensorInfo, m_Data.m_Parameters.m_Function, in Execute()
DArgMinMax.hpp17 void ArgMinMax(Decoder<float>& in, OUT *out, const TensorInfo& inputTensorInfo,
DCMakeLists.txt10 ArgMinMax.cpp
11 ArgMinMax.hpp
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Darg_min_max.h36 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
81 ArgMinMax(input1_shape, input1_data, input2_data, output_shape, output_data, in ArgMinMax()
/external/armnn/src/backends/reference/test/
DArgMinMaxTests.cpp21 ArgMinMax(*armnn::MakeDecoder<float>(inputInfo, inputValues.data()),
41 ArgMinMax(*armnn::MakeDecoder<float>(inputInfo, inputValues.data()),
/external/armnn/src/armnn/layers/
DArgMinMaxLayer.cpp20 : LayerWithParameters(1, 1, LayerType::ArgMinMax, param, name) in ArgMinMaxLayer()
29 return factory.CreateWorkload(LayerType::ArgMinMax, descriptor, PrepInfoAndDesc(descriptor)); in CreateWorkload()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc138 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min) { in ArgMinMax() function
169 return ArgMinMax(input, output_type, axis, /*is_min=*/false); in ArgMax()
173 return ArgMinMax(input, output_type, axis, /*is_min=*/true); in ArgMin()
Darithmetic.h88 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min);
Darithmetic_test.cc54 ArgMinMax(x, primitive_util::NativeToPrimitiveType<NativeT>(), axis, in TestArgMinMax()
/external/armnn/src/armnnDeserializer/test/
DDeserializeArgMinMax.cpp53 layerType: "ArgMinMax", in ArgMinMaxFixture()
/external/armnn/docs/
D05_02_deserializer_serializer.dox24 - ArgMinMax
117 - ArgMinMax
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops.cc67 xla::ArgMinMax(input, index_xla_type, axis, /*is_min=*/is_min_); in Compile()
/external/armnn/src/backends/neon/
DNeonTensorHandleFactory.hpp17 LayerType::ArgMinMax,
/external/armnn/src/armnn/test/
DTestNameAndDescriptorLayerVisitor.hpp61 DECLARE_TEST_NAME_AND_DESCRIPTOR_LAYER_VISITOR_CLASS(ArgMinMax)
DTestNameAndDescriptorLayerVisitor.cpp281 TEST_CASE_CHECK_LAYER_VISITOR_NAME_AND_DESCRIPTOR(ArgMinMax, CheckArgMinMaxLayerVisitorNameAndDescr…
314 TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR_AND_DESCRIPTOR(ArgMinMax,
/external/armnn/src/armnn/
DLayersFwd.hpp112 DECLARE_LAYER(ArgMinMax)
/external/armnn/delegate/classic/
DCMakeLists.txt13 src/ArgMinMax.hpp
/external/armnn/src/backends/reference/
Dbackend.mk25 workloads/ArgMinMax.cpp \
/external/armnn/include/armnn/
DTypes.hpp403 X(ArgMinMax) \
/external/tensorflow/tensorflow/lite/kernels/
Darg_min_max.cc140 optimized_ops::ArgMinMax( \ in Eval()
/external/ComputeLibrary/tests/validation/NEON/
DArgMinMax.cpp47 TEST_SUITE(ArgMinMax) in TEST_SUITE() argument
/external/ComputeLibrary/tests/validation/CL/
DArgMinMax.cpp60 TEST_SUITE(ArgMinMax) in TEST_SUITE() argument
/external/ComputeLibrary/tests/validation/
DCMakeLists.txt108 NEON/ArgMinMax.cpp
/external/armnn/src/backends/backendsCommon/test/layerTests/
DArgMinMaxTestImpl.cpp44 …unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateWorkload(armnn::LayerType::ArgMinMax, in ArgMinMaxTestCommon()

12