diff options
Diffstat (limited to 'src/armnn/test/OptimizerTests.cpp')
-rw-r--r-- | src/armnn/test/OptimizerTests.cpp | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/src/armnn/test/OptimizerTests.cpp b/src/armnn/test/OptimizerTests.cpp index a7277b78b5..6a13dc6456 100644 --- a/src/armnn/test/OptimizerTests.cpp +++ b/src/armnn/test/OptimizerTests.cpp @@ -478,11 +478,10 @@ void CreateDepthwiseConvolution2dGraph(Graph &graph, const unsigned int* inputSh { armnn::TensorInfo inputInfo(4, inputShape, DataType::Float32); armnn::TensorInfo outputInfo(4, outputShape, DataType::Float32); + armnn::TensorInfo weightsInfo(TensorShape(4, weightsShape), armnn::DataType::Float32, 0.0f, 0, true); std::vector<float> weightsVector(18); - armnn::ConstTensor weights( - armnn::TensorInfo(4, weightsShape, armnn::DataType::Float32, 0.0f, 0, true), - weightsVector); + armnn::ConstTensor weights(weightsInfo, weightsVector); DepthwiseConvolution2dDescriptor desc; desc.m_BiasEnabled = false; @@ -490,15 +489,19 @@ void CreateDepthwiseConvolution2dGraph(Graph &graph, const unsigned int* inputSh desc.m_StrideY = 1; desc.m_DataLayout = dataLayout; - Layer* input = graph.AddLayer<InputLayer>(0, "input"); - input->GetOutputSlot().SetTensorInfo(inputInfo); - + InputLayer* input = graph.AddLayer<InputLayer>(0, "input"); DepthwiseConvolution2dLayer* layer = graph.AddLayer<DepthwiseConvolution2dLayer>(desc, "depthwiseConv2d"); - layer->m_Weight = std::make_unique<armnn::ScopedTensorHandle>(weights); + ConstantLayer* weightsLayer = graph.AddLayer<ConstantLayer>("weights"); + OutputLayer* output = graph.AddLayer<OutputLayer>(0, "output"); + + input->GetOutputSlot().SetTensorInfo(inputInfo); layer->GetOutputSlot().SetTensorInfo(outputInfo); + weightsLayer->GetOutputSlot().SetTensorInfo(weightsInfo); + + weightsLayer->m_LayerOutput = std::make_unique<armnn::ScopedTensorHandle>(weights); - Layer* output = graph.AddLayer<OutputLayer>(0, "output"); input->GetOutputSlot().Connect(layer->GetInputSlot(0)); + weightsLayer->GetOutputSlot().Connect(layer->GetInputSlot(1)); layer->GetOutputSlot().Connect(output->GetInputSlot(0)); } |