`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=_DefaultTracer())[source]#

Initialization

register(key: str | None, fn: elasticai.creator.torch2ir.torch2ir.TypeHandler) → elasticai.creator.torch2ir.torch2ir.TypeHandler[source]#

convert(model: torch.nn.Module) → tuple[elasticai.creator.torch2ir.torch2ir.DataGraph, elasticai.creator.torch2ir.torch2ir.Registry][source]#

__call__(model: torch.nn.Module) → tuple[elasticai.creator.torch2ir.torch2ir.DataGraph, elasticai.creator.torch2ir.torch2ir.Registry][source]#

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