Class used to create and manipulate ONNX nodes. More...
#include "darknet_node.hpp"

Public Member Functions | |
| Node ()=delete | |
| Node (const Darknet::CfgSection §ion, const Darknet::VInt &v) | |
Constructor for many int constants. | |
| Node (const Darknet::CfgSection §ion, const float f, const size_t bit_size=32) | |
Constructor for a single float constant. | |
| Node (const Darknet::CfgSection §ion, const std::string &append="") | |
| Constructor. | |
| Node (const Node &rhs)=delete | |
| Node (const std::string &n) | |
| Constructor. | |
| ~Node () | |
| Destructor. | |
| Node & | add_attribute_FLOAT (const std::string &key, const float val) |
| Node & | add_attribute_INT (const std::string &key, const int val) |
| Node & | add_attribute_INTS (const std::string &key, const Darknet::VInt &val) |
| Node & | add_attribute_STR (const std::string &key, const std::string &val) |
| Node & | add_input (const std::string &input) |
| Node & | add_input (int idx) |
| Look up the node's name from the given index and use that as input. | |
| Node & | doc (const Darknet::CfgSection §ion) |
| Node & | doc (const std::string &str) |
| Node & | doc_append (const std::string &str) |
| Node & | init (const std::string &n) |
| Node & | operator= (const Node &rhs)=delete |
| Node & | set_output (const std::string &out="") |
| Node & | type (const std::string &type) |
Static Public Member Functions | |
| static std::string | get_output_for_layer_index (const int idx) |
Public Attributes | |
| size_t | counter |
| size_t | layer_index |
| std::string | name |
| onnx::NodeProto * | node |
| std::string | output |
Static Public Attributes | |
| static std::string | cfg_filename = "unknown" |
| static onnx::GraphProto * | graph = nullptr |
| static size_t | node_counter = 0 |
| static Darknet::MIdStr | output_per_layer_index |
| Keep track of the most recent output name for each of the layers. | |
Class used to create and manipulate ONNX nodes.
|
delete |
|
delete |
| Darknet::Node::Node | ( | const std::string & | n | ) |
Constructor.

| Darknet::Node::Node | ( | const Darknet::CfgSection & | section, |
| const std::string & | append = "" |
||
| ) |
Constructor.

| Darknet::Node::Node | ( | const Darknet::CfgSection & | section, |
| const float | f, | ||
| const size_t | bit_size = 32 |
||
| ) |
Constructor for a single float constant.

| Darknet::Node::Node | ( | const Darknet::CfgSection & | section, |
| const Darknet::VInt & | v | ||
| ) |
Constructor for many int constants.

| Darknet::Node::~Node | ( | ) |
Destructor.
| Darknet::Node & Darknet::Node::add_attribute_FLOAT | ( | const std::string & | key, |
| const float | val | ||
| ) |

| Darknet::Node & Darknet::Node::add_attribute_INT | ( | const std::string & | key, |
| const int | val | ||
| ) |


| Darknet::Node & Darknet::Node::add_attribute_INTS | ( | const std::string & | key, |
| const Darknet::VInt & | val | ||
| ) |


| Darknet::Node & Darknet::Node::add_attribute_STR | ( | const std::string & | key, |
| const std::string & | val | ||
| ) |

| Darknet::Node & Darknet::Node::add_input | ( | const std::string & | input | ) |
| Darknet::Node & Darknet::Node::add_input | ( | int | idx | ) |
Look up the node's name from the given index and use that as input.

| Darknet::Node & Darknet::Node::doc | ( | const Darknet::CfgSection & | section | ) |
| Darknet::Node & Darknet::Node::doc | ( | const std::string & | str | ) |

| Darknet::Node & Darknet::Node::doc_append | ( | const std::string & | str | ) |
|
static |

| Darknet::Node & Darknet::Node::init | ( | const std::string & | n | ) |

| Darknet::Node & Darknet::Node::set_output | ( | const std::string & | out = "" | ) |
| Darknet::Node & Darknet::Node::type | ( | const std::string & | type | ) |

|
static |
| size_t Darknet::Node::counter |
|
static |
| size_t Darknet::Node::layer_index |
| std::string Darknet::Node::name |
| onnx::NodeProto* Darknet::Node::node |
|
static |
| std::string Darknet::Node::output |
|
static |
Keep track of the most recent output name for each of the layers.