1 files changed, 7 insertions, 0 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index 6e73f5275f..25ccbee45a 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -577,6 +577,13 @@ public:
                               m_OutputBindings[outputIndex].second.GetQuantizationOffset());
     }
 
+    QuantizationParams GetInputQuantizationParams(unsigned int inputIndex = 0u) const
+    {
+        CheckInputIndexIsValid(inputIndex);
+        return std::make_pair(m_InputBindings[inputIndex].second.GetQuantizationScale(),
+                              m_InputBindings[inputIndex].second.GetQuantizationOffset());
+    }
+
     std::vector<QuantizationParams> GetAllQuantizationParams() const
     {
         std::vector<QuantizationParams> quantizationParams;