__init__.py 553 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
from .add import scatter_add_, scatter_add
rusty1s's avatar
rusty1s committed
2
3
4
5
6
7
from .sub import scatter_sub_, scatter_sub
from .mul import scatter_mul_, scatter_mul
from .div import scatter_div_, scatter_div
from .mean import scatter_mean_, scatter_mean
from .max import scatter_max_, scatter_max
from .min import scatter_min_, scatter_min
rusty1s's avatar
rusty1s committed
8

rusty1s's avatar
rusty1s committed
9
10
__all__ = [
    'scatter_add_', 'scatter_add', 'scatter_sub_', 'scatter_sub',
rusty1s's avatar
rusty1s committed
11
    'scatter_mul_', 'scatter_mul', 'scatter_div_', 'scatter_div',
rusty1s's avatar
rusty1s committed
12
13
    'scatter_mean_', 'scatter_mean', 'scatter_max_', 'scatter_max',
    'scatter_min_', 'scatter_min'
rusty1s's avatar
rusty1s committed
14
]