aboutsummaryrefslogtreecommitdiff
path: root/delegate/src/Quantization.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/src/Quantization.hpp')
-rw-r--r--delegate/src/Quantization.hpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/delegate/src/Quantization.hpp b/delegate/src/Quantization.hpp
index 565f1e37c0..3c274c6ff5 100644
--- a/delegate/src/Quantization.hpp
+++ b/delegate/src/Quantization.hpp
@@ -73,6 +73,15 @@ TfLiteStatus VisitDequantizeOperator(DelegateData& delegateData,
armnn::IOutputSlot& outputSlot = dequantizeLayer->GetOutputSlot(0);
outputSlot.SetTensorInfo(outputTensorInfo);
+ auto inputsTensorsProcess = ProcessInputs(dequantizeLayer,
+ delegateData,
+ tfLiteContext,
+ tfLiteNode);
+ if (inputsTensorsProcess == kTfLiteError)
+ {
+ return inputsTensorsProcess;
+ }
+
return Connect(dequantizeLayer, tfLiteNode, delegateData);
}