Home
last modified time | relevance | path

Searched refs:ArgMinMaxFunction (Results 1 – 25 of 27) sorted by relevance

12

/external/armnn/src/backends/backendsCommon/test/
DArgMinMaxEndToEndTestImpl.hpp19 armnn::ArgMinMaxFunction function, in CreateArgMinMaxNetwork()
43 armnn::ArgMinMaxFunction function, in ArgMinMaxEndToEndImpl()
80 armnn::ArgMinMaxFunction::Max, in ArgMaxEndToEndSimple()
98 armnn::ArgMinMaxFunction::Min, in ArgMinEndToEndSimple()
123 armnn::ArgMinMaxFunction::Max, in ArgMaxAxis0EndToEnd()
148 armnn::ArgMinMaxFunction::Min, in ArgMinAxis0EndToEnd()
173 armnn::ArgMinMaxFunction::Max, in ArgMaxAxis1EndToEnd()
198 armnn::ArgMinMaxFunction::Min, in ArgMinAxis1EndToEnd()
224 armnn::ArgMinMaxFunction::Max, in ArgMaxAxis2EndToEnd()
250 armnn::ArgMinMaxFunction::Min, in ArgMinAxis2EndToEnd()
[all …]
/external/armnn/src/backends/reference/workloads/
DArgMinMax.cpp17 const TensorInfo& outputTensorInfo, ArgMinMaxFunction function, int axis) in ArgMinMax()
37 if ((function == armnn::ArgMinMaxFunction::Min && value < tmpValue) || in ArgMinMax()
38 (function == armnn::ArgMinMaxFunction::Max && value > tmpValue)) { in ArgMinMax()
50 const TensorInfo& outputTensorInfo, ArgMinMaxFunction function, int axis);
53 const TensorInfo& outputTensorInfo, ArgMinMaxFunction function, int axis);
DArgMinMax.hpp18 const TensorInfo& outputTensorInfo, ArgMinMaxFunction function, int axis);
/external/armnn/src/backends/reference/test/
DArgMinMaxTests.cpp25 armnn::ArgMinMaxFunction::Min,
45 armnn::ArgMinMaxFunction::Max,
/external/armnn/src/backends/backendsCommon/test/layerTests/
DArgMinMaxTestImpl.cpp23 armnn::ArgMinMaxFunction argMinMaxFunction, in ArgMinMaxTestCommon()
88 armnn::ArgMinMaxFunction::Max, in ArgMaxSimpleTest()
116 armnn::ArgMinMaxFunction::Min, in ArgMinSimpleTest()
152 armnn::ArgMinMaxFunction::Min, in ArgMinChannelTest()
188 armnn::ArgMinMaxFunction::Max, in ArgMaxChannelTest()
224 armnn::ArgMinMaxFunction::Max, in ArgMaxHeightTest()
260 armnn::ArgMinMaxFunction::Min, in ArgMinWidthTest()
/external/armnn/src/backends/cl/workloads/
DClArgMinMaxWorkload.cpp42 if (descriptor.m_Function == ArgMinMaxFunction::Max) in ClArgMinMaxWorkloadValidate()
75 if (m_Data.m_Parameters.m_Function == ArgMinMaxFunction::Max) in ClArgMinMaxWorkload()
/external/armnn/src/backends/neon/workloads/
DNeonArgMinMaxWorkload.cpp42 if (descriptor.m_Function == ArgMinMaxFunction::Max) in NeonArgMinMaxWorkloadValidate()
74 if (m_Data.m_Parameters.m_Function == ArgMinMaxFunction::Max) in NeonArgMinMaxWorkload()
/external/armnn/delegate/classic/src/
DArgMinMax.hpp68 desc.m_Function = armnn::ArgMinMaxFunction::Max; in VisitArgMinMaxOperator()
81 desc.m_Function = armnn::ArgMinMaxFunction::Min; in VisitArgMinMaxOperator()
/external/armnn/include/armnn/
DTypesUtils.hpp51 constexpr char const* GetArgMinMaxFunctionAsCString(ArgMinMaxFunction function) in GetArgMinMaxFunctionAsCString()
55 case ArgMinMaxFunction::Max: return "Max"; in GetArgMinMaxFunctionAsCString()
56 case ArgMinMaxFunction::Min: return "Min"; in GetArgMinMaxFunctionAsCString()
DTypes.hpp102 enum class ArgMinMaxFunction enum
DDescriptors.hpp70 : m_Function(ArgMinMaxFunction::Min) in ArgMinMaxDescriptor()
81 ArgMinMaxFunction m_Function;
/external/android-nn-driver/1.2/
DHalPolicy.cpp57 return ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Max); in ConvertOperation()
59 return ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Min); in ConvertOperation()
197 armnn::ArgMinMaxFunction argMinMaxFunction) in ConvertArgMinMax()
DHalPolicy.hpp45 armnn::ArgMinMaxFunction argMinMaxFunction);
/external/android-nn-driver/1.3/
DHalPolicy.cpp29 return ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Max); in ConvertOperation()
31 return ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Min); in ConvertOperation()
185 armnn::ArgMinMaxFunction argMinMaxFunction) in ConvertArgMinMax()
DHalPolicy.hpp42 armnn::ArgMinMaxFunction argMinMaxFunction);
/external/armnn/shim/sl/canonical/
DConverter.hpp39 armnn::ArgMinMaxFunction argMinMaxFunction);
DConverter.cpp30 return ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Max); in ConvertOperation()
32 return ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Min); in ConvertOperation()
256 armnn::ArgMinMaxFunction argMinMaxFunction) in ConvertArgMinMax()
/external/armnn/src/armnnSerializer/
DSerializer.cpp86 serializer::ArgMinMaxFunction GetFlatBufferArgMinMaxFunction(armnn::ArgMinMaxFunction function) in GetFlatBufferArgMinMaxFunction()
90 case armnn::ArgMinMaxFunction::Max: in GetFlatBufferArgMinMaxFunction()
91 return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Max; in GetFlatBufferArgMinMaxFunction()
92 case armnn::ArgMinMaxFunction::Min: in GetFlatBufferArgMinMaxFunction()
94 return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Min; in GetFlatBufferArgMinMaxFunction()
DArmnnSchema.fbs27 enum ArgMinMaxFunction : byte {
231 argMinMaxFunction:ArgMinMaxFunction;
/external/armnn/src/armnn/test/
DTestNameAndDescriptorLayerVisitor.cpp51 descriptor.m_Function = armnn::ArgMinMaxFunction::Max; in GetDescriptor()
DShapeInferenceTests.cpp190 descriptor.m_Function = ArgMinMaxFunction::Min;
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_descriptors.i86 ArgMinMaxFunction m_Function;
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.hpp114 …void ParseArgMinMax(size_t subgraphIndex, size_t operatorIndex, armnn::ArgMinMaxFunction argMinMax…
DTfLiteParser.cpp4188 ParseArgMinMax(subgraphIndex, operatorIndex, armnn::ArgMinMaxFunction::Min); in ParseArgMin()
4193 ParseArgMinMax(subgraphIndex, operatorIndex, armnn::ArgMinMaxFunction::Max); in ParseArgMax()
4196 void TfLiteParserImpl::ParseArgMinMax(size_t subgraphIndex, size_t operatorIndex, ArgMinMaxFunction in ParseArgMinMax()
4250 auto layerName = argMinMaxFunction == ArgMinMaxFunction::Max ? "ArgMax:{}:{}" : "ArgMin:{}:{}"; in ParseArgMinMax()
/external/armnn/src/armnnDeserializer/
DDeserializer.cpp504 armnn::ArgMinMaxFunction ToArgMinMaxFunction(armnnSerializer::ArgMinMaxFunction function) in ToArgMinMaxFunction()
508 case armnnSerializer::ArgMinMaxFunction::ArgMinMaxFunction_Max: in ToArgMinMaxFunction()
509 return armnn::ArgMinMaxFunction::Max; in ToArgMinMaxFunction()
510 case armnnSerializer::ArgMinMaxFunction::ArgMinMaxFunction_Min: in ToArgMinMaxFunction()
512 return armnn::ArgMinMaxFunction::Min; in ToArgMinMaxFunction()

12