/aosp_15_r20/external/armnn/src/armnn/test/optimizations/ |
H A D | ConvertConstDequantisationLayersToConstLayersTest.cpp | 106 Convolution2dDescriptor convolution2dDescriptor; variable 107 convolution2dDescriptor.m_BiasEnabled = false; 108 convolution2dDescriptor.m_StrideX = 1; 109 convolution2dDescriptor.m_StrideY = 1; 110 convolution2dDescriptor.m_DataLayout = DataLayout::NHWC; 111 auto conv2d = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "conv2d");
|
H A D | FuseBatchNormTests.cpp | 114 ConvDescriptorType convolution2dDescriptor; in CreateNetwork() local 115 convolution2dDescriptor.m_BiasEnabled = false; in CreateNetwork() 116 convolution2dDescriptor.m_DataLayout = DataLayout::NHWC; in CreateNetwork() 117 convolution2dDescriptor.m_StrideX = 1; in CreateNetwork() 118 convolution2dDescriptor.m_StrideY = 1; in CreateNetwork() 164 convolution2dDescriptor, in CreateNetwork() 183 … convolution2dDescriptor, in CreateNetwork()
|
H A D | FoldPadTests.cpp | 42 Convolution2dDescriptor convolution2dDescriptor; variable 43 convolution2dDescriptor.m_BiasEnabled = false; 44 convolution2dDescriptor.m_StrideX = 1; 45 convolution2dDescriptor.m_StrideY = 1; 46 convolution2dDescriptor.m_DataLayout = DataLayout::NHWC; 55 …Convolution2dLayer* conv2dLayer = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "con…
|
/aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
H A D | armnn_network.i | 551 const armnn::DepthwiseConvolution2dDescriptor& convolution2dDescriptor, 1187 …nectableLayer* AddConvolution2dLayer(const armnn::Convolution2dDescriptor& convolution2dDescriptor, 1190 return $self->AddConvolution2dLayer(convolution2dDescriptor, name); 1206 const armnn::DepthwiseConvolution2dDescriptor& convolution2dDescriptor, 1209 return $self->AddDepthwiseConvolution2dLayer(convolution2dDescriptor, name);
|
/aosp_15_r20/external/armnn/src/armnn/test/ |
H A D | ConstTensorLayerVisitor.hpp | 23 explicit TestConvolution2dLayerVisitor(const Convolution2dDescriptor& convolution2dDescriptor, in TestConvolution2dLayerVisitor() argument 26 , m_Descriptor(convolution2dDescriptor) in TestConvolution2dLayerVisitor() 55 void CheckDescriptor(const Convolution2dDescriptor& convolution2dDescriptor); 96 void CheckDescriptor(const DepthwiseConvolution2dDescriptor& convolution2dDescriptor);
|
H A D | ConstTensorLayerVisitor.cpp | 14 …tConvolution2dLayerVisitor::CheckDescriptor(const Convolution2dDescriptor &convolution2dDescriptor) in CheckDescriptor() argument 16 CHECK(m_Descriptor.m_PadLeft == convolution2dDescriptor.m_PadLeft); in CheckDescriptor() 17 CHECK(m_Descriptor.m_PadRight == convolution2dDescriptor.m_PadRight); in CheckDescriptor() 18 CHECK(m_Descriptor.m_PadTop == convolution2dDescriptor.m_PadTop); in CheckDescriptor() 19 CHECK(m_Descriptor.m_PadBottom == convolution2dDescriptor.m_PadBottom); in CheckDescriptor() 20 CHECK(m_Descriptor.m_StrideX == convolution2dDescriptor.m_StrideX); in CheckDescriptor() 21 CHECK(m_Descriptor.m_StrideY == convolution2dDescriptor.m_StrideY); in CheckDescriptor() 22 CHECK(m_Descriptor.m_BiasEnabled == convolution2dDescriptor.m_BiasEnabled); in CheckDescriptor() 23 CHECK(m_Descriptor.m_DataLayout == convolution2dDescriptor.m_DataLayout); in CheckDescriptor() 27 const DepthwiseConvolution2dDescriptor& convolution2dDescriptor) in CheckDescriptor() argument [all …]
|
H A D | OptimizerTests.cpp | 857 Convolution2dDescriptor convolution2dDescriptor; variable 858 convolution2dDescriptor.m_BiasEnabled = false; 859 convolution2dDescriptor.m_DataLayout = DataLayout::NHWC; 889 auto conv = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "convolution"); 906 if (convolution2dDescriptor.m_BiasEnabled) 922 if (convolution2dDescriptor.m_BiasEnabled) 966 Convolution2dDescriptor convolution2dDescriptor; variable 970 auto conv = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "convolution");
|
/aosp_15_r20/external/armnn/src/armnn/ |
H A D | Network.hpp | 77 IConnectableLayer* AddConvolution2dLayer(const Convolution2dDescriptor& convolution2dDescriptor, 88 …er* AddDepthwiseConvolution2dLayer(const DepthwiseConvolution2dDescriptor& convolution2dDescriptor,
|
H A D | Network.cpp | 271 …tableLayer* INetwork::AddConvolution2dLayer(const Convolution2dDescriptor& convolution2dDescriptor, in AddConvolution2dLayer() argument 274 return pNetworkImpl->AddConvolution2dLayer(convolution2dDescriptor, name); in AddConvolution2dLayer() 292 const DepthwiseConvolution2dDescriptor& convolution2dDescriptor, in AddDepthwiseConvolution2dLayer() argument 295 return pNetworkImpl->AddDepthwiseConvolution2dLayer(convolution2dDescriptor, name); in AddDepthwiseConvolution2dLayer() 2114 …leLayer* NetworkImpl::AddConvolution2dLayer(const Convolution2dDescriptor& convolution2dDescriptor, in AddConvolution2dLayer() argument 2117 return m_Graph->AddLayer<Convolution2dLayer>(convolution2dDescriptor, name); in AddConvolution2dLayer() 2143 const DepthwiseConvolution2dDescriptor& convolution2dDescriptor, in AddDepthwiseConvolution2dLayer() argument 2146 return m_Graph->AddLayer<DepthwiseConvolution2dLayer>(convolution2dDescriptor, name); in AddDepthwiseConvolution2dLayer()
|
/aosp_15_r20/external/armnn/python/pyarmnn/test/ |
H A D | test_network.py | 256 layer = net.AddConvolution2dLayer(convolution2dDescriptor=ann.Convolution2dDescriptor()) 263 layer = net.AddConvolution2dLayer(convolution2dDescriptor=ann.Convolution2dDescriptor(),
|
/aosp_15_r20/external/armnn/include/armnn/ |
H A D | INetwork.hpp | 384 IConnectableLayer* AddConvolution2dLayer(const Convolution2dDescriptor& convolution2dDescriptor, 405 …er* AddDepthwiseConvolution2dLayer(const DepthwiseConvolution2dDescriptor& convolution2dDescriptor,
|
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
H A D | DynamicBackendTests.hpp | 1452 Convolution2dDescriptor convolution2dDescriptor; in CreateReferenceDynamicBackendTestImpl() local 1457 convolution2dDescriptor); in CreateReferenceDynamicBackendTestImpl() 1534 Convolution2dDescriptor convolution2dDescriptor; in CreateSampleDynamicBackendTestImpl() local 1539 convolution2dDescriptor); in CreateSampleDynamicBackendTestImpl()
|