aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorColm Donelan <colm.donelan@arm.com>2021-11-09 20:39:10 +0000
committerDavid Monahan <David.Monahan@arm.com>2021-11-10 10:14:37 +0000
commit0f04892cba2c2dd25c249d67311390675fb35ee9 (patch)
tree17ed347f277571f95e919073f937f24bced513e2
parent60c8d93f33c373e56beb06573f1e389e92de1e9b (diff)
downloadarmnn-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.cpp5
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);
}
}