diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 125a763ff4..15ca36d906 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -4205,7 +4205,10 @@ BindingPointInfo TfLiteParserImpl::GetNetworkInputBindingInfo(size_t subgraphId, if (input.second->name == name) { auto bindingId = GenerateLayerBindingId(subgraphId, input.first); - return std::make_pair(bindingId, ToTensorInfo(input.second)); + auto inputTensorInfo = ToTensorInfo(input.second); + // Input tensors are always treated as constant tensors during network execution. + inputTensorInfo.SetConstant(true); + return std::make_pair(bindingId, inputTensorInfo); } } |