mirrors / dmlc--dgl

项目文件夹

文件

T

Xin Yao 5fcd7f290f [Feature] Enable UVA for GPU PinSAGE and RandomWalk (#3857 )

* enable uva for pinsage sampler

* unit test

* modify some checks on the python side

* remove legacy random walk code

* update unit test

* update unit test

* fix unit test

* adjust checks

* move some checks to c++

* move max_nodes check to cuda kernel

* fix ci for tf

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>

2022-04-11 20:01:23 +08:00

4 行

157 B

Python

原始文件 永久链接 Blame 文件历史

	`from .sampler import NeighborSampler, LayerSampler, EdgeSampler`
	`from .dis_sampler import SamplerSender, SamplerReceiver`
	`from .dis_sampler import SamplerPool`