Visualization Stage#

Visualization emits graph artifacts for debugging and inspection. This stage is intended to help developers understand transform output without changing inference semantics.

Visualize Namespace#

Transform key: visualize_namespace

Source module: tensorrt_llm._torch.auto_deploy.transform.library.visualization

Configured modes: graph

class tensorrt_llm._torch.auto_deploy.transform.library.visualization.VisualizeNamespace( config: TransformConfig, )[source]#

Bases: BaseTransform

Transform to visualize the graph using Model Explorer.

This transform exports the graph module to an ExportedProgram and launches Model Explorer for interactive visualization. The visualization helps debug and understand the graph structure after AutoDeploy transformations.

YAML configuration

Uses the common TransformConfig fields documented in Core Transform APIs.