Searched refs:convDescriptor (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/src/backends/backendsCommon/test/ |
| D | OptimizationViewsTests.cpp | 63 Convolution2dDescriptor convDescriptor; variable
65 Layer* const convLayer1 = baseGraph.AddLayer<Convolution2dLayer>(convDescriptor, "conv1");
66 Layer* const convLayer2 = baseGraph.AddLayer<Convolution2dLayer>(convDescriptor, "conv2");
134 DepthwiseConvolution2dDescriptor convDescriptor; variable
140 view.GetINetwork()->AddDepthwiseConvolution2dLayer(convDescriptor,
144 view.GetINetwork()->AddDepthwiseConvolution2dLayer(convDescriptor,
198 Convolution2dDescriptor convDescriptor; variable
200 Layer* const convLayer1 = baseGraph.AddLayer<Convolution2dLayer>(convDescriptor, "conv1");
201 Layer* const convLayer2 = baseGraph.AddLayer<Convolution2dLayer>(convDescriptor, "conv2");
|
| /external/armnn/src/armnn/optimizations/ |
| D | FuseBatchNorm.hpp | 53 auto convDescriptor = convLayer->GetParameters(); in Run() local
64 armnnUtils::DataLayoutIndexed dataLayout(convDescriptor.m_DataLayout); in Run()
107 else if (convDescriptor.m_DataLayout == DataLayout::NHWC) in Run()
130 bool biasWasEnabledBeforeOpt = convDescriptor.m_BiasEnabled; in Run()
154 convDescriptor.m_BiasEnabled = true; in Run()
168 convDescriptor, in Run()
|
| /external/armnn/src/armnn/test/ |
| D | SubgraphViewTests.cpp | 248 Convolution2dDescriptor convDescriptor; variable
249 Layer* const convLayer1 = graph.AddLayer<Convolution2dLayer>(convDescriptor, "conv1");
250 Layer* const convLayer2 = graph.AddLayer<Convolution2dLayer>(convDescriptor, "conv2");
285 Convolution2dDescriptor convDescriptor; variable
286 Layer* const convLayer1 = graph.AddLayer<Convolution2dLayer>(convDescriptor, "conv1");
287 Layer* const convLayer2 = graph.AddLayer<Convolution2dLayer>(convDescriptor, "conv2");
343 Convolution2dDescriptor convDescriptor; variable
344 Layer* const convLayer1 = graph.AddLayer<Convolution2dLayer>(convDescriptor, "conv1");
345 Layer* const convLayer2 = graph.AddLayer<Convolution2dLayer>(convDescriptor, "conv2");
388 Convolution2dDescriptor convDescriptor; variable
[all …]
|
| /external/armnn/src/armnn/test/optimizations/ |
| D | FoldPadTests.cpp | 681 Convolution2dDescriptor convDescriptor; variable
682 convDescriptor.m_DataLayout = DataLayout::NHWC;
683 convDescriptor.m_StrideX = 1;
684 convDescriptor.m_StrideY = 1;
685 convDescriptor.m_BiasEnabled = true;
697 IConnectableLayer* conv2dLayer = network->AddConvolution2dLayer(convDescriptor, "Conv2D");
805 DepthwiseConvolution2dDescriptor convDescriptor; variable
806 convDescriptor.m_DataLayout = DataLayout::NHWC;
807 convDescriptor.m_StrideX = 1;
808 convDescriptor.m_StrideY = 1;
[all …]
|