diff options
Diffstat (limited to 'src/profiling/test/FileOnlyProfilingDecoratorTests.cpp')
| -rw-r--r-- | src/profiling/test/FileOnlyProfilingDecoratorTests.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp b/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp index 0d6f8422fc..a96de66ecd 100644 --- a/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp +++ b/src/profiling/test/FileOnlyProfilingDecoratorTests.cpp @@ -90,9 +90,11 @@ TEST_CASE("TestFileOnlyProfiling") outputData[i] = 3.0; } + 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 { @@ -213,9 +215,11 @@ TEST_CASE("DumpOutgoingValidFileEndToEnd") outputData[i] = 3.0; } + 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 { |