RModelParser_PyTorch.cxx File Reference

#include "TMVA/RModelParser_PyTorch.h"
#include <Python.h>

Include dependency graph for RModelParser_PyTorch.cxx:

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

Typedefs
using	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::PyTorchMethodMap = std::unordered_map<std::string, std::unique_ptr<ROperator> (*)(PyObject* fNode)>

Functions
std::unique_ptr< ROperator >	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::MakePyTorchConv (PyObject *fNode)
	Prepares a ROperator_Conv object.
std::unique_ptr< ROperator >	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::MakePyTorchGemm (PyObject *fNode)
	Prepares a ROperator_Gemm object.
std::unique_ptr< ROperator >	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::MakePyTorchNode (PyObject *fNode)
	Prepares equivalent ROperator with respect to PyTorch ONNX node.
std::unique_ptr< ROperator >	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::MakePyTorchRelu (PyObject *fNode)
	Prepares a ROperator_Relu object.
std::unique_ptr< ROperator >	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::MakePyTorchSelu (PyObject *fNode)
	Prepares a ROperator_Selu object.
std::unique_ptr< ROperator >	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::MakePyTorchSigmoid (PyObject *fNode)
	Prepares a ROperator_Sigmoid object.
std::unique_ptr< ROperator >	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::MakePyTorchTranspose (PyObject *fNode)
	Prepares a ROperator_Transpose object.
RModel	TMVA::Experimental::SOFIE::PyTorch::Parse (std::string filepath, std::vector< std::vector< size_t > > inputShapes)
	Overloaded Parser function for translating PyTorch .pt model into a RModel object.
RModel	TMVA::Experimental::SOFIE::PyTorch::Parse (std::string filepath, std::vector< std::vector< size_t > > inputShapes, std::vector< ETensorType > dtype)
	Parser function for translating PyTorch .pt model into a RModel object.

Variables
const PyTorchMethodMap	TMVA::Experimental::SOFIE::PyTorch::INTERNAL::mapPyTorchNode