Definition at line 28 of file RModelParser_ONNX.hxx.
|
| RModelParser_ONNX () noexcept |
|
| ~RModelParser_ONNX () |
|
void | CheckGraph (const onnx::GraphProto &g, int &level, std::map< std::string, int > &missingOperators) |
|
bool | CheckModel (std::string filename, bool verbose=false) |
|
std::vector< std::string > | GetRegisteredOperators () |
|
ETensorType | GetTensorType (const std::string &name) |
|
bool | IsRegisteredOperator (const std::string &name) |
|
bool | IsRegisteredTensorType (const std::string &) |
|
std::unique_ptr< onnx::ModelProto > | LoadModel (std::string filename) |
|
RModel | Parse (std::string filename, bool verbose=false) |
|
void | ParseONNXGraph (RModel &model, const onnx::GraphProto &g, std::string name="") |
|
std::unique_ptr< ROperator > | ParseOperator (const size_t, const onnx::GraphProto &, const std::vector< size_t > &, const std::vector< int > &) |
|
void | RegisterOperator (const std::string &name, ParserFuncSignature func) |
|
void | RegisterTensorType (const std::string &, ETensorType) |
|
bool | Verbose () const |
|
#include <TMVA/RModelParser_ONNX.hxx>
◆ RModelParser_ONNX()
TMVA::Experimental::SOFIE::RModelParser_ONNX::RModelParser_ONNX |
( |
| ) |
|
|
noexcept |
◆ ~RModelParser_ONNX()
TMVA::Experimental::SOFIE::RModelParser_ONNX::~RModelParser_ONNX |
( |
| ) |
|
|
default |
◆ CheckGraph()
void TMVA::Experimental::SOFIE::RModelParser_ONNX::CheckGraph |
( |
const onnx::GraphProto & | g, |
|
|
int & | level, |
|
|
std::map< std::string, int > & | missingOperators ) |
◆ CheckModel()
bool TMVA::Experimental::SOFIE::RModelParser_ONNX::CheckModel |
( |
std::string | filename, |
|
|
bool | verbose = false ) |
◆ GetRegisteredOperators()
std::vector< std::string > TMVA::Experimental::SOFIE::RModelParser_ONNX::GetRegisteredOperators |
( |
| ) |
|
◆ GetTensorType()
ETensorType TMVA::Experimental::SOFIE::RModelParser_ONNX::GetTensorType |
( |
const std::string & | name | ) |
|
◆ IsRegisteredOperator()
bool TMVA::Experimental::SOFIE::RModelParser_ONNX::IsRegisteredOperator |
( |
const std::string & | name | ) |
|
◆ IsRegisteredTensorType()
bool TMVA::Experimental::SOFIE::RModelParser_ONNX::IsRegisteredTensorType |
( |
const std::string & | name | ) |
|
◆ LoadModel()
std::unique_ptr< onnx::ModelProto > TMVA::Experimental::SOFIE::RModelParser_ONNX::LoadModel |
( |
std::string | filename | ) |
|
◆ Parse()
RModel TMVA::Experimental::SOFIE::RModelParser_ONNX::Parse |
( |
std::string | filename, |
|
|
bool | verbose = false ) |
◆ ParseONNXGraph()
void TMVA::Experimental::SOFIE::RModelParser_ONNX::ParseONNXGraph |
( |
RModel & | model, |
|
|
const onnx::GraphProto & | g, |
|
|
std::string | name = "" ) |
◆ ParseOperator()
std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::RModelParser_ONNX::ParseOperator |
( |
const size_t | i, |
|
|
const onnx::GraphProto & | graphproto, |
|
|
const std::vector< size_t > & | nodes, |
|
|
const std::vector< int > & | children ) |
◆ RegisterOperator()
void TMVA::Experimental::SOFIE::RModelParser_ONNX::RegisterOperator |
( |
const std::string & | name, |
|
|
ParserFuncSignature | func ) |
◆ RegisterTensorType()
void TMVA::Experimental::SOFIE::RModelParser_ONNX::RegisterTensorType |
( |
const std::string & | name, |
|
|
ETensorType | type ) |
◆ Verbose()
bool TMVA::Experimental::SOFIE::RModelParser_ONNX::Verbose |
( |
| ) |
const |
|
inline |
◆ fFusedOperators
std::vector<bool> TMVA::Experimental::SOFIE::RModelParser_ONNX::fFusedOperators |
|
private |
◆ fOperatorsMapImpl
std::unique_ptr<OperatorsMapImpl> TMVA::Experimental::SOFIE::RModelParser_ONNX::fOperatorsMapImpl |
|
private |
◆ fTensorTypeMap
std::unordered_map<std::string, ETensorType> TMVA::Experimental::SOFIE::RModelParser_ONNX::fTensorTypeMap |
|
private |
◆ fVerbose
bool TMVA::Experimental::SOFIE::RModelParser_ONNX::fVerbose = false |
|
private |
tmva/sofie_parsers/inc/TMVA/RModelParser_ONNX.hxx
tmva/sofie_parsers/src/RModelParser_ONNX.cxx