Home
last modified time | relevance | path

Searched refs:convolution2dDescriptor (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DConvertConstDequantisationLayersToConstLayersTest.cpp106 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 DFuseBatchNormTests.cpp114 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()
183convolution2dDescriptor, in CreateNetwork()
H A DFoldPadTests.cpp42 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 Darmnn_network.i551 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 DConstTensorLayerVisitor.hpp23 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 DConstTensorLayerVisitor.cpp14 …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 DOptimizerTests.cpp857 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 DNetwork.hpp77 IConnectableLayer* AddConvolution2dLayer(const Convolution2dDescriptor& convolution2dDescriptor,
88 …er* AddDepthwiseConvolution2dLayer(const DepthwiseConvolution2dDescriptor& convolution2dDescriptor,
H A DNetwork.cpp271 …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 Dtest_network.py256 layer = net.AddConvolution2dLayer(convolution2dDescriptor=ann.Convolution2dDescriptor())
263 layer = net.AddConvolution2dLayer(convolution2dDescriptor=ann.Convolution2dDescriptor(),
/aosp_15_r20/external/armnn/include/armnn/
H A DINetwork.hpp384 IConnectableLayer* AddConvolution2dLayer(const Convolution2dDescriptor& convolution2dDescriptor,
405 …er* AddDepthwiseConvolution2dLayer(const DepthwiseConvolution2dDescriptor& convolution2dDescriptor,
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DDynamicBackendTests.hpp1452 Convolution2dDescriptor convolution2dDescriptor; in CreateReferenceDynamicBackendTestImpl() local
1457 convolution2dDescriptor); in CreateReferenceDynamicBackendTestImpl()
1534 Convolution2dDescriptor convolution2dDescriptor; in CreateSampleDynamicBackendTestImpl() local
1539 convolution2dDescriptor); in CreateSampleDynamicBackendTestImpl()