tglite.op¶
|
Reindex the data of edges based on the target node indices in the TBlock. |
|
Computes segmented softmax on given data using edge information from the block. |
|
Computes segmented reduction (e.g. |
|
Aggregates the features of the source node indices in TBlock, using the specified aggregation operation ('sum' or 'mean'). |
|
Segmented operation to reduce source nodes for each destination node by a certain property, such as latest timestamp. |
|
Prefetch data (e.g. |
|
Performs pull-style multi-hop aggregation from the tail block back towards the given block by applying function to each block, using the key to pass along results. |
|
Performs push-style multi-hop propagation from the given block to the tail block by applying function to each block. |
|
Applies the deduplication optimization to the TBlock by rewriting the destination nodes. |
|
Applies the caching optimization to the TBlock by rewriting the destination nodes and using ctx as scratch space. |
|
Generates a tensor of precomputed zero values encoded by the specified encoder, used for creating a batch of zero time encodings. |
|
Encodes a tensor of time values using the provided encoder, leverage precomputed values if available in the context's time table. |