Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
onnx::TensorProto Class Reference

Definition at line 356 of file onnx.hxx.

Public Types

enum  DataLocation { DEFAULT = 0 , EXTERNAL = 1 }
 
enum  DataType {
  UNDEFINED = 0 , FLOAT = 1 , UINT8 = 2 , INT8 = 3 ,
  UINT16 = 4 , INT16 = 5 , INT32 = 6 , INT64 = 7 ,
  STRING = 8 , BOOL = 9 , FLOAT16 = 10 , DOUBLE = 11 ,
  UINT32 = 12 , UINT64 = 13 , COMPLEX64 = 14 , COMPLEX128 = 15 ,
  BFLOAT16 = 16
}
 

Public Member Functions

DataLocation data_location () const
 
int data_type () const
 
int64_t dims (int i) const
 
int dims_size () const
 
const std::vector< double > & double_data () const
 
double double_data (int i) const
 
int double_data_size () const
 
const std::vector< StringStringEntryProto > & external_data () const
 
const std::vector< float > & float_data () const
 
float float_data (int i) const
 
int float_data_size () const
 
const std::vector< int32_t > & int32_data () const
 
int32_t int32_data (int i) const
 
int int32_data_size () const
 
const std::vector< int64_t > & int64_data () const
 
int64_t int64_data (int i) const
 
int int64_data_size () const
 
const std::string & name () const
 
void ParseFrom (detail::WireReader r)
 
const std::string & raw_data () const
 

Private Attributes

DataLocation fDataLocation = DEFAULT
 
int fDataType = 0
 
std::vector< int64_t > fDims
 
std::vector< doublefDoubleData
 
std::vector< StringStringEntryProtofExternalData
 
std::vector< float > fFloatData
 
std::vector< int32_t > fInt32Data
 
std::vector< int64_t > fInt64Data
 
std::string fName
 
std::string fRawData
 

#include </github/home/ROOT-CI/src/tmva/sofie_parsers/src/onnx.hxx>

Member Enumeration Documentation

◆ DataLocation

Enumerator
DEFAULT 
EXTERNAL 

Definition at line 377 of file onnx.hxx.

◆ DataType

Enumerator
UNDEFINED 
FLOAT 
UINT8 
INT8 
UINT16 
INT16 
INT32 
INT64 
STRING 
BOOL 
FLOAT16 
DOUBLE 
UINT32 
UINT64 
COMPLEX64 
COMPLEX128 
BFLOAT16 

Definition at line 358 of file onnx.hxx.

Member Function Documentation

◆ data_location()

DataLocation onnx::TensorProto::data_location ( ) const
inline

Definition at line 387 of file onnx.hxx.

◆ data_type()

int onnx::TensorProto::data_type ( ) const
inline

Definition at line 383 of file onnx.hxx.

◆ dims()

int64_t onnx::TensorProto::dims ( int i) const
inline

Definition at line 385 of file onnx.hxx.

◆ dims_size()

int onnx::TensorProto::dims_size ( ) const
inline

Definition at line 384 of file onnx.hxx.

◆ double_data() [1/2]

const std::vector< double > & onnx::TensorProto::double_data ( ) const
inline

Definition at line 395 of file onnx.hxx.

◆ double_data() [2/2]

double onnx::TensorProto::double_data ( int i) const
inline

Definition at line 400 of file onnx.hxx.

◆ double_data_size()

int onnx::TensorProto::double_data_size ( ) const
inline

Definition at line 391 of file onnx.hxx.

◆ external_data()

const std::vector< StringStringEntryProto > & onnx::TensorProto::external_data ( ) const
inline

Definition at line 388 of file onnx.hxx.

◆ float_data() [1/2]

const std::vector< float > & onnx::TensorProto::float_data ( ) const
inline

Definition at line 394 of file onnx.hxx.

◆ float_data() [2/2]

float onnx::TensorProto::float_data ( int i) const
inline

Definition at line 399 of file onnx.hxx.

◆ float_data_size()

int onnx::TensorProto::float_data_size ( ) const
inline

Definition at line 390 of file onnx.hxx.

◆ int32_data() [1/2]

const std::vector< int32_t > & onnx::TensorProto::int32_data ( ) const
inline

Definition at line 396 of file onnx.hxx.

◆ int32_data() [2/2]

int32_t onnx::TensorProto::int32_data ( int i) const
inline

Definition at line 401 of file onnx.hxx.

◆ int32_data_size()

int onnx::TensorProto::int32_data_size ( ) const
inline

Definition at line 392 of file onnx.hxx.

◆ int64_data() [1/2]

const std::vector< int64_t > & onnx::TensorProto::int64_data ( ) const
inline

Definition at line 397 of file onnx.hxx.

◆ int64_data() [2/2]

int64_t onnx::TensorProto::int64_data ( int i) const
inline

Definition at line 402 of file onnx.hxx.

◆ int64_data_size()

int onnx::TensorProto::int64_data_size ( ) const
inline

Definition at line 393 of file onnx.hxx.

◆ name()

const std::string & onnx::TensorProto::name ( ) const
inline

Definition at line 382 of file onnx.hxx.

◆ ParseFrom()

void onnx::TensorProto::ParseFrom ( detail::WireReader r)
inline

Definition at line 404 of file onnx.hxx.

◆ raw_data()

const std::string & onnx::TensorProto::raw_data ( ) const
inline

Definition at line 386 of file onnx.hxx.

Member Data Documentation

◆ fDataLocation

DataLocation onnx::TensorProto::fDataLocation = DEFAULT
private

Definition at line 434 of file onnx.hxx.

◆ fDataType

int onnx::TensorProto::fDataType = 0
private

Definition at line 431 of file onnx.hxx.

◆ fDims

std::vector<int64_t> onnx::TensorProto::fDims
private

Definition at line 432 of file onnx.hxx.

◆ fDoubleData

std::vector<double> onnx::TensorProto::fDoubleData
private

Definition at line 437 of file onnx.hxx.

◆ fExternalData

std::vector<StringStringEntryProto> onnx::TensorProto::fExternalData
private

Definition at line 435 of file onnx.hxx.

◆ fFloatData

std::vector<float> onnx::TensorProto::fFloatData
private

Definition at line 436 of file onnx.hxx.

◆ fInt32Data

std::vector<int32_t> onnx::TensorProto::fInt32Data
private

Definition at line 438 of file onnx.hxx.

◆ fInt64Data

std::vector<int64_t> onnx::TensorProto::fInt64Data
private

Definition at line 439 of file onnx.hxx.

◆ fName

std::string onnx::TensorProto::fName
private

Definition at line 430 of file onnx.hxx.

◆ fRawData

std::string onnx::TensorProto::fRawData
private

Definition at line 433 of file onnx.hxx.

  • tmva/sofie_parsers/src/onnx.hxx