diff options
Diffstat (limited to 'src/armnn/test/OptimizerTests.cpp')
-rw-r--r-- | src/armnn/test/OptimizerTests.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/armnn/test/OptimizerTests.cpp b/src/armnn/test/OptimizerTests.cpp index 3dd55279c6..b78863dddc 100644 --- a/src/armnn/test/OptimizerTests.cpp +++ b/src/armnn/test/OptimizerTests.cpp @@ -885,11 +885,11 @@ TEST_CASE("OptimizeForExclusiveConnectionsFuseTest") // Define the network Graph graph; - auto input = graph.AddLayer<InputLayer>(0, "input"); + auto input = graph.AddLayer<InputLayer>(0, "input"); auto weightsLayer = graph.AddLayer<ConstantLayer>("Weights"); - auto conv = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "convolution"); - auto batchNorm = graph.AddLayer<BatchNormalizationLayer>(batchNormDescriptor, "batchNorm"); - auto output = graph.AddLayer<OutputLayer>(0, "output"); + auto conv = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "convolution"); + auto batchNorm = graph.AddLayer<BatchNormalizationLayer>(batchNormDescriptor, "batchNorm"); + auto output = graph.AddLayer<OutputLayer>(0, "output"); // Set layer information input->GetOutputSlot().SetTensorInfo(inputInfo); |