Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::onnx::OnnxAttributes Class Reference

Public Member Functions

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

Detailed Description

Definition at line 55 of file backend.h.


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