Lines Matching refs:convolutionLayer

85     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));
155     Connect(convolutionLayer, output, TensorInfo({2, 9, 2, 10}, armnn::DataType::Float32));
162     convolutionLayer->ReleaseConstantData();