IVGCVSW-1961: Add converter method for SQUEEZE to V1.1 section of HalPolicy
Change-Id: I15dffef32d394b13e57df134000b7dca4b8788af
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