nvneural::IOnnxGenerationActivation Class Referenceabstract
ONNX export interface for IActivationFunction. More...
#include <nvneural/OnnxTypes.h>
Inheritance diagram for nvneural::IOnnxGenerationActivation:
Public Member Functions | |
| virtual NeuralResult | generateActivationOnnx (IOnnxGenerationHost *pOnnxHost, const ILayer *pLayer) const noexcept=0 |
| Generates ONNX operators for this activation function. More... | |
 Public Member Functions inherited from nvneural::IRefObject | |
| virtual RefCount | addRef () const noexcept=0 |
| Increments the object's reference count. More... | |
| virtual const void * | queryInterface (TypeId interface) const noexcept=0 |
| This is an overloaded member function, provided for convenience. It differs from the above function only in what argument(s) it accepts. | |
| virtual void * | queryInterface (TypeId interface) noexcept=0 |
| Retrieves a new object interface pointer. More... | |
| virtual RefCount | release () const noexcept=0 |
| Decrements the object's reference count and destroy the object if the reference count reaches zero. More... | |
Static Public Attributes | |
| static const IRefObject::TypeId | typeID = 0x5cb7974e7d598a97ul |
| Interface TypeId for InterfaceOf purposes. | |
| Static Public Attributes inherited from nvneural::IRefObject | |
| static const TypeId | typeID = 0x14ecc3f9de638e1dul |
| Interface TypeId for InterfaceOf purposes. | |
Additional Inherited Members | |
| Public Types inherited from nvneural::IRefObject | |
| 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. | |
| Protected Member Functions inherited from nvneural::IRefObject | |
| virtual | ~IRefObject ()=default |
| A protected destructor prevents accidental stack-allocation of IRefObjects or use with other smart pointer classes like std::unique_ptr. | |
Detailed Description
ONNX export interface for IActivationFunction.
Activation functions should implement this interface to support export to ONNX. By default, activations that do not implement this interface will be replaced by opaque placeholder instances.
Member Function Documentation
◆ generateActivationOnnx()
|
pure virtualnoexcept |
Generates ONNX operators for this activation function.
- Parameters
-
pOnnxHost ONNX generation host; provides the graph interface pLayer Layer attached to this activation and containing relevant coefficients
- Returns
- NeuralResult::Success normally. Failure if something went wrong and the export should be canceled. You may return NeuralResult::Unsupported to treat this activation as though it had no IOnnxGenerationActivation interface. This is useful if the activation inherits a larger parent class but needs to opt out of ONNX generation. Do not modify the graph if you use this path.
The documentation for this class was generated from the following file:
- Core/Inc/nvneural/OnnxTypes.h
