Home
last modified time | relevance | path

Searched refs:PoolingAlgorithm (Results 1 – 25 of 33) 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 …]
/external/armnn/src/armnnSerializer/
DSerializerUtils.cpp122 armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgo… in GetFlatBufferPoolingAlgorithm()
126 case armnn::PoolingAlgorithm::Average: in GetFlatBufferPoolingAlgorithm()
127 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average; in GetFlatBufferPoolingAlgorithm()
128 case armnn::PoolingAlgorithm::L2: in GetFlatBufferPoolingAlgorithm()
129 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2; in GetFlatBufferPoolingAlgorithm()
130 case armnn::PoolingAlgorithm::Max: in GetFlatBufferPoolingAlgorithm()
132 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max; in GetFlatBufferPoolingAlgorithm()
DSerializerUtils.hpp23 armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgo…
DArmnnSchema.fbs398 enum PoolingAlgorithm : byte {
415 poolType:PoolingAlgorithm;
/external/armnn/delegate/src/
DPooling.hpp51 armnn::PoolingAlgorithm poolingAlgorithm; in VisitPoolingOperator()
55 poolingAlgorithm = armnn::PoolingAlgorithm::Average; in VisitPoolingOperator()
58 poolingAlgorithm = armnn::PoolingAlgorithm::L2; in VisitPoolingOperator()
61 poolingAlgorithm = armnn::PoolingAlgorithm::Max; in VisitPoolingOperator()
/external/armnn/src/backends/aclCommon/
DArmComputeUtils.hpp119 inline arm_compute::PoolingType ConvertPoolingAlgorithmToAclPoolingType(PoolingAlgorithm poolingAlg… in ConvertPoolingAlgorithmToAclPoolingType()
125 case PoolingAlgorithm::Max: return PoolingType::MAX; in ConvertPoolingAlgorithmToAclPoolingType()
126 case PoolingAlgorithm::Average: return PoolingType::AVG; in ConvertPoolingAlgorithmToAclPoolingType()
127 case PoolingAlgorithm::L2: return PoolingType::L2; in ConvertPoolingAlgorithmToAclPoolingType()
/external/armnn/include/armnn/
DTypesUtils.hpp95 constexpr char const* GetPoolingAlgorithmAsCString(PoolingAlgorithm pooling) in GetPoolingAlgorithmAsCString()
99 case PoolingAlgorithm::Average: return "Average"; in GetPoolingAlgorithmAsCString()
100 case PoolingAlgorithm::Max: return "Max"; in GetPoolingAlgorithmAsCString()
101 case PoolingAlgorithm::L2: return "L2"; in GetPoolingAlgorithmAsCString()
DTypes.hpp104 enum class PoolingAlgorithm enum
DDescriptors.hpp327 : m_PoolType(PoolingAlgorithm::Max) in Pooling2dDescriptor()
358 PoolingAlgorithm m_PoolType;
/external/armnn/src/backends/backendsCommon/test/layerTests/
DPooling2dTestImpl.cpp135 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in SimpleMaxPooling2dSize3x3Stride2x4TestCommon()
261 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in SimpleMaxPooling2dTestCommon()
333 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in SimpleAveragePooling2dTestCommon()
404 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in LargeTensorsAveragePooling2dTestCommon()
457 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in SimpleL2Pooling2dTestCommon()
519 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize3Stride1TestCommon()
555 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize3Stride3TestCommon()
597 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize3Stride4TestCommon()
636 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize7TestCommon()
674 descriptor.m_PoolType = armnn::PoolingAlgorithm::L2; in L2Pooling2dSize9TestCommon()
[all …]
DPooling2dTestImpl.hpp327 armnn::PoolingAlgorithm poolingType);
335 armnn::PoolingAlgorithm poolingType);
343 armnn::PoolingAlgorithm poolingType);
DAdditionTestImpl.cpp567 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in AdditionAfterMaxPoolTest()
/external/android-nn-driver/1.2/
DHalPolicy.cpp166 …return ConvertPooling2d<hal_1_2::HalPolicy>(operation, __func__, PoolingAlgorithm::Average, model,… in ConvertAveragePool2d()
274 …return ConvertPooling2d<hal_1_2::HalPolicy>(operation, __func__, PoolingAlgorithm::L2, model, data… in ConvertL2Pool2d()
300 …return ConvertPooling2d<hal_1_2::HalPolicy>(operation, __func__, PoolingAlgorithm::Max, model, dat… in ConvertMaxPool2d()
/external/android-nn-driver/1.3/
DHalPolicy.cpp176 …return ConvertPooling2d<hal_1_3::HalPolicy>(operation, __func__, PoolingAlgorithm::Average, model,… in ConvertAveragePool2d()
306 …return ConvertPooling2d<hal_1_3::HalPolicy>(operation, __func__, PoolingAlgorithm::L2, model, data… in ConvertL2Pool2d()
338 …return ConvertPooling2d<hal_1_3::HalPolicy>(operation, __func__, PoolingAlgorithm::Max, model, dat… in ConvertMaxPool2d()
/external/android-nn-driver/1.0/
DHalPolicy.cpp85 …return ConvertPooling2d<hal_1_0::HalPolicy>(operation, __func__, armnn::PoolingAlgorithm::Average,… in ConvertAveragePool2d()
507 …return ConvertPooling2d<hal_1_0::HalPolicy>(operation, __func__, armnn::PoolingAlgorithm::L2, mode… in ConvertL2Pool2d()
513 …return ConvertPooling2d<hal_1_0::HalPolicy>(operation, __func__, armnn::PoolingAlgorithm::Max, mod… in ConvertMaxPool2d()
/external/armnn/src/backends/neon/test/
DNeonLayerTests.cpp1373 …O_TEST_CASE_WITH_THF(CompareMaxPooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::Max)
1375 PoolingAlgorithm::Max)
1377 PoolingAlgorithm::Average)
1379 PoolingAlgorithm::Average)
1380 …UTO_TEST_CASE_WITH_THF(CompareL2Pooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::L2)
1382 PoolingAlgorithm::L2)
DNeonTensorHandleTests.cpp176 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in BOOST_AUTO_TEST_CASE()
493 descriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in BOOST_AUTO_TEST_CASE()
/external/armnn/src/armnn/test/
DTestNameAndDescriptorLayerVisitor.cpp171 descriptor.m_PoolType = armnn::PoolingAlgorithm::Max; in GetDescriptor()
/external/armnn/src/backends/cl/test/
DClLayerTests.cpp1280 …EF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling2dWithRef, ComparePooling2dTest, PoolingAlgorithm::Max)
1283 ComparePooling2dTest, PoolingAlgorithm::Average)
1285 PoolingAlgorithm::Average)
1287 …_REF_AUTO_TEST_CASE_WITH_THF(CompareL2Pooling2dWithRef, ComparePooling2dTest, PoolingAlgorithm::L2)
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.hpp119 void ParsePool(size_t subgraphIndex, size_t operatorIndex, armnn::PoolingAlgorithm algorithm);
DTfLiteParser.cpp1165 ParsePool(subgraphIndex, operatorIndex, PoolingAlgorithm::Average); in ParseAveragePool2D()
1248 ParsePool(subgraphIndex, operatorIndex, PoolingAlgorithm::Max); in ParseMaxPool2D()
1313 PoolingAlgorithm algorithm) in ParsePool()
1326 case PoolingAlgorithm::Average: in ParsePool()
1330 case PoolingAlgorithm::Max: in ParsePool()
/external/armnn/src/backends/backendsCommon/test/
DOptimizeSubgraphViewTests.cpp214 poolingDescriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in BuildFullyUnsupportedSubgraph1()
249 poolingDescriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in BuildFullyUnsupportedSubgraph2()
377 poolingDescriptor.m_PoolType = armnn::PoolingAlgorithm::Average; in BuildPartiallySupportedSubgraph()
/external/armnn/src/armnnTfParser/
DTfParser.hpp165 armnn::PoolingAlgorithm pooltype);
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_descriptors.i660 PoolingAlgorithm m_PoolType;
/external/armnn/src/armnnOnnxParser/
DOnnxParser.cpp1192 desc.m_PoolType = PoolingAlgorithm::Average; in ParseAveragePool()
1515 desc.m_PoolType = PoolingAlgorithm::Average; in ParseGlobalAveragePool()
1539 desc.m_PoolType = PoolingAlgorithm::Max; in ParseMaxPool()

12