diff options
Diffstat (limited to 'src/backends/cl/test/Fp16SupportTest.cpp')
-rw-r--r-- | src/backends/cl/test/Fp16SupportTest.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/backends/cl/test/Fp16SupportTest.cpp b/src/backends/cl/test/Fp16SupportTest.cpp index 1974d4d856..b30a447f9f 100644 --- a/src/backends/cl/test/Fp16SupportTest.cpp +++ b/src/backends/cl/test/Fp16SupportTest.cpp @@ -88,10 +88,12 @@ TEST_CASE("Fp16AdditionTest") 100.0_h, 200.0_h, 300.0_h, 400.0_h }; + TensorInfo inputTensorInfo = runtime->GetInputTensorInfo(netId, 0); + inputTensorInfo.SetConstant(true); InputTensors inputTensors { - {0,ConstTensor(runtime->GetInputTensorInfo(netId, 0), input1Data.data())}, - {1,ConstTensor(runtime->GetInputTensorInfo(netId, 0), input2Data.data())} + {0,ConstTensor(inputTensorInfo, input1Data.data())}, + {1,ConstTensor(inputTensorInfo, input2Data.data())} }; std::vector<Half> outputData(input1Data.size()); |