Unverified Commit c5119480 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Sparse] Remove unnecessary API exposure. (#5135)



* remove unnecessary exposure.

* remove_add_power

* remove_warning

* reorder
Co-authored-by: default avatarSteve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
parent c6228c01
......@@ -5,9 +5,6 @@ dgl.sparse
`dgl.sparse` is a library for sparse operators that are commonly used in GNN models.
.. warning::
This is an experimental package. The sparse operators provided in this library do not guarantee the same performance as their message-passing api counterparts.
Sparse matrix class
-------------------------
.. currentmodule:: dgl.sparse
......@@ -85,16 +82,15 @@ Attributes and methods
SparseMatrix.dtype
SparseMatrix.device
SparseMatrix.val
SparseMatrix.__repr__
SparseMatrix.row
SparseMatrix.col
SparseMatrix.indices
SparseMatrix.coo
SparseMatrix.csr
SparseMatrix.csc
SparseMatrix.coalesce
SparseMatrix.has_duplicate
SparseMatrix.dense
SparseMatrix.transpose
SparseMatrix.t
SparseMatrix.T
SparseMatrix.to
......@@ -104,7 +100,6 @@ Attributes and methods
SparseMatrix.double
SparseMatrix.int
SparseMatrix.long
SparseMatrix.transpose
SparseMatrix.reduce
SparseMatrix.sum
SparseMatrix.smax
......@@ -112,7 +107,6 @@ Attributes and methods
SparseMatrix.smean
SparseMatrix.neg
SparseMatrix.softmax
SparseMatrix.__matmul__
Diagonal matrix class
-------------------------
......@@ -140,12 +134,11 @@ Attributes and methods
DiagMatrix.dtype
DiagMatrix.device
DiagMatrix.val
DiagMatrix.__repr__
DiagMatrix.as_sparse
DiagMatrix.dense
DiagMatrix.transpose
DiagMatrix.t
DiagMatrix.T
DiagMatrix.transpose
DiagMatrix.to
DiagMatrix.cuda
DiagMatrix.cpu
......@@ -155,7 +148,6 @@ Attributes and methods
DiagMatrix.long
DiagMatrix.neg
DiagMatrix.inv
DiagMatrix.__matmul__
Operators
---------
......@@ -164,16 +156,6 @@ Operators
.. autosummary::
:toctree: ../../generated/
sp_add
sp_mul
sp_power
diag_add
diag_sub
diag_mul
diag_div
diag_power
add
power
spmm
bspmm
spspmm
......
......@@ -4,8 +4,6 @@ from typing import Union
from .diag_matrix import diag, DiagMatrix
from .sparse_matrix import SparseMatrix
__all__ = ["diag_add", "diag_sub", "diag_mul", "diag_div", "diag_power"]
def diag_add(
D1: DiagMatrix, D2: Union[DiagMatrix, SparseMatrix]
......
......@@ -6,8 +6,6 @@ import torch
from .diag_matrix import DiagMatrix
from .sparse_matrix import SparseMatrix, val_like
__all__ = ["sp_add", "sp_mul", "sp_power"]
def spsp_add(A, B):
"""Invoke C++ sparse library for addition"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment