Home
last modified time | relevance | path

Searched refs:PoolingAlgorithm (Results 1 – 25 of 48) sorted by relevance

12

/external/armnn/src/backends/reference/workloads/
DPooling2d.cpp20 using PoolingAlgorithm = armnn::PoolingAlgorithm; typedef
22 float DefaultInitializer(PoolingAlgorithm algorithm) in DefaultInitializer()
26 case PoolingAlgorithm::Max: in DefaultInitializer()
30 case PoolingAlgorithm::Average: in DefaultInitializer()
31 case PoolingAlgorithm::L2: in DefaultInitializer()
44 Accumulator GetAccumulator(PoolingAlgorithm algorithm) in GetAccumulator()
48 case PoolingAlgorithm::Max: in GetAccumulator()
57 case PoolingAlgorithm::Average: in GetAccumulator()
64 case PoolingAlgorithm::L2: in GetAccumulator()
80 Executor GetExecutor(PoolingAlgorithm algorithm) in GetExecutor()
[all …]
DPooling3d.cpp20 using PoolingAlgorithm = armnn::PoolingAlgorithm; typedef
22 float DefaultInitializer(PoolingAlgorithm algorithm) in DefaultInitializer()
26 case PoolingAlgorithm::Max: in DefaultInitializer()
30 case PoolingAlgorithm::Average: in DefaultInitializer()
31 case PoolingAlgorithm::L2: in DefaultInitializer()
44 Accumulator GetAccumulator(PoolingAlgorithm algorithm) in GetAccumulator()
48 case PoolingAlgorithm::Max: in GetAccumulator()
57 case PoolingAlgorithm::Average: in GetAccumulator()
64 case PoolingAlgorithm::L2: in GetAccumulator()
80 Executor GetExecutor(PoolingAlgorithm algorithm) in GetExecutor()
[all …]
/external/armnn/src/armnnSerializer/
DSerializerUtils.cpp158 armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgo… in GetFlatBufferPoolingAlgorithm()
162 case armnn::PoolingAlgorithm::Average: in GetFlatBufferPoolingAlgorithm()
163 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average; in GetFlatBufferPoolingAlgorithm()
164 case armnn::PoolingAlgorithm::L2: in GetFlatBufferPoolingAlgorithm()
165 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2; in GetFlatBufferPoolingAlgorithm()
166 case armnn::PoolingAlgorithm::Max: in GetFlatBufferPoolingAlgorithm()
168 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max; in GetFlatBufferPoolingAlgorithm()
DSerializerUtils.hpp25 armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgo…
/external/armnn/delegate/classic/src/
DPooling.hpp66 armnn::PoolingAlgorithm poolingAlgorithm; in VisitPooling2dOperator()
70 poolingAlgorithm = armnn::PoolingAlgorithm::Average; in VisitPooling2dOperator()
73 poolingAlgorithm = armnn::PoolingAlgorithm::L2; in VisitPooling2dOperator()
76 poolingAlgorithm = armnn::PoolingAlgorithm::Max; in VisitPooling2dOperator()
177 armnn::PoolingAlgorithm poolingAlgorithm; in VisitPooling3dOperator()
180 poolingAlgorithm = armnn::PoolingAlgorithm::Max; in VisitPooling3dOperator()
184 poolingAlgorithm = armnn::PoolingAlgorithm::Average; in VisitPooling3dOperator()
/external/armnn/src/backends/backendsCommon/test/
DPooling2dEndToEndTestImpl.hpp24 PoolingAlgorithm poolAlg = PoolingAlgorithm::Max, in CreatePooling2dNetwork()
111 inputShape, outputShape, padMethod, PoolingAlgorithm::Average); in AvgPool2dEndToEnd()
152 inputShape, outputShape, padMethod, PoolingAlgorithm::Average); in AvgPool2dEndToEndFloat16()
/external/armnn/include/armnn/
DTypesUtils.hpp115 constexpr char const* GetPoolingAlgorithmAsCString(PoolingAlgorithm pooling) in GetPoolingAlgorithmAsCString()
119 case PoolingAlgorithm::Average: return "Average"; in GetPoolingAlgorithmAsCString()
120 case PoolingAlgorithm::Max: return "Max"; in GetPoolingAlgorithmAsCString()
121 case PoolingAlgorithm::L2: return "L2"; in GetPoolingAlgorithmAsCString()
DDescriptors.hpp362 : m_PoolType(PoolingAlgorithm::Max) in Pooling2dDescriptor()
393 PoolingAlgorithm m_PoolType;
422 : m_PoolType(PoolingAlgorithm::Max) in Pooling3dDescriptor()
461 PoolingAlgorithm m_PoolType;
/external/armnn/src/backends/backendsCommon/test/layerTests/
DPooling2dTestImpl.cpp140 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in SimpleMaxPooling2dSize3x3Stride2x4TestCommon()
266 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in SimpleMaxPooling2dTestCommon()
335 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in SimpleAveragePooling2dTestCommon()
403 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in LargeTensorsAveragePooling2dTestCommon()
453 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in SimpleL2Pooling2dTestCommon()
512 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize3Stride1TestCommon()
549 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize3Stride3TestCommon()
592 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize3Stride4TestCommon()
632 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize7TestCommon()
671 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize9TestCommon()
[all …]
DPooling3dTestImpl.cpp154 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
276 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in SimpleMaxPooling3dTestCommon()
349 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in IgnorePaddingSimpleMaxPooling3dTestCommon()
434 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in SimpleAveragePooling3dTestCommon()
507 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in LargeTensorsAveragePooling3dTestCommon()
560 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in IgnorePaddingSimpleAveragePooling3dTestCommon()
646 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in SimpleL2Pooling3dTestCommon()
719 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in IgnorePaddingSimpleL2Pooling3dTestCommon()
843 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPoolTestCommon()
890 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in AsymmetricNonSquareMaxPooling3dTestCommon()
[all …]
DPooling3dTestImpl.hpp273 armnn::PoolingAlgorithm poolingType,
282 armnn::PoolingAlgorithm poolingType,
291 armnn::PoolingAlgorithm poolingType,
DPooling2dTestImpl.hpp327 armnn::PoolingAlgorithm poolingType);
335 armnn::PoolingAlgorithm poolingType);
343 armnn::PoolingAlgorithm poolingType);
/external/armnn/src/backends/tosaCommon/operatorMappings/
DPooling2DOperator.cpp13 …std::string poolType = (poolDescriptor->m_PoolType == PoolingAlgorithm::Max) ? "Op_MAX" : "Op_AVG"; in ConvertPooling2DToTosaOperator()
14 … Op opcode = (poolDescriptor->m_PoolType == PoolingAlgorithm::Max) ? Op_MAX_POOL2D : Op_AVG_POOL2D; in ConvertPooling2DToTosaOperator()
/external/armnn/src/backends/tosaCommon/test/
DOneToManyMappingTests.cpp17 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average;
51 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average;
DOneToOneMappingTests.cpp361 descriptor.m_PoolType = PoolingAlgorithm::Average;
396 descriptor.m_PoolType = PoolingAlgorithm::Average;
435 descriptor.m_PoolType = PoolingAlgorithm::Max;
465 descriptor.m_PoolType = PoolingAlgorithm::Max;
/external/armnn/src/backends/tosaCommon/
DTosaMappings.cpp56 (poolDesc->m_PoolType == PoolingAlgorithm::Average) && in GetTosaMapping()
59 if (poolDesc->m_PoolType == PoolingAlgorithm::L2) in GetTosaMapping()
/external/armnn/src/armnn/optimizations/
DFoldPadIntoLayer2d.hpp71 return (descriptor.m_PoolType == PoolingAlgorithm::Max) in IsNeutralElement()
121 if (poolDescriptor.m_PoolType != PoolingAlgorithm::Max && in TryFoldPadIntoLayer2d()
132 poolDescriptor.m_PoolType == PoolingAlgorithm::Average && in TryFoldPadIntoLayer2d()
/external/armnn/src/backends/aclCommon/
DArmComputeUtils.hpp153 inline arm_compute::PoolingType ConvertPoolingAlgorithmToAclPoolingType(PoolingAlgorithm poolingAlg… in ConvertPoolingAlgorithmToAclPoolingType()
159 case PoolingAlgorithm::Max: return PoolingType::MAX; in ConvertPoolingAlgorithmToAclPoolingType()
160 case PoolingAlgorithm::Average: return PoolingType::AVG; in ConvertPoolingAlgorithmToAclPoolingType()
161 case PoolingAlgorithm::L2: return PoolingType::L2; in ConvertPoolingAlgorithmToAclPoolingType()
/external/armnn/src/armnn/test/optimizations/
DFoldPadTests.cpp209 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Average;
294 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Average;
365 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Average;
438 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Max;
500 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Average;
576 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Average;
DFoldPadIntoQuantizedAveragePooling2DTests.cpp43 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Average; in FoldPadIntoQuantizedAvgPoolTest()
/external/armnn/src/backends/tosaReference/test/
DTosaRefLayerSupportTests.cpp249 desc.m_PoolType = PoolingAlgorithm::Max;
276 desc.m_PoolType = PoolingAlgorithm::Average;
319 desc.m_PoolType = PoolingAlgorithm::Average;
/external/armnn/src/backends/neon/test/
DNeonLayerTests.cpp1700 …O_TEST_CASE_WITH_THF(CompareMaxPooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::Max)
1702 PoolingAlgorithm::Max)
1704 PoolingAlgorithm::Average)
1706 PoolingAlgorithm::Average)
1707 …UTO_TEST_CASE_WITH_THF(CompareL2Pooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::L2)
1709 PoolingAlgorithm::L2)
1711 …EF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling3dWithRef, ComparePooling3dTest, PoolingAlgorithm::Max,
1714 PoolingAlgorithm::Average, DataLayout::NDHWC)
1715 …_REF_AUTO_TEST_CASE_WITH_THF(CompareL2Pooling3dWithRef, ComparePooling3dTest, PoolingAlgorithm::L2,
/external/android-nn-driver/1.2/
DHalPolicy.cpp206 …return ConvertPooling2d<hal_1_2::HalPolicy>(operation, __func__, PoolingAlgorithm::Average, model,… in ConvertAveragePool2d()
329 …return ConvertPooling2d<hal_1_2::HalPolicy>(operation, __func__, PoolingAlgorithm::L2, model, data… in ConvertL2Pool2d()
355 …return ConvertPooling2d<hal_1_2::HalPolicy>(operation, __func__, PoolingAlgorithm::Max, model, dat… in ConvertMaxPool2d()
/external/android-nn-driver/1.0/
DHalPolicy.cpp79 …return ConvertPooling2d<hal_1_0::HalPolicy>(operation, __func__, armnn::PoolingAlgorithm::Average,… in ConvertAveragePool2d()
513 …return ConvertPooling2d<hal_1_0::HalPolicy>(operation, __func__, armnn::PoolingAlgorithm::L2, mode… in ConvertL2Pool2d()
519 …return ConvertPooling2d<hal_1_0::HalPolicy>(operation, __func__, armnn::PoolingAlgorithm::Max, mod… in ConvertMaxPool2d()
/external/android-nn-driver/1.3/
DHalPolicy.cpp194 …return ConvertPooling2d<hal_1_3::HalPolicy>(operation, __func__, PoolingAlgorithm::Average, model,… in ConvertAveragePool2d()
338 …return ConvertPooling2d<hal_1_3::HalPolicy>(operation, __func__, PoolingAlgorithm::L2, model, data… in ConvertL2Pool2d()
379 …return ConvertPooling2d<hal_1_3::HalPolicy>(operation, __func__, PoolingAlgorithm::Max, model, dat… in ConvertMaxPool2d()

12