Logo ROOT  
Reference Guide
Loading...
Searching...
No Matches
RModelParser_PyTorch.h File Reference

Namespaces

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

Functions

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.