项目文件夹

文件

T

文件历史

Da Zheng 223795eb82 [Distributed] Distributed heterograph training (#3069 )

* support hetero RGCN.

* fix.

* simplify code.

* sample_neighbors return heterograph directly.

* avoid using to_heterogeneous.

* compute canonical etypes in advance.

* fix tests.

* fix.

* fix distributed data loader for heterograph.

* use NodeDataLoader.

* fix bugs in partitioning on heterogeneous graphs.

* fix lint.

* fix tests.

* fix.

* fix.

* fix bugs.

* fix tests.

* fix.

* enable coo for distributed.

* fix.

* fix.

* fix.

* fix.

* fix.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-71-112.ec2.internal>
Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>

2021-11-02 09:18:15 +00:00

__init__.py

[Feature] Autograd of gspmm and gsddmm on PyTorch/MXNet/Tensorflow (#1680 )

2020-07-13 23:21:22 +08:00

sparse_optim.py

[Feature] Add support for sparse embedding (#2451 )

2021-01-28 00:26:49 +08:00

sparse.py

[Feature] Support direct creation from CSR and CSC (#3045 )

2021-06-25 10:05:47 +08:00

tensor.py

[Distributed] Distributed heterograph training (#3069 )

2021-11-02 09:18:15 +00:00