HDK
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Public Member Functions | List of all members
onnxruntime::OnnxRuntimeOpSchemaRegistry Class Reference

#include <schema_registry.h>

+ Inheritance diagram for onnxruntime::OnnxRuntimeOpSchemaRegistry:

Public Member Functions

 OnnxRuntimeOpSchemaRegistry ()=default
 
common::Status SetBaselineAndOpsetVersionForDomain (const std::string &domain, int baseline_opset_version, int opset_version)
 
DomainToVersionMap GetLatestOpsetVersions (bool is_onnx_only) const override
 
common::Status RegisterOpSet (std::vector< ONNX_NAMESPACE::OpSchema > &schemas, const std::string &domain, int baseline_opset_version, int opset_version)
 
void GetSchemaAndHistory (const std::string &key, int maxInclusiveVersion, const std::string &domain, const ONNX_NAMESPACE::OpSchema **latest_schema, int *earliest_opset_where_unchanged) const override
 
bool empty () const
 
- Public Member Functions inherited from onnxruntime::IOnnxRuntimeOpSchemaCollection
const ONNX_NAMESPACE::OpSchema * GetSchema (const std::string &key, const int maxInclusiveVersion, const std::string &domain) const final
 

Detailed Description

OnnxRuntimeOpSchemaRegistry is used to provide supplement for built-in ONNX schemas. Each OnnxRuntimeOpSchemaRegistry must register complete opsets delta from a baseline version to max opset version. (Please notice that baseline opsets are not include in the delta)

For example, ONNXRuntime is build with ONNX 1.2 which is at opset7, to use ONNX opset8 and opset9, user could create a OnnxRuntimeOpSchemaRegistry and config it as {baseline_opset_version = 7, opset_version = 9} it means this OnnxRuntimeOpSchemaRegistry contains the complete delta from opset7 to opset9.

Definition at line 74 of file schema_registry.h.

Constructor & Destructor Documentation

onnxruntime::OnnxRuntimeOpSchemaRegistry::OnnxRuntimeOpSchemaRegistry ( )
default

Member Function Documentation

bool onnxruntime::OnnxRuntimeOpSchemaRegistry::empty ( void  ) const
inline

Definition at line 98 of file schema_registry.h.

DomainToVersionMap onnxruntime::OnnxRuntimeOpSchemaRegistry::GetLatestOpsetVersions ( bool  is_onnx_only) const
overridevirtual

Implements onnxruntime::IOnnxRuntimeOpSchemaCollection.

void onnxruntime::OnnxRuntimeOpSchemaRegistry::GetSchemaAndHistory ( const std::string &  key,
int  maxInclusiveVersion,
const std::string &  domain,
const ONNX_NAMESPACE::OpSchema **  latest_schema,
int *  earliest_opset_where_unchanged 
) const
overridevirtual

Implements onnxruntime::IOnnxRuntimeOpSchemaCollection.

common::Status onnxruntime::OnnxRuntimeOpSchemaRegistry::RegisterOpSet ( std::vector< ONNX_NAMESPACE::OpSchema > &  schemas,
const std::string &  domain,
int  baseline_opset_version,
int  opset_version 
)
common::Status onnxruntime::OnnxRuntimeOpSchemaRegistry::SetBaselineAndOpsetVersionForDomain ( const std::string &  domain,
int  baseline_opset_version,
int  opset_version 
)
The documentation for this class was generated from the following file: