Lines Matching refs:num_hops

29 def check_uniform(out, num_hops, max_num_vertices):  argument
41         assert(data <= num_hops)
43 def check_non_uniform(out, num_hops, max_num_vertices):  argument
58         assert(data <= num_hops)
80 …out = mx.nd.contrib.dgl_csr_neighbor_uniform_sample(a, seed, num_args=2, num_hops=1, num_neighbor=…
82     check_uniform(out, num_hops=1, max_num_vertices=5)
89 …out = mx.nd.contrib.dgl_csr_neighbor_uniform_sample(a, seed, num_args=2, num_hops=1, num_neighbor=…
91     check_uniform(out, num_hops=1, max_num_vertices=4)
98 …out = mx.nd.contrib.dgl_csr_neighbor_uniform_sample(a, seed, num_args=2, num_hops=2, num_neighbor=…
100     check_uniform(out, num_hops=2, max_num_vertices=3)
107 …out = mx.nd.contrib.dgl_csr_neighbor_uniform_sample(a, seed, num_args=2, num_hops=1, num_neighbor=…
109     check_uniform(out, num_hops=1, max_num_vertices=5)
116 …out = mx.nd.contrib.dgl_csr_neighbor_uniform_sample(a, seed, num_args=2, num_hops=1, num_neighbor=…
118     check_uniform(out, num_hops=1, max_num_vertices=5)
125 …out = mx.nd.contrib.dgl_csr_neighbor_uniform_sample(a, seed, num_args=2, num_hops=2, num_neighbor=…
127     check_uniform(out, num_hops=2, max_num_vertices=5)
134 …out = mx.nd.contrib.dgl_csr_neighbor_uniform_sample(a, seed, num_args=2, num_hops=1, num_neighbor=…
136     check_uniform(out, num_hops=1, max_num_vertices=5)
151 …out = mx.nd.contrib.dgl_csr_neighbor_non_uniform_sample(a, prob, seed, num_args=3, num_hops=1, num…
153     check_non_uniform(out, num_hops=1, max_num_vertices=5)
156 …out = mx.nd.contrib.dgl_csr_neighbor_non_uniform_sample(a, prob, seed, num_args=3, num_hops=1, num…
158     check_non_uniform(out, num_hops=1, max_num_vertices=4)
161 …out = mx.nd.contrib.dgl_csr_neighbor_non_uniform_sample(a, prob, seed, num_args=3, num_hops=2, num…
163     check_non_uniform(out, num_hops=2, max_num_vertices=4)
166 …out = mx.nd.contrib.dgl_csr_neighbor_non_uniform_sample(a, prob, seed, num_args=3, num_hops=1, num…
168     check_non_uniform(out, num_hops=1, max_num_vertices=5)
171 …out = mx.nd.contrib.dgl_csr_neighbor_non_uniform_sample(a, prob, seed, num_args=3, num_hops=1, num…
173     check_non_uniform(out, num_hops=1, max_num_vertices=5)
176 …out = mx.nd.contrib.dgl_csr_neighbor_non_uniform_sample(a, prob, seed, num_args=3, num_hops=2, num…
178     check_non_uniform(out, num_hops=2, max_num_vertices=5)
181 …out = mx.nd.contrib.dgl_csr_neighbor_non_uniform_sample(a, prob, seed, num_args=3, num_hops=1, num…
183     check_non_uniform(out, num_hops=1, max_num_vertices=5)