Definition at line 12 of file RModel.hxx.
Public Member Functions | |
RModel ()=default | |
Default constructor. | |
RModel (const RModel &other)=delete | |
RModel (RModel &&other) | |
RModel (std::string function_name) | |
RModel (std::string name, std::string parsedtime) | |
~RModel ()=default | |
void | AddDynamicTensor (std::string tensor_name, ETensorType type, std::vector< Dim > shape) |
void | AddInitializedTensor (std::string tensor_name, ETensorType type, std::vector< std::size_t > shape, std::shared_ptr< void > data) |
template<typename T > | |
void | AddInitializedTensor (std::string tensor_name, ETensorType type, std::vector< std::size_t > shape, T *raw_data) |
void | AddInputTensorInfo (std::string input_name, ETensorType type, std::vector< Dim > shape) |
void | AddInputTensorInfo (std::string input_name, ETensorType type, std::vector< size_t > shape) |
void | AddInputTensorName (std::string name) |
void | AddIntermediateTensor (std::string tensor_name, ETensorType type, std::vector< Dim > dim_shape) |
void | AddIntermediateTensor (std::string tensor_name, ETensorType type, std::vector< std::size_t > shape) |
void | AddOperator (std::unique_ptr< ROperator > op, int order_execution=-1) |
void | AddOperatorReference (ROperator *op, int order_execution=-1) |
void | AddOutputTensorNameList (std::vector< std::string > output_tensor_names) |
bool | CheckIfTensorAlreadyExist (std::string tensor_name) |
ClassDefNV (RModel, 2) | |
void | Generate (Options options=Options::kDefault, int batchSize=-1, int pos=0) |
void | Generate (std::underlying_type_t< Options > options, int batchSize=-1, long pos=0) |
void | GenerateDynamicTensorInfo () |
void | GenerateInitializedTensorInfo () |
void | GenerateIntermediateTensorInfo () |
void | GenerateOutput () |
std::vector< Dim > | GetDynamicTensorShape (std::string name) |
std::shared_ptr< void > | GetInitializedTensorData (std::string tensor_name) |
const std::vector< std::string > & | GetInputTensorNames () const |
std::vector< std::string > | GetOutputTensorNames () |
const std::vector< std::string > & | GetOutputTensorNames () const |
const std::vector< size_t > & | GetTensorShape (std::string name) |
const ETensorType & | GetTensorType (std::string name) |
void | HeadInitializedTensors (std::string name, int n_print=50) |
void | Initialize (int batchSize=-1, bool verbose=false) |
bool | IsDynamicTensor (const std::string &name) const |
bool | IsInitializedTensor (const std::string &name) const |
bool | IsInputTensor (const std::string &name) const |
RModel & | operator= (const RModel &other)=delete |
RModel & | operator= (RModel &&other) |
void | OutputGenerated (std::string filename="", bool append=false) |
void | PrintDynamicTensors () |
void | PrintInitializedTensors () |
void | PrintIntermediateTensors () |
void | PrintOutputTensors () |
void | PrintRequiredInputTensors () |
void | ReadInitializedTensorsFromFile (long) |
void | SetFilename (std::string filename) |
void | UpdateInitializedTensor (std::string tensor_name, ETensorType type, std::vector< std::size_t > shape, std::shared_ptr< void > data) |
void | UpdateOutputTensorList (std::vector< std::string > curr_output_tensor, std::vector< std::string > modify_output_tensor) |
bool | UseSession () const |
long | WriteInitializedTensorsToFile (std::string filename="") |
Public Member Functions inherited from TMVA::Experimental::SOFIE::RModel_Base | |
RModel_Base ()=default | |
Default constructor. | |
RModel_Base (std::string function_name) | |
RModel_Base (std::string name, std::string parsedtime) | |
void | AddBlasRoutines (std::vector< std::string > routines) |
void | AddNeededCustomHeader (std::string filename) |
void | AddNeededStdLib (std::string libname) |
void | GenerateHeaderInfo (std::string &hgname) |
std::string | GetFilename () |
void | OutputGenerated (std::string filename="", bool append=false) |
void | PrintGenerated () |
std::string | ReturnGenerated () |
void | SetFilename (std::string filename) |
Private Attributes | |
std::unordered_map< std::string, DynamicTensorInfo > | fDynamicTensorInfos |
std::unordered_map< std::string, InitializedTensor > | fInitializedTensors |
std::unordered_map< std::string, InputTensorInfo > | fInputTensorInfos |
std::vector< std::string > | fInputTensorNames |
std::unordered_map< std::string, TensorInfo > | fIntermediateTensorInfos |
std::vector< std::unique_ptr< ROperator > > | fOperators |
std::vector< std::string > | fOutputTensorNames |
std::unordered_map< std::string, TensorInfo > | fReadyInputTensorInfos |
std::unordered_map< std::string, std::string > | fShapeParams |
const std::string | SP = " " |
Additional Inherited Members | |
Protected Attributes inherited from TMVA::Experimental::SOFIE::RModel_Base | |
const std::unordered_set< std::string > | fAllowedStdLib = {"vector", "algorithm", "cmath"} |
std::unordered_set< std::string > | fCustomOpHeaders |
std::string | fFileName |
std::string | fGC |
bool | fIsGNN = false |
bool | fIsGNNComponent = false |
std::string | fName = "UnnamedModel" |
std::unordered_set< std::string > | fNeededBlasRoutines |
std::unordered_set< std::string > | fNeededStdLib = {"vector"} |
std::string | fParseTime |
bool | fUseSession = true |
bool | fUseWeightFile = true |
WeightFileType | fWeightFile = WeightFileType::Text |
#include <TMVA/RModel.hxx>
TMVA::Experimental::SOFIE::RModel::RModel | ( | RModel && | other | ) |
Definition at line 23 of file RModel.cxx.
|
delete |
|
default |
|
default |
Default constructor.
Needed to allow serialization of ROOT objects. See https://root.cern/manual/io_custom_classes/#restrictions-on-types-root-io-can-handle
|
inline |
Definition at line 43 of file RModel.hxx.
|
inline |
Definition at line 46 of file RModel.hxx.
void TMVA::Experimental::SOFIE::RModel::AddDynamicTensor | ( | std::string | tensor_name, |
ETensorType | type, | ||
std::vector< Dim > | shape | ||
) |
Definition at line 204 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::AddInitializedTensor | ( | std::string | tensor_name, |
ETensorType | type, | ||
std::vector< std::size_t > | shape, | ||
std::shared_ptr< void > | data | ||
) |
Definition at line 161 of file RModel.cxx.
|
inline |
Definition at line 65 of file RModel.hxx.
void TMVA::Experimental::SOFIE::RModel::AddInputTensorInfo | ( | std::string | input_name, |
ETensorType | type, | ||
std::vector< Dim > | shape | ||
) |
Definition at line 125 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::AddInputTensorInfo | ( | std::string | input_name, |
ETensorType | type, | ||
std::vector< size_t > | shape | ||
) |
Definition at line 135 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::AddInputTensorName | ( | std::string | name | ) |
Definition at line 144 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::AddIntermediateTensor | ( | std::string | tensor_name, |
ETensorType | type, | ||
std::vector< Dim > | dim_shape | ||
) |
Definition at line 187 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::AddIntermediateTensor | ( | std::string | tensor_name, |
ETensorType | type, | ||
std::vector< std::size_t > | shape | ||
) |
Definition at line 195 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::AddOperator | ( | std::unique_ptr< ROperator > | op, |
int | order_execution = -1 |
||
) |
Definition at line 148 of file RModel.cxx.
|
inline |
Definition at line 56 of file RModel.hxx.
void TMVA::Experimental::SOFIE::RModel::AddOutputTensorNameList | ( | std::vector< std::string > | output_tensor_names | ) |
Definition at line 225 of file RModel.cxx.
bool TMVA::Experimental::SOFIE::RModel::CheckIfTensorAlreadyExist | ( | std::string | tensor_name | ) |
Definition at line 116 of file RModel.cxx.
TMVA::Experimental::SOFIE::RModel::ClassDefNV | ( | RModel | , |
2 | |||
) |
|
inline |
Definition at line 101 of file RModel.hxx.
void TMVA::Experimental::SOFIE::RModel::Generate | ( | std::underlying_type_t< Options > | options, |
int | batchSize = -1 , |
||
long | pos = 0 |
||
) |
Definition at line 542 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::GenerateDynamicTensorInfo | ( | ) |
Definition at line 405 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::GenerateInitializedTensorInfo | ( | ) |
Definition at line 337 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::GenerateIntermediateTensorInfo | ( | ) |
Definition at line 364 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::GenerateOutput | ( | ) |
Definition at line 418 of file RModel.cxx.
std::vector< Dim > TMVA::Experimental::SOFIE::RModel::GetDynamicTensorShape | ( | std::string | name | ) |
Definition at line 79 of file RModel.cxx.
std::shared_ptr< void > TMVA::Experimental::SOFIE::RModel::GetInitializedTensorData | ( | std::string | tensor_name | ) |
Definition at line 248 of file RModel.cxx.
|
inline |
Definition at line 106 of file RModel.hxx.
|
inline |
Definition at line 115 of file RModel.hxx.
|
inline |
Definition at line 107 of file RModel.hxx.
const std::vector< size_t > & TMVA::Experimental::SOFIE::RModel::GetTensorShape | ( | std::string | name | ) |
Definition at line 56 of file RModel.cxx.
const ETensorType & TMVA::Experimental::SOFIE::RModel::GetTensorType | ( | std::string | name | ) |
Definition at line 91 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::HeadInitializedTensors | ( | std::string | name, |
int | n_print = 50 |
||
) |
Definition at line 897 of file RModel.cxx.
Definition at line 257 of file RModel.cxx.
bool TMVA::Experimental::SOFIE::RModel::IsDynamicTensor | ( | const std::string & | name | ) | const |
Definition at line 177 of file RModel.cxx.
bool TMVA::Experimental::SOFIE::RModel::IsInitializedTensor | ( | const std::string & | name | ) | const |
Definition at line 172 of file RModel.cxx.
bool TMVA::Experimental::SOFIE::RModel::IsInputTensor | ( | const std::string & | name | ) | const |
Definition at line 181 of file RModel.cxx.
Definition at line 39 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::OutputGenerated | ( | std::string | filename = "" , |
bool | append = false |
||
) |
Definition at line 933 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::PrintDynamicTensors | ( | ) |
Definition at line 870 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::PrintInitializedTensors | ( | ) |
Definition at line 840 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::PrintIntermediateTensors | ( | ) |
Definition at line 855 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::PrintOutputTensors | ( | ) |
Definition at line 885 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::PrintRequiredInputTensors | ( | ) |
Definition at line 810 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::ReadInitializedTensorsFromFile | ( | long | pos | ) |
Definition at line 641 of file RModel.cxx.
|
inline |
Definition at line 116 of file RModel.hxx.
void TMVA::Experimental::SOFIE::RModel::UpdateInitializedTensor | ( | std::string | tensor_name, |
ETensorType | type, | ||
std::vector< std::size_t > | shape, | ||
std::shared_ptr< void > | data | ||
) |
Definition at line 239 of file RModel.cxx.
void TMVA::Experimental::SOFIE::RModel::UpdateOutputTensorList | ( | std::vector< std::string > | curr_output_tensor, |
std::vector< std::string > | modify_output_tensor | ||
) |
Definition at line 232 of file RModel.cxx.
|
inline |
Definition at line 136 of file RModel.hxx.
long TMVA::Experimental::SOFIE::RModel::WriteInitializedTensorsToFile | ( | std::string | filename = "" | ) |
Definition at line 716 of file RModel.cxx.
|
private |
Definition at line 20 of file RModel.hxx.
|
private |
Definition at line 18 of file RModel.hxx.
|
private |
Definition at line 16 of file RModel.hxx.
|
private |
Definition at line 24 of file RModel.hxx.
|
private |
Definition at line 19 of file RModel.hxx.
|
private |
Definition at line 26 of file RModel.hxx.
|
private |
Definition at line 23 of file RModel.hxx.
|
private |
Definition at line 17 of file RModel.hxx.
|
private |
Definition at line 22 of file RModel.hxx.
|
private |
Definition at line 28 of file RModel.hxx.