diff options
author | Colm Donelan <colm.donelan@arm.com> | 2021-11-09 20:39:10 +0000 |
---|---|---|
committer | David Monahan <David.Monahan@arm.com> | 2021-11-10 10:14:37 +0000 |
commit | 0f04892cba2c2dd25c249d67311390675fb35ee9 (patch) | |
tree | 17ed347f277571f95e919073f937f24bced513e2 | |
parent | 60c8d93f33c373e56beb06573f1e389e92de1e9b (diff) | |
download | armnn-0f04892cba2c2dd25c249d67311390675fb35ee9.tar.gz |
IVGCVSW-6420 Setting a const tensor flag was missing inthe TfLite parser.
* The TfLiteparser was returning non const TensorInfo's for input tensor
BindingPointInfo.
Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: I20942aaef3afe7c91cf19fc3c007bec35828e69e
-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); } } |