tests/compute/test_apply_edges_hetero.py · 88f5a8bec8f76593f6b7729817e3e10781a8a5f6 · OpenDAS / dgl

"test/test_max_min.py" did not exist on "e4d17fe89c4758f21b450965d15ec08ab4698fac"

[Feature] Add heterogeneous graph API for `edge_softmax` (#3571) · 88f5a8be

Israt Nisa authored Dec 19, 2021



* edge_softmax_hetero forwar+cpu+norm=dst

* convert eids to list

* addedunittest

* added unittest

* added backward. Not tested correctness

* minor

* changed reducer to max from sum

* bugfix

* docstring

* add GPU unittest

* output converted to dict from tuple

* lint check
Co-authored-by: Israt Nisa <nisisrat@amazon.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>

88f5a8be

test_apply_edges_hetero.py 5.4 KB

Replace test_apply_edges_hetero.py