Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
TMVA::Experimental::SOFIE Namespace Reference

Namespaces

namespace  BLAS
 
namespace  INTERNAL
 
namespace  PyKeras
 
namespace  PyTorch
 
namespace  UTILITY
 

Classes

struct  Dim
 
struct  InitializedTensor
 
struct  InputTensorInfo
 
struct  RAttributes_Pool
 
class  RModel
 
class  RModelParser_ONNX
 
class  ROperator
 
class  ROperator_Add
 
class  ROperator_BatchNormalization
 
class  ROperator_Conv
 
class  ROperator_Gemm
 
class  ROperator_GRU
 Gated Recurrent Unit operator. More...
 
class  ROperator_LSTM
 Long Short-Term Memory operator. More...
 
class  ROperator_Pool
 
class  ROperator_Relu
 
class  ROperator_Reshape
 
class  ROperator_RNN
 Recurrent Neural Network operator. More...
 
class  ROperator_Selu
 
class  ROperator_Sigmoid
 
class  ROperator_Slice
 
class  ROperator_Transpose
 
struct  TensorInfo
 

Typedefs

typedef std::int64_t int_t
 

Enumerations

enum class  ETensorType {
  UNDEFINED = 0 , FLOAT = 1 , UNINT8 = 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 , COMPLEX28 = 15 ,
  BFLOAT16 = 16
}
 
enum  PoolOpMode { InvalidPool , MaxPool , AveragePool , GlobalAveragePool }
 
enum  ReshapeOpMode { Reshape , Flatten , Squeeze , Unsqueeze }
 

Functions

std::vector< DimConvertShapeToDim (std::vector< size_t > shape)
 
std::size_t ConvertShapeToLength (std::vector< size_t > shape)
 
std::string ConvertShapeToString (std::vector< size_t > shape)
 
ETensorType ConvertStringToType (std::string type)
 
std::string ConvertTypeToString (ETensorType type)
 
template<typename T >
ETensorType GetTemplatedType (T)
 

Typedef Documentation

◆ int_t

typedef std::int64_t TMVA::Experimental::SOFIE::int_t

Definition at line 26 of file SOFIE_common.hxx.

Enumeration Type Documentation

◆ ETensorType

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

Definition at line 21 of file SOFIE_common.hxx.

◆ PoolOpMode

Enumerator
InvalidPool 
MaxPool 
AveragePool 
GlobalAveragePool 

Definition at line 31 of file ROperator_Pool.hxx.

◆ ReshapeOpMode

Enumerator
Reshape 
Flatten 
Squeeze 
Unsqueeze 

Definition at line 14 of file ROperator_Reshape.hxx.

Function Documentation

◆ ConvertShapeToDim()

std::vector< Dim > TMVA::Experimental::SOFIE::ConvertShapeToDim ( std::vector< size_t >  shape)

Definition at line 9 of file SOFIE_common.cxx.

◆ ConvertShapeToLength()

std::size_t TMVA::Experimental::SOFIE::ConvertShapeToLength ( std::vector< size_t >  shape)

Definition at line 17 of file SOFIE_common.cxx.

◆ ConvertShapeToString()

std::string TMVA::Experimental::SOFIE::ConvertShapeToString ( std::vector< size_t >  shape)

Definition at line 43 of file SOFIE_common.cxx.

◆ ConvertStringToType()

ETensorType TMVA::Experimental::SOFIE::ConvertStringToType ( std::string  type)

Definition at line 34 of file SOFIE_common.cxx.

◆ ConvertTypeToString()

std::string TMVA::Experimental::SOFIE::ConvertTypeToString ( ETensorType  type)

Definition at line 23 of file SOFIE_common.cxx.

◆ GetTemplatedType()

template<typename T >
ETensorType TMVA::Experimental::SOFIE::GetTemplatedType ( )

Definition at line 88 of file SOFIE_common.hxx.