RModelParser_ONNX.cxx File Reference

#include "TMVA/RModelParser_ONNX.hxx"
#include "onnx_proto3.pb.h"
#include <string>
#include <memory>
#include <cassert>

Include dependency graph for RModelParser_ONNX.cxx:

Namespaces
namespace	TMVA
	create variable transformations
namespace	TMVA::Experimental
namespace	TMVA::Experimental::SOFIE
namespace	TMVA::Experimental::SOFIE::INTERNAL

Typedefs
using	TMVA::Experimental::SOFIE::INTERNAL::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 >	TMVA::Experimental::SOFIE::INTERNAL::make_ROperator (size_t idx, const onnx::GraphProto &graphproto, std::unordered_map< std::string, ETensorType > &tensor_type)
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)
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)
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)
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)
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)
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)
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)
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)
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)
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)
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)
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)
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)
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)

Variables
const factoryMethodMap	TMVA::Experimental::SOFIE::INTERNAL::mapOptypeOperator