diff options
Diffstat (limited to 'src/armnnSerializer/test/ActivationSerializationTests.cpp')
-rw-r--r-- | src/armnnSerializer/test/ActivationSerializationTests.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnnSerializer/test/ActivationSerializationTests.cpp b/src/armnnSerializer/test/ActivationSerializationTests.cpp index fb99e0bc3e..341752dd67 100644 --- a/src/armnnSerializer/test/ActivationSerializationTests.cpp +++ b/src/armnnSerializer/test/ActivationSerializationTests.cpp @@ -84,9 +84,11 @@ TEST_CASE("ActivationSerialization") run->LoadNetwork(networkIdentifier, std::move(deserializedOptimized)); std::vector<float> inputData {0.0f, -5.3f, 42.0f, -42.0f}; + armnn::TensorInfo inputTensorInfo = run->GetInputTensorInfo(networkIdentifier, 0); + inputTensorInfo.SetConstant(true); armnn::InputTensors inputTensors { - {0, armnn::ConstTensor(run->GetInputTensorInfo(networkIdentifier, 0), inputData.data())} + {0, armnn::ConstTensor(inputTensorInfo, inputData.data())} }; std::vector<float> expectedOutputData {0.0f, 0.0f, 42.0f, 0.0f}; |