From ac4f1bf9603967212333cf9c9e5487145e7c91bc Mon Sep 17 00:00:00 2001 From: Scott McKay Date: Mon, 5 Sep 2022 09:37:14 +1000 Subject: [PATCH] Update max opset for NNAPI and CoreML. (#12831) Update max opset for NNAPI and CoreML. Changes in opsets 16 and 17 don't require any updates. --- .../core/providers/coreml/builders/impl/base_op_builder.h | 4 ++-- .../nnapi/nnapi_builtin/builders/op_support_checker.cc | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/onnxruntime/core/providers/coreml/builders/impl/base_op_builder.h b/onnxruntime/core/providers/coreml/builders/impl/base_op_builder.h index e8ec01f34c..059e767f4a 100644 --- a/onnxruntime/core/providers/coreml/builders/impl/base_op_builder.h +++ b/onnxruntime/core/providers/coreml/builders/impl/base_op_builder.h @@ -46,7 +46,7 @@ class BaseOpBuilder : public IOpBuilder { virtual bool HasSupportedInputsImpl(const Node& node, const logging::Logger& logger) const; virtual int GetMinSupportedOpSet(const Node& /* node */) const { return 1; } - virtual int GetMaxSupportedOpSet(const Node& /* node */) const { return 15; } + virtual int GetMaxSupportedOpSet(const Node& /* node */) const { return 17; } private: bool HasSupportedOpSet(const Node& node, const logging::Logger& logger) const; @@ -54,4 +54,4 @@ class BaseOpBuilder : public IOpBuilder { }; } // namespace coreml -} // namespace onnxruntime \ No newline at end of file +} // namespace onnxruntime diff --git a/onnxruntime/core/providers/nnapi/nnapi_builtin/builders/op_support_checker.cc b/onnxruntime/core/providers/nnapi/nnapi_builtin/builders/op_support_checker.cc index f5a033766a..54c4cc5a52 100644 --- a/onnxruntime/core/providers/nnapi/nnapi_builtin/builders/op_support_checker.cc +++ b/onnxruntime/core/providers/nnapi/nnapi_builtin/builders/op_support_checker.cc @@ -346,7 +346,7 @@ class BaseOpSupportChecker : public IOpSupportChecker { const OpSupportCheckParams& params) const; virtual int GetMinSupportedOpSet(const NodeUnit& /* node_unit */) const { return 1; } - virtual int GetMaxSupportedOpSet(const NodeUnit& /* node_unit */) const { return 15; } + virtual int GetMaxSupportedOpSet(const NodeUnit& /* node_unit */) const { return 17; } // Check if this node_unit's type is supported // SingleNode type NodeUnit is supported