elasticai.creator.torch2ir.torch2ir#

Module Contents#

Classes#

Functions#

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]#