megengine.traced_module.trace_module

trace_module(mod, *args, **kwargs)

Traces module mod and returns corresponding TracedModule.

Parameters:

• mod (Module) – the module will be converted to TracedModule.

• args (Tuple[Any]) – the positional arguments passed to forward method of mod.

• kwargs (Dict[str, Any]) – the keyword arguments passed to forward method of mod.

Return type:

TracedModule