analyzer
Analyzer to look for layer-wise dependencies to identify connected symbols.
Functions
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: