aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/test/ActivationSerializationTests.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/test/ActivationSerializationTests.cpp')
-rw-r--r--src/armnnSerializer/test/ActivationSerializationTests.cpp4
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};