Abstract Interfaces

Abstract InterfaceLocationDescription
apply_actv athena__misc_types

Interface for activation function

apply_attributes_actv athena__misc_types

Interface for loading ONNX attributes

build_onnx_expanded_gnn_layer athena__container_layer

Build one expanded-ONNX GNN layer from a node cluster.

build_onnx_expanded_nop_layer athena__container_layer

Build one expanded-ONNX NOP layer from a node cluster.

classify_onnx_expanded_gnn_layer athena__container_layer

Return true when this creator handles the given expanded-ONNX GNN prefix.

classify_onnx_expanded_nop_layer athena__container_layer

Return true when this creator handles the given expanded-ONNX NOP prefix.

compute_accuracy_function athena__accuracy

Compute the accuracy of a model

create_gnn_from_onnx_layer athena__container_layer

Create a GNN layer from ONNX metadata and initialisers

export_attributes_actv athena__misc_types

Interface for exporting ONNX attributes

generate_mask athena__base_layer

Generate dropout mask

initialiser_subroutine athena__misc_types

Interface for initialiser function

regularise athena__regulariser

Regularise the parameters

reset_actv athena__misc_types

Interface for resetting activation function attributes and variables