diff options
Diffstat (limited to 'samples/DynamicSample.cpp')
-rw-r--r-- | samples/DynamicSample.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/samples/DynamicSample.cpp b/samples/DynamicSample.cpp index ffcc9de083..8a6ff92706 100644 --- a/samples/DynamicSample.cpp +++ b/samples/DynamicSample.cpp @@ -62,10 +62,12 @@ int main() }; std::vector<float> outputData(2); + TensorInfo inputTensorInfo = run->GetInputTensorInfo(networkIdentifier, 0); + inputTensorInfo.SetConstant(true); InputTensors inputTensors { - {0,armnn::ConstTensor(run->GetInputTensorInfo(networkIdentifier, 0), input0Data.data())}, - {1,armnn::ConstTensor(run->GetInputTensorInfo(networkIdentifier, 0), input1Data.data())} + {0,armnn::ConstTensor(inputTensorInfo, input0Data.data())}, + {1,armnn::ConstTensor(inputTensorInfo, 0), input1Data.data())} }; OutputTensors outputTensors { |