Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions devtools/visualization/TARGETS
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,15 @@ runtime.python_library(
srcs = [
"__init__.py",
"visualization_utils.py",
"html_visualization.py",
"trace_passes.py",
],
visibility = ["PUBLIC"],
deps = [
"//caffe2:torch",
"//executorch/exir:lib",
"//executorch/exir/_serialize:lib",
"//executorch/devtools/etrecord:lib",
],
)

Expand Down
26 changes: 20 additions & 6 deletions devtools/visualization/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,25 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# model_explorer-based visualization (requires model_explorer pip package)
try:
from executorch.devtools.visualization.visualization_utils import ( # noqa: F401
ModelExplorerServer,
SingletonModelExplorerServer,
visualize,
visualize_graph,
visualize_model_explorer,
)
except ImportError:
pass

from executorch.devtools.visualization.visualization_utils import ( # noqa: F401
ModelExplorerServer,
SingletonModelExplorerServer,
visualize,
visualize_graph,
visualize_model_explorer,
# Self-contained HTML visualization (no external dependencies)
from executorch.devtools.visualization.html_visualization import ( # noqa: F401
extract_from_exported_program,
extract_from_etrecord,
extract_from_pt2,
extract_from_pte,
generate_html,
generate_multi_pass_html,
visualize_edge_manager,
)
Loading
Loading