diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp
index a812183..165c63b 100644
--- a/ConversionUtils.hpp
+++ b/ConversionUtils.hpp
@@ -464,11 +464,15 @@
 using namespace android::nn;
 
 template<typename HalOperation, typename HalModel>
-const Operand* GetInputOperand(const HalOperation& operation, uint32_t inputIndex, const HalModel& model)
+const Operand* GetInputOperand(const HalOperation& operation, uint32_t inputIndex, const HalModel& model,
+                               bool failOnIndexOutOfBounds = true)
 {
     if (inputIndex >= operation.inputs.size())
     {
-        Fail("%s: invalid input index: %i out of %i", __func__, inputIndex, operation.inputs.size());
+        if (failOnIndexOutOfBounds)
+        {
+            Fail("%s: invalid input index: %i out of %i", __func__, inputIndex, operation.inputs.size());
+        }
         return nullptr;
     }
 
@@ -1036,4 +1040,4 @@
     }
 }
 
-} // namespace armnn_driver
\ No newline at end of file
+} // namespace armnn_driver