| /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()
|