diff options
Diffstat (limited to 'delegate/src/DelegateUtils.hpp')
| -rw-r--r-- | delegate/src/DelegateUtils.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/delegate/src/DelegateUtils.hpp b/delegate/src/DelegateUtils.hpp index b04baac36e..2d1651842a 100644 --- a/delegate/src/DelegateUtils.hpp +++ b/delegate/src/DelegateUtils.hpp @@ -482,6 +482,11 @@ armnn::ConstTensor CreateConstTensor(const TfLiteTensor* tfLiteTensor, "TfLiteArmnnDelegate: Not constant allocation type: " + std::to_string(tfLiteTensor->allocation_type)); } + if(tflite::IsConstantTensor(tfLiteTensor)) + { + tensorInfo.SetConstant(); + } + if (permutationVector.has_value() && permutationVector.value().GetSize() > 0 && permutationData != nullptr) { // Permute tensor info |