MCPcopy
hub / github.com/snap-stanford/GraphGym / register_layer

Function register_layer

graphgym/register.py:74–76  ·  view source on GitHub ↗

r"""Registers a GNN layer in GraphGym.

(key: str, module: Any = None)

Source from the content-addressed store, hash-verified

72
73
74def register_layer(key: str, module: Any = None):
75 r"""Registers a GNN layer in GraphGym."""
76 return register_base(layer_dict, key, module)
77
78
79def register_pooling(key: str, module: Any = None):

Callers 5

idconv.pyFile · 0.90
example.pyFile · 0.90
sageinitconv.pyFile · 0.90
generalconv_ogb.pyFile · 0.90
attconv.pyFile · 0.90

Calls 1

register_baseFunction · 0.85

Tested by

no test coverage detected