Logo ROOT  

Reference Guide
 
Loading...
Searching...
No Matches
Typedefs | Functions | Variables
TMVA::Experimental::SOFIE::INTERNAL Namespace Reference

Typedefs

using factoryMethodMap = std::unordered_map< std::string, std::unique_ptr< ROperator >(*)(const onnx::NodeProto &, const onnx::GraphProto &, std::unordered_map< std::string, ETensorType > &)>
 

Functions

std::unique_ptr< ROperator > make_ROperator (size_t idx, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Add (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_BatchNormalization (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Conv (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Gemm (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_GRU (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_LSTM (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Pool (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Relu (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Reshape (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_RNN (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Selu (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Sigmoid (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Slice (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 
std::unique_ptr< ROperator > make_ROperator_Transpose (const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
 

Variables

const factoryMethodMap mapOptypeOperator
 

Typedef Documentation

◆ factoryMethodMap

using TMVA::Experimental::SOFIE::INTERNAL::factoryMethodMap = typedef std::unordered_map<std::string, std::unique_ptr<ROperator> (*)(const onnx::NodeProto&, const onnx::GraphProto&, std::unordered_map<std::string, ETensorType>&)>

Definition at line 30 of file RModelParser_ONNX.cxx.

Function Documentation

◆ make_ROperator()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator ( size_t  idx,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 56 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Add()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Add ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 69 of file RModelParser_ONNX.cxx.

◆ make_ROperator_BatchNormalization()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_BatchNormalization ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 870 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Conv()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Conv ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 389 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Gemm()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Gemm ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 240 of file RModelParser_ONNX.cxx.

◆ make_ROperator_GRU()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_GRU ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 297 of file RModelParser_ONNX.cxx.

◆ make_ROperator_LSTM()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_LSTM ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 767 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Pool()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Pool ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 453 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Relu()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Relu ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 144 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Reshape()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Reshape ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 530 of file RModelParser_ONNX.cxx.

◆ make_ROperator_RNN()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_RNN ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 679 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Selu()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Selu ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 176 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Sigmoid()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Sigmoid ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 208 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Slice()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Slice ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 598 of file RModelParser_ONNX.cxx.

◆ make_ROperator_Transpose()

std::unique_ptr< ROperator > TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Transpose ( const onnx::NodeProto &  nodeproto,
const onnx::GraphProto &  graphproto,
std::unordered_map< std::string, ETensorType > &  tensor_type 
)

Definition at line 104 of file RModelParser_ONNX.cxx.

Variable Documentation

◆ mapOptypeOperator

const factoryMethodMap TMVA::Experimental::SOFIE::INTERNAL::mapOptypeOperator
Initial value:
= {
{"Gemm", &make_ROperator_Gemm},
{"Transpose", &make_ROperator_Transpose},
{"Relu", &make_ROperator_Relu},
{"Conv", &make_ROperator_Conv},
{"RNN", &make_ROperator_RNN},
{"Selu", &make_ROperator_Selu},
{"Sigmoid", &make_ROperator_Sigmoid},
{"LSTM", &make_ROperator_LSTM},
{"GRU", &make_ROperator_GRU},
{"BatchNormalization", &make_ROperator_BatchNormalization},
{"AveragePool", &make_ROperator_Pool},
{"GlobalAveragePool", &make_ROperator_Pool},
{"MaxPool", &make_ROperator_Pool},
{"Add", &make_ROperator_Add},
{"Reshape", &make_ROperator_Reshape},
{"Flatten", &make_ROperator_Reshape},
{"Slice", &make_ROperator_Slice},
{"Squeeze", &make_ROperator_Reshape},
{"Unsqueeze", &make_ROperator_Reshape},
{"Flatten", &make_ROperator_Reshape}
}
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Gemm
std::unique_ptr< ROperator > make_ROperator_Gemm(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:240
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_LSTM
std::unique_ptr< ROperator > make_ROperator_LSTM(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:767
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Pool
std::unique_ptr< ROperator > make_ROperator_Pool(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:453
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_GRU
std::unique_ptr< ROperator > make_ROperator_GRU(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:297
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Reshape
std::unique_ptr< ROperator > make_ROperator_Reshape(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:530
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_BatchNormalization
std::unique_ptr< ROperator > make_ROperator_BatchNormalization(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:870
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Sigmoid
std::unique_ptr< ROperator > make_ROperator_Sigmoid(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:208
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Add
std::unique_ptr< ROperator > make_ROperator_Add(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:69
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Slice
std::unique_ptr< ROperator > make_ROperator_Slice(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:598
TMVA::Experimental::SOFIE::INTERNAL::make_ROperator_Transpose
std::unique_ptr< ROperator > make_ROperator_Transpose(const onnx::NodeProto &nodeproto, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
Definition RModelParser_ONNX.cxx:104

Definition at line 31 of file RModelParser_ONNX.cxx.