diff options
Diffstat (limited to 'src/armnn/test/DebugCallbackTest.cpp')
-rw-r--r-- | src/armnn/test/DebugCallbackTest.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/test/DebugCallbackTest.cpp b/src/armnn/test/DebugCallbackTest.cpp index 48e2c15a79..69b9736f92 100644 --- a/src/armnn/test/DebugCallbackTest.cpp +++ b/src/armnn/test/DebugCallbackTest.cpp @@ -71,9 +71,11 @@ TEST_CASE("RuntimeRegisterDebugCallback") std::vector<float> inputData({-2, -1, 0, 1, 2}); std::vector<float> outputData(5); + TensorInfo inputTensorInfo = runtime->GetInputTensorInfo(netId, 0); + inputTensorInfo.SetConstant(true); InputTensors inputTensors { - {0, ConstTensor(runtime->GetInputTensorInfo(netId, 0), inputData.data())} + {0, ConstTensor(inputTensorInfo, inputData.data())} }; OutputTensors outputTensors { |