aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorColm Donelan <colm.donelan@arm.com>2021-11-09 20:39:10 +0000
committerColm Donelan <colm.donelan@arm.com>2021-11-09 20:39:10 +0000
commit4bc993b54a0d9d3595827425e6171c94546eb371 (patch)
tree17ed347f277571f95e919073f937f24bced513e2
parent7f82e085cb0c7eee321922ce756049ac9c387fb7 (diff)
downloadarmnn-4bc993b54a0d9d3595827425e6171c94546eb371.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);
}
}