Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Public Member Functions | List of all members
dragon::onnx::ONNXAttributes Class Reference

#include <onnx_backend.h>

Public Member Functions

 ONNXAttributes (const NodeProto &node)
 
bool HasAttribute (const string &key) const
 
AttributeProto * AddRewrittenAttribute (const string &key)
 
google::protobuf::RepeatedPtrField< Argument > AttrToArg (std::function< string(const string &)> mapper) const
 
template<typename T >
get (const string &key) const
 
template<typename T >
get (const string &key, const T &default_value) const
 
const AttributeProto * remove (const string &key)
 
template<>
float get (const string &key) const
 
template<>
google::protobuf::RepeatedField< float > get (const string &key) const
 
template<>
int64_t get (const string &key) const
 
template<>
string get (const string &key) const
 
template<>
google::protobuf::RepeatedPtrField< string > get (const string &key) const
 
template<>
google::protobuf::RepeatedField< google::protobuf::int64 > get (const string &key) const
 
template<>
const TensorProto * get (const std::string &key) const
 

Constructor & Destructor Documentation

◆ ONNXAttributes()

dragon::onnx::ONNXAttributes::ONNXAttributes ( const NodeProto &  node)

Member Function Documentation

◆ AddRewrittenAttribute()

AttributeProto* dragon::onnx::ONNXAttributes::AddRewrittenAttribute ( const string &  key)
inline

◆ AttrToArg()

google::protobuf::RepeatedPtrField< Argument > dragon::onnx::ONNXAttributes::AttrToArg ( std::function< string(const string &)>  mapper) const

◆ get() [1/9]

float dragon::onnx::ONNXAttributes::get ( const string &  key) const

float <- get(key)

◆ get() [2/9]

google::protobuf::RepeatedField< float > dragon::onnx::ONNXAttributes::get ( const string &  key) const

floats <- get(key)

◆ get() [3/9]

int64_t dragon::onnx::ONNXAttributes::get ( const string &  key) const

int64 <- get(key)

◆ get() [4/9]

template<>
string dragon::onnx::ONNXAttributes::get ( const string &  key) const

string <- get(key)

◆ get() [5/9]

google::protobuf::RepeatedPtrField< string > dragon::onnx::ONNXAttributes::get ( const string &  key) const

strings <- get(key)

◆ get() [6/9]

google::protobuf::RepeatedField< google::protobuf::int64 > dragon::onnx::ONNXAttributes::get ( const string &  key) const

ints <- get(key)

◆ get() [7/9]

template<typename T >
T dragon::onnx::ONNXAttributes::get ( const string &  key) const

◆ get() [8/9]

template<typename T >
T dragon::onnx::ONNXAttributes::get ( const string &  key,
const T &  default_value 
) const
inline

◆ get() [9/9]

template<>
const TensorProto* dragon::onnx::ONNXAttributes::get ( const std::string &  key) const

◆ HasAttribute()

bool dragon::onnx::ONNXAttributes::HasAttribute ( const string &  key) const
inline

◆ remove()

const AttributeProto* dragon::onnx::ONNXAttributes::remove ( const string &  key)
inline

The documentation for this class was generated from the following files: