#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"
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. | |