diff options
Diffstat (limited to 'src/armnn/test/CreateWorkload.hpp')
-rw-r--r-- | src/armnn/test/CreateWorkload.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp index 00257ea090..0048646d45 100644 --- a/src/armnn/test/CreateWorkload.hpp +++ b/src/armnn/test/CreateWorkload.hpp @@ -926,7 +926,7 @@ std::unique_ptr<BatchToSpaceNdWorkload> CreateBatchToSpaceNdWorkloadTest(armnn:: Layer* const output = graph.AddLayer<OutputLayer>(0, "output"); // Connects up. - armnn::TensorInfo tensorInfo({1, 1}, DataType); + armnn::TensorInfo tensorInfo({1, 1, 1, 1}, DataType); Connect(input, layer, tensorInfo); Connect(layer, output, tensorInfo); |