deepmd_gnn.pt_expt

PyTorch exportable 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_expt.load() → None

Entry point placeholder; importing this module registers plugins.

deepmd_gnn.pt_expt._is_partially_initialized(module_name: str, attr_name: str) → bool

deepmd_gnn.pt_expt._register() → None