add other backend's bspmm

BUPT-GAMMA · Jun 26, 2024 · a460dbf · a460dbf
1 parent 4c317a9
commit a460dbf
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/gammagl/mpops/mindspore.py b/gammagl/mpops/mindspore.py
@@ -60,3 +60,6 @@ def segment_max(x, segment_ids, num_segments=None):
 
 def gspmm(index, weight=None, x=None, reduce='sum'):
     pass
+
+def bspmm(index, weight=None, x=None, reduce='sum'):
+    pass
diff --git a/gammagl/mpops/paddle.py b/gammagl/mpops/paddle.py
@@ -222,3 +222,6 @@ def _scatter(x, index, updates, overwrite=True):
 
 def gspmm(index, weight=None, x=None, reduce='sum'):
     pass
+
+def bspmm(index, weight=None, x=None, reduce='sum'):
+    pass
diff --git a/gammagl/mpops/tensorflow.py b/gammagl/mpops/tensorflow.py
@@ -60,3 +60,6 @@ def segment_min(x, segment_ids, num_segments=None):
 
 def gspmm(index, weight=None, x=None, reduce='sum'):
     pass
+
+def bspmm(index, weight=None, x=None, reduce='sum'):
+    pass