blob: fe6f3a4603169f622e953a793f606f7fa74a9cbb [file] [log] [blame]
# torch.ao is a package with a lot of interdependencies.
# We will use lazy import to avoid cyclic dependencies here.
__all__ = [
"nn",
"ns",
"quantization",
"pruning",
]
def __getattr__(name):
if name in __all__:
import importlib
return importlib.import_module("." + name, __name__)
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")