Tool-provided interface representing a NodeProto in an ONNX graph.
More...
#include <nvneural/OnnxTypes.h>
|
using | RefCount = std::uint32_t |
| Typedef used to track the number of active references to an object.
|
|
using | TypeId = std::uint64_t |
| Every interface must define a unique TypeId. This should be randomized.
|
|
virtual | ~IRefObject ()=default |
| A protected destructor prevents accidental stack-allocation of IRefObjects or use with other smart pointer classes like std::unique_ptr.
|
|
Tool-provided interface representing a NodeProto in an ONNX graph.
◆ addInput()
virtual NeuralResult nvneural::IOnnxGenerationGraphNode::addInput |
( |
const char * |
pName | ) |
|
|
pure virtualnoexcept |
◆ addOutput()
virtual NeuralResult nvneural::IOnnxGenerationGraphNode::addOutput |
( |
const char * |
pName | ) |
|
|
pure virtualnoexcept |
Adds an output reference to the node.
Names should come from the IOnnxGenerationHost::getLayerOutputName namespace. To represent a missing output, pass the empty string ("").
With few exceptions, all layers should emit a node output with the same name as that returned by IOnnxGenerationHost::getLayerOutputName(this).
- Parameters
-
◆ createAttribute()
Creates a new attribute for this node and returns it.
- Parameters
-
◆ setDocString()
virtual NeuralResult nvneural::IOnnxGenerationGraphNode::setDocString |
( |
const char * |
pDocString | ) |
|
|
pure virtualnoexcept |
Assigns a human-readable documentation string to the node.
This function is optional.
- Parameters
-
pDocString | New documentation string. Markdown is allowed. |
◆ setName()
virtual NeuralResult nvneural::IOnnxGenerationGraphNode::setName |
( |
const char * |
pName | ) |
|
|
pure virtualnoexcept |
Sets the name of the node.
This is optional in the current ONNX revision.
- Parameters
-
pName | Node name. This is typically a name derived from ILayer::name. |
◆ setOperatorSetDomain()
virtual NeuralResult nvneural::IOnnxGenerationGraphNode::setOperatorSetDomain |
( |
const char * |
pDomain | ) |
|
|
pure virtualnoexcept |
Sets the operator set domain used by this node.
This is typically unnecessary; new nodes are initialized to a domain of "" (the onnx.ai default set).
- Parameters
-
pDomain | Operator set domain to assign. |
◆ setOperatorType()
virtual NeuralResult nvneural::IOnnxGenerationGraphNode::setOperatorType |
( |
const char * |
pOpType | ) |
|
|
pure virtualnoexcept |
Sets the operator type of the node.
This is mandatory.
- Parameters
-
pOpType | Operator type string. |
The documentation for this class was generated from the following file: