deepmd_gnn.pt#

PyTorch backend plugin registration.

Functions#

load(→ None)

Entry point placeholder; importing this module registers plugins.

_is_partially_initialized(→ bool)

_register(→ None)

Module Contents#

deepmd_gnn.pt.load() None[source]#

Entry point placeholder; importing this module registers plugins.

deepmd_gnn.pt._is_partially_initialized(module_name: str, attr_name: str) bool[source]#
deepmd_gnn.pt._register() None[source]#