emit_onnx_nodes_base Interface

interface
private module subroutine emit_onnx_nodes_base(this, prefix, nodes, num_nodes, max_nodes, inits, num_inits, max_inits, input_name, is_last_layer, format)

Arguments

Type IntentOptional Attributes Name
class(base_layer_type), intent(in) :: this

Instance of the layer
character(len=*), intent(in) :: prefix

Node name prefix (e.g. "node_2")
type(onnx_node_type), intent(inout), dimension(:) :: nodes

Accumulator for ONNX nodes
integer, intent(inout) :: num_nodes

Current number of nodes
integer, intent(in) :: max_nodes

Maximum capacity
type(onnx_initialiser_type), intent(inout), dimension(:) :: inits

Accumulator for ONNX initialisers
integer, intent(inout) :: num_inits

Current number of initialisers
integer, intent(in) :: max_inits

Maximum capacity
character(len=*), intent(in), optional :: input_name

Upstream tensor name used by sequential expanded ONNX format
logical, intent(in), optional :: is_last_layer

Whether this is the last non-input layer in the network
integer, intent(in), optional :: format

Export format selector 1 = ONNX athena abstract format (default) 2 = ONNX expanded format

Description

Emit ONNX JSON nodes for this layer Default implementation does nothing; override for GNN/NOP layers