analyzer

Analyzer to look for layer-wise dependencies to identify connected symbols.

Functions

analyze_symbols

Trace layer dependencies and return a valid symbolic mapping.

analyze_symbols(model, concrete_args=None, _strict=True)

Trace layer dependencies and return a valid symbolic mapping.

The tracing process happens in-place.

Parameters:
  • model (Module) – Model to be traced.

  • concrete_args (Dict[str, Any] | None) – [FOR DEBUG ONLY!] Concrete arguments to model.forward() passed to the symbolic tracer.

  • _strict (bool) – [FOR DEBUG ONLY!] Whether to enforce strict dependency tracing.

Return type:

SymMap