Module gnn 

Graph Neural Network kernels.

This module provides GPU-accelerated GNN algorithms:

• GNNInference - Message passing neural network inference
• GraphAttention - Graph attention network (GAT) layers

Structs

GATResult

Result of graph attention.

GATWeights

Attention weights for GAT layer.

GNNConfig

Configuration for GNN inference.

GNNInference

GNN Inference kernel.

GNNResult

Result of GNN inference.

GNNWeights

GNN layer weights (simulated).

GraphAttention

Graph Attention kernel.

GraphAttentionConfig

Configuration for graph attention.

Enums

ActivationType

Activation function type.

AggregationType

Message aggregation type.