trace — Model Optimizer 0.27.1 (original) (raw)
Modules
| modelopt.torch.trace.analyzer | Analyzer to look for layer-wise dependencies to identify connected symbols. |
|---|---|
| modelopt.torch.trace.modules | Module containing layer-specific symbolic information and implementations. |
| modelopt.torch.trace.plugins | Handles tracing plugins for third-party modules. |
| modelopt.torch.trace.symbols | Utilities to describe symbols found in common torch modules. |
| modelopt.torch.trace.tracer | Tracing module for building a set of partial Graphs describing the model. |
Tracing module for layer-wise dependency analysis.