cugraph-ops Python API reference

This page provides a list of all publicly accessible modules, methods and classes through pylibcugraphops.* namespace.

API Documentation

• Graph types
  • CSC Graph
  • Heterogenous CSC Graph
  • Bipartite Graph
  • Heterogenous Bipartite Graph
• Operators for Message-Passing
  • Simple Neighborhood Aggregator (SAGEConv)
  • Weighted Neighborhood Aggregation
  • Heterogenous Aggregator using Basis Decomposition (RGCNConv)
  • Graph Attention (GATConv/GATv2Conv)
  • Transformer-like Graph Attention (TransformerConv)
  • Directional Message-Passing (DMPNN)
  • Update Edges: Concatenation or Sum of Edge and Node Features
• Dimenet operators
  • Radial Basis Functions
  • Edge-to-Edge Aggregation
• PyTorch Autograd Wrappers
  • Simple Neighborhood Aggregator (SAGEConv)
  • Graph Attention (GATConv/GATv2Conv)
  • Heterogenous Aggregator using Basis Decomposition (RGCNConv)
  • Update Edges: Concatenation or Sum of Edge and Node Features