saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update max opset for NNAPI and CoreML. (#12831)

Browse source 
Update max opset for NNAPI and CoreML. Changes in opsets 16 and 17 don't require any updates.
This commit is contained in:
Scott McKay 2022-09-05 09:37:14 +10:00 committed by GitHub
parent 9e47eb68e0
commit ac4f1bf960
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/coreml/builders/impl/base_op_builder.h
View file

@ -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
} // namespace onnxruntime

2
onnxruntime/core/providers/nnapi/nnapi_builtin/builders/op_support_checker.cc
View file

@ -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