diff options
Diffstat (limited to 'src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp b/src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp index 764983f3b9..8ef5ecc203 100644 --- a/src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp +++ b/src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp @@ -62,8 +62,10 @@ void AsyncThreadedEndToEndTestImpl(INetworkPtr network, inputTensors.reserve(inputTensorData.size()); for (auto&& it : inputTensorData[i]) { + TensorInfo inputTensorInfo = runtime->GetInputTensorInfo(networkId, it.first); + inputTensorInfo.SetConstant(true); inputTensors.push_back({it.first, - ConstTensor(runtime->GetInputTensorInfo(networkId, it.first), it.second.data())}); + ConstTensor(inputTensorInfo, it.second.data())}); } outputTensors.reserve(expectedOutputData.size()); @@ -146,8 +148,10 @@ void AsyncEndToEndTestImpl(INetworkPtr network, inputTensors.reserve(inputTensorData.size()); for (auto&& it : inputTensorData) { + TensorInfo inputTensorInfo = runtime->GetInputTensorInfo(networkId, it.first); + inputTensorInfo.SetConstant(true); inputTensors.push_back({it.first, - ConstTensor(runtime->GetInputTensorInfo(networkId, it.first), it.second.data())}); + ConstTensor(inputTensorInfo, it.second.data())}); } OutputTensors outputTensors; |