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