diff options
Diffstat (limited to 'src/armnn/test/NetworkTests.cpp')
| -rw-r--r-- | src/armnn/test/NetworkTests.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/test/NetworkTests.cpp b/src/armnn/test/NetworkTests.cpp index 25dab596fd..c1927e3601 100644 --- a/src/armnn/test/NetworkTests.cpp +++ b/src/armnn/test/NetworkTests.cpp @@ -74,7 +74,7 @@ TEST_CASE("NetworkModification") unsigned int dims[] = { 10,1,1,1 }; std::vector<float> convWeightsData(10); - armnn::ConstTensor weights(armnn::TensorInfo(4, dims, armnn::DataType::Float32), convWeightsData); + armnn::ConstTensor weights(armnn::TensorInfo(4, dims, armnn::DataType::Float32, 0.0f, 0, true), convWeightsData); armnn::Convolution2dDescriptor convDesc2d; armnn::IConnectableLayer* const convLayer = net.AddConvolution2dLayer(convDesc2d, @@ -123,7 +123,7 @@ TEST_CASE("NetworkModification") armnn::BatchNormalizationDescriptor batchNormDesc; - armnn::TensorInfo tensorInfo({ 1 }, armnn::DataType::Float32); + armnn::TensorInfo tensorInfo({ 1 }, armnn::DataType::Float32, 0.0f, 0, true); std::vector<float> data(tensorInfo.GetNumBytes() / sizeof(float)); armnn::ConstTensor invalidTensor(tensorInfo, data); |