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< Dim > | ConvertShapeToDim (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 std::int64_t TMVA::Experimental::SOFIE::int_t |
Definition at line 26 of file SOFIE_common.hxx.
|
strong |
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.
Enumerator | |
---|---|
InvalidPool | |
MaxPool | |
AveragePool | |
GlobalAveragePool |
Definition at line 31 of file ROperator_Pool.hxx.
Enumerator | |
---|---|
Reshape | |
Flatten | |
Squeeze | |
Unsqueeze |
Definition at line 14 of file ROperator_Reshape.hxx.
std::vector< Dim > TMVA::Experimental::SOFIE::ConvertShapeToDim | ( | std::vector< size_t > | shape | ) |
Definition at line 9 of file SOFIE_common.cxx.
std::size_t TMVA::Experimental::SOFIE::ConvertShapeToLength | ( | std::vector< size_t > | shape | ) |
Definition at line 17 of file SOFIE_common.cxx.
std::string TMVA::Experimental::SOFIE::ConvertShapeToString | ( | std::vector< size_t > | shape | ) |
Definition at line 43 of file SOFIE_common.cxx.
ETensorType TMVA::Experimental::SOFIE::ConvertStringToType | ( | std::string | type | ) |
Definition at line 34 of file SOFIE_common.cxx.
std::string TMVA::Experimental::SOFIE::ConvertTypeToString | ( | ETensorType | type | ) |
Definition at line 23 of file SOFIE_common.cxx.
ETensorType TMVA::Experimental::SOFIE::GetTemplatedType | ( | T | ) |
Definition at line 88 of file SOFIE_common.hxx.