diff options
author | Colm Donelan <colm.donelan@arm.com> | 2021-11-09 20:39:10 +0000 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2021-11-09 20:39:10 +0000 |
commit | 4bc993b54a0d9d3595827425e6171c94546eb371 (patch) | |
tree | 17ed347f277571f95e919073f937f24bced513e2 | |
parent | 7f82e085cb0c7eee321922ce756049ac9c387fb7 (diff) | |
download | armnn-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.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); } } |