Searched refs:pooling2dDescriptor (Results 1 – 10 of 10) sorted by relevance
| /external/armnn/src/armnn/test/optimizations/ |
| D | FoldPadTests.cpp | 208 Pooling2dDescriptor pooling2dDescriptor; variable
209 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Average;
210 pooling2dDescriptor.m_PoolWidth = 3;
211 pooling2dDescriptor.m_PoolHeight = 3;
212 pooling2dDescriptor.m_StrideX = 1;
213 pooling2dDescriptor.m_StrideY = 1;
214 pooling2dDescriptor.m_DataLayout = DataLayout::NHWC;
216 Pooling2dLayer* pool2dLayer = graph.AddLayer<Pooling2dLayer>(pooling2dDescriptor, "pool2d");
229 (pool2dLayer->GetParameters() == pooling2dDescriptor); in __anonca9d4ff90502()
257 … return (pool2dLayerParamsNoPad == pooling2dDescriptor) && (pool2dLayerParams.m_PadLeft == 1) && in __anonca9d4ff90602()
[all …]
|
| D | FoldPadIntoQuantizedAveragePooling2DTests.cpp | 42 Pooling2dDescriptor pooling2dDescriptor; in FoldPadIntoQuantizedAvgPoolTest() local
43 pooling2dDescriptor.m_PoolType = PoolingAlgorithm::Average; in FoldPadIntoQuantizedAvgPoolTest()
44 pooling2dDescriptor.m_PoolWidth = 3; in FoldPadIntoQuantizedAvgPoolTest()
45 pooling2dDescriptor.m_PoolHeight = 3; in FoldPadIntoQuantizedAvgPoolTest()
46 pooling2dDescriptor.m_StrideX = 1; in FoldPadIntoQuantizedAvgPoolTest()
47 pooling2dDescriptor.m_StrideY = 1; in FoldPadIntoQuantizedAvgPoolTest()
48 pooling2dDescriptor.m_DataLayout = DataLayout::NHWC; in FoldPadIntoQuantizedAvgPoolTest()
50 IConnectableLayer* pool2dLayer = network->AddPooling2dLayer(pooling2dDescriptor, "pool2d"); in FoldPadIntoQuantizedAvgPoolTest()
83 … return (pool2dLayerParamsNoPad == pooling2dDescriptor) && (pool2dLayerParams.m_PadLeft == 1) && in FoldPadIntoQuantizedAvgPoolTest()
|
| /external/armnn/src/armnnSerializer/ |
| D | Serializer.cpp | 1080 const armnn::Pooling2dDescriptor& pooling2dDescriptor, in SerializePooling2dLayer() argument
1088 GetFlatBufferPoolingAlgorithm(pooling2dDescriptor.m_PoolType), in SerializePooling2dLayer()
1089 pooling2dDescriptor.m_PadLeft, in SerializePooling2dLayer()
1090 pooling2dDescriptor.m_PadRight, in SerializePooling2dLayer()
1091 pooling2dDescriptor.m_PadTop, in SerializePooling2dLayer()
1092 pooling2dDescriptor.m_PadBottom, in SerializePooling2dLayer()
1093 pooling2dDescriptor.m_PoolWidth, in SerializePooling2dLayer()
1094 pooling2dDescriptor.m_PoolHeight, in SerializePooling2dLayer()
1095 pooling2dDescriptor.m_StrideX, in SerializePooling2dLayer()
1096 pooling2dDescriptor.m_StrideY, in SerializePooling2dLayer()
[all …]
|
| D | Serializer.hpp | 255 const armnn::Pooling2dDescriptor& pooling2dDescriptor,
|
| /external/armnn/src/armnnDeserializer/ |
| D | Deserializer.hpp | 64 … static armnn::Pooling2dDescriptor GetPooling2dDescriptor(Pooling2dDescriptor pooling2dDescriptor,
|
| D | Deserializer.cpp | 2531 auto pooling2dDescriptor = GetPooling2dDescriptor(pooling2dDes, layerIndex); in ParsePooling2d() local
2533 IConnectableLayer* layer = m_Network->AddPooling2dLayer(pooling2dDescriptor, layerName.c_str()); in ParsePooling2d()
|
| /external/armnn/src/armnn/ |
| D | Network.hpp | 158 IConnectableLayer* AddPooling2dLayer(const Pooling2dDescriptor& pooling2dDescriptor,
|
| D | Network.cpp | 349 IConnectableLayer* INetwork::AddPooling2dLayer(const Pooling2dDescriptor& pooling2dDescriptor, in AddPooling2dLayer() argument
352 return pNetworkImpl->AddPooling2dLayer(pooling2dDescriptor, name); in AddPooling2dLayer()
2165 IConnectableLayer* NetworkImpl::AddPooling2dLayer(const Pooling2dDescriptor& pooling2dDescriptor, in AddPooling2dLayer() argument
2168 return m_Graph->AddLayer<Pooling2dLayer>(pooling2dDescriptor, name); in AddPooling2dLayer()
|
| /external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| D | armnn_network.i | 835 … armnn::IConnectableLayer* AddPooling2dLayer(const armnn::Pooling2dDescriptor& pooling2dDescriptor,
|
| /external/armnn/include/armnn/ |
| D | INetwork.hpp | 486 IConnectableLayer* AddPooling2dLayer(const Pooling2dDescriptor& pooling2dDescriptor,
|