Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
RModelParser_PyTorch.h File Reference
#include "TMVA/RModel.hxx"
#include "TMVA/SOFIE_common.hxx"
#include "TMVA/Types.h"
#include "TMVA/OperatorList.hxx"
#include "TMVA/PyMethodBase.h"
#include "Rtypes.h"
#include "TString.h"
Include dependency graph for RModelParser_PyTorch.h:
This graph shows which files directly or indirectly include this file:

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.