File tgn.h
Go to the source code of this file
#include <torch/nn/module.h>#include <torch/types.h>#include <cstddef>#include <memory>#include <optional>#include <utility>#include <vector>#include "tguf.h"
Namespaces
| Type | Name |
|---|---|
| namespace | tgn Temporal Graph Networks: A High-performance implementation. |
Classes
| Type | Name |
|---|---|
| struct | TGNConfig Configuration parameters for the TGN model architecture. |
| class | TGNImpl The core Temporal Graph Network module. Manages node memory state and temporal neighborhood aggregation. |
The documentation for this class was generated from the following file include/tgn.h