Allow graphs with ONNX opseet 18 to be executed by ORT

microsoft · Sep 7, 2022 · c213a27 · c213a27
1 parent 400195a
commit c213a27
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 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
@@ -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 17; }
+  virtual int GetMaxSupportedOpSet(const Node& /* node */) const { return 18; }
 
  private:
   bool HasSupportedOpSet(const Node& node, const logging::Logger& logger) const;

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
@@ -338,7 +338,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 17; }
+  virtual int GetMaxSupportedOpSet(const NodeUnit& /* node_unit */) const { return 18; }
 
   // Check if this node_unit's type is supported
   // SingleNode type NodeUnit is supported