slapo.tracer

Functions:

trace(model, **kwargs)

Traces a model to a GraphModule.

slapo.tracer.trace(model, **kwargs)

Traces a model to a GraphModule.

Parameters

• model (nn.Module) –

• kwargs (dict[str, Any]) –