项目文件夹

文件

T

文件历史

Minjie Wang 22167f7203 [Refactor] Enable new kernel in all message passing APIs (#1953 )

* WIP: frame refactor

* new frame

* simple update_all builtin

* move all subgraph routines into the same file

* sddmm & spmm schedule; node & edge udf

* degree bucketing

* some tricky 0deg corner cases

* bug in frame append

* merge test_hetero_basics and test_basics

* some code rearange

* fix test_heterograph

* add mean spmm

* enable all builtin combinations

* pass gpu test

* pass pytorch tests

* wip

* fix some pt debugging codes

* fix bug in mxnet backward

* pass all mxnet utests

* passed tf tests

* docstring

* lint

* lint

* fix broadcasting bugs

* add warning and clamp for mean reducer

* add test for zero-degree mean

* address comments

* lint

* small fix

2020-08-07 15:40:25 +08:00

__init__.py

[Lint] Pylint (#330 )

2019-01-01 15:15:41 -05:00

base.py

[Refactor] Enable new kernel in all message passing APIs (#1953 )

2020-08-07 15:40:25 +08:00

message.py

[Refactor] Enable new kernel in all message passing APIs (#1953 )

2020-08-07 15:40:25 +08:00

reducer.py

[Refactor] Enable new kernel in all message passing APIs (#1953 )

2020-08-07 15:40:25 +08:00