dmlc--dgl
3efb5d8ecf
* Add HeteroGraphConv * add custom aggregator; some docstring * debugging * rm print * fix some acc bugs * fix initialization problem in weight basis * passed tests * lint * fix graphconv flag; add error message * add mxnet heteroconv * more fix for mx * lint * fix torch cuda test * fix mx test_nn * add exhaust test for graphconv * add tf heteroconv * fix comment
7 行
156 B
Python
7 行
156 B
Python
"""Package for Tensorflow-specific NN modules."""
|
|
from .conv import *
|
|
from .softmax import *
|
|
from .utils import *
|
|
from .glob import *
|
|
from .hetero import *
|