diff options
Diffstat (limited to 'ConversionUtils.hpp')
-rw-r--r-- | ConversionUtils.hpp | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index a812183d..165c63b7 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -464,11 +464,15 @@ namespace armnn_driver 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 @@ bool ConvertPooling2d(const HalOperation& operation, } } -} // namespace armnn_driver
\ No newline at end of file +} // namespace armnn_driver |