`elasticai.creator.torch2ir.torch2ir`#

Module Contents#

Classes#

Functions#

get_default_converter

API#

exception elasticai.creator.torch2ir.torch2ir.LoweringError(message: str)[source]#: Bases: Exception

class elasticai.creator.torch2ir.torch2ir.Torch2Ir(tracer: torch.fx.Tracer = _DefaultTracer())[source]#

Initialization

register(module_type: str, handler: collections.abc.Callable[[torch.nn.Module], dict]) → collections.abc.Callable[[torch.nn.Module], dict][source]#: The handlers are used to extract the attributes of the module

register_handlers(handlers: collections.abc.Iterable[collections.abc.Callable[[torch.nn.Module], dict]]) → elasticai.creator.torch2ir.torch2ir.Torch2Ir[source]#

convert(model: torch.nn.Module) → collections.abc.Iterator[elasticai.creator.torch2ir.core.Implementation][source]#

__call__(model: torch.nn.Module) → collections.abc.Iterator[elasticai.creator.torch2ir.core.Implementation][source]#

elasticai.creator.torch2ir.torch2ir.get_default_converter() → elasticai.creator.torch2ir.torch2ir.Torch2Ir[source]#