Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Classes | Typedefs | Functions | Variables
dragon::onnx Namespace Reference

Classes

class  ConversionContext
 
class  ONNXAttributes
 
class  ONNXBackend
 
struct  ONNXImporterReturns
 
struct  ONNXNode
 

Typedefs

using ValueInfoMap = Map< string, ValueInfoProto >
 
using InitializerMap = Map< string, const TensorProto * >
 

Functions

void CopyAttrValueToArg (Argument *arg, const AttributeProto &attr)
 
bool AlmostEqual (double a, double b)
 
template<class T >
bool TryConvertingTensorRawValues (const TensorProto &onnx_tensor, google::protobuf::RepeatedField< T > *field)
 
template<class Tx , class Ty >
bool TryConvertingTensorRawValues_v2 (const TensorProto &onnx_tensor, google::protobuf::RepeatedField< Ty > *field)
 
template<typename T >
void ConvertIntegralValue (const TensorProto &onnx_tensor, Argument *values)
 
template<>
void ConvertIntegralValue< google::protobuf::int64 > (const TensorProto &onnx_tensor, Argument *values)
 
template<>
void ConvertIntegralValue< google::protobuf::uint64 > (const TensorProto &onnx_tensor, Argument *values)
 

Variables

const int kKnownOpsetVersion = 9
 

Typedef Documentation

◆ InitializerMap

using dragon::onnx::InitializerMap = typedef Map<string, const TensorProto*>

◆ ValueInfoMap

using dragon::onnx::ValueInfoMap = typedef Map<string, ValueInfoProto>

Function Documentation

◆ AlmostEqual()

bool dragon::onnx::AlmostEqual ( double  a,
double  b 
)

◆ ConvertIntegralValue()

template<typename T >
void dragon::onnx::ConvertIntegralValue ( const TensorProto &  onnx_tensor,
Argument *  values 
)

◆ ConvertIntegralValue< google::protobuf::int64 >()

template<>
void dragon::onnx::ConvertIntegralValue< google::protobuf::int64 > ( const TensorProto &  onnx_tensor,
Argument *  values 
)

◆ ConvertIntegralValue< google::protobuf::uint64 >()

template<>
void dragon::onnx::ConvertIntegralValue< google::protobuf::uint64 > ( const TensorProto &  onnx_tensor,
Argument *  values 
)

◆ CopyAttrValueToArg()

void dragon::onnx::CopyAttrValueToArg ( Argument *  arg,
const AttributeProto &  attr 
)

argument <- attribute

◆ TryConvertingTensorRawValues()

template<class T >
bool dragon::onnx::TryConvertingTensorRawValues ( const TensorProto &  onnx_tensor,
google::protobuf::RepeatedField< T > *  field 
)

◆ TryConvertingTensorRawValues_v2()

template<class Tx , class Ty >
bool dragon::onnx::TryConvertingTensorRawValues_v2 ( const TensorProto &  onnx_tensor,
google::protobuf::RepeatedField< Ty > *  field 
)

Variable Documentation

◆ kKnownOpsetVersion

const int dragon::onnx::kKnownOpsetVersion = 9