Home
last modified time | relevance | path

Searched refs:convolutionLayer (Results 1 – 3 of 3) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/
DLayerReleaseConstantDataTest.cpp85 auto* const convolutionLayer = graph.AddLayer<Convolution2dLayer>(layerDesc, "convolution"); variable
103 Connect(input, convolutionLayer, TensorInfo({ 2, 3, 8, 16 }, armnn::DataType::Float32));
104 weightsLayer->GetOutputSlot().Connect(convolutionLayer->GetInputSlot(1));
105 biasLayer->GetOutputSlot().Connect(convolutionLayer->GetInputSlot(2));
106 Connect(convolutionLayer, output, TensorInfo({ 2, 2, 2, 10 }, armnn::DataType::Float32));
113 convolutionLayer->ReleaseConstantData();
134 …auto* const convolutionLayer = graph.AddLayer<DepthwiseConvolution2dLayer>(layerDesc, "convolution… variable
152 Connect(input, convolutionLayer, TensorInfo({2, 3, 8, 16}, armnn::DataType::Float32));
153 weightsLayer->GetOutputSlot().Connect(convolutionLayer->GetInputSlot(1));
154 biasLayer->GetOutputSlot().Connect(convolutionLayer->GetInputSlot(2));
[all …]
/external/armnn/src/backends/neon/workloads/
DNeonConvolution2dWorkload.cpp107 auto convolutionLayer = std::make_unique<arm_compute::NEConvolutionLayer>(memoryManager); in NeonConvolution2dWorkload() local
108 convolutionLayer->configure(&input, in NeonConvolution2dWorkload()
119 convolutionLayer->get_convolution_method(input.info(), in NeonConvolution2dWorkload()
147 m_ConvolutionLayer.reset(convolutionLayer.release()); in NeonConvolution2dWorkload()
DNeonConvolution3dWorkload.cpp84 auto convolutionLayer = std::make_unique<arm_compute::NEConv3D>(); in NeonConvolution3dWorkload() local
85 convolutionLayer->configure(&input, in NeonConvolution3dWorkload()
103 m_ConvolutionLayer.reset(convolutionLayer.release()); in NeonConvolution3dWorkload()