Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
SafetensorsReader Class Reference

Definition at line 947 of file SOFIE_common_helpers.cxx.

Public Member Functions

 SafetensorsReader (SafetensorsBlob blob)
 
template<class T >
void Read (const std::string &name, std::vector< T > &target, std::size_t expectedLength, const std::string &expectedDtype)
 

Private Attributes

const char * fData
 
std::size_t fSize
 
std::unordered_map< std::string, SafetensorsTensorInfo > fTensors
 

Constructor & Destructor Documentation

◆ SafetensorsReader()

SafetensorsReader::SafetensorsReader ( SafetensorsBlob blob)
inlineexplicit

Definition at line 949 of file SOFIE_common_helpers.cxx.

Member Function Documentation

◆ Read()

template<class T >
void SafetensorsReader::Read ( const std::string & name,
std::vector< T > & target,
std::size_t expectedLength,
const std::string & expectedDtype )
inline

Definition at line 973 of file SOFIE_common_helpers.cxx.

Member Data Documentation

◆ fData

const char* SafetensorsReader::fData
private

Definition at line 993 of file SOFIE_common_helpers.cxx.

◆ fSize

std::size_t SafetensorsReader::fSize
private

Definition at line 994 of file SOFIE_common_helpers.cxx.

◆ fTensors

std::unordered_map<std::string, SafetensorsTensorInfo> SafetensorsReader::fTensors
private

Definition at line 995 of file SOFIE_common_helpers.cxx.

  • tmva/sofie/src/SOFIE_common_helpers.cxx