nncf
nncf copied to clipboard
[PT] Introduce GraphBuilderMode
Changes
Add GraphBuilderMode to build graph of pytorch model based on FunctionHookMode
Reason for changes
Related tickets
154914
Tests
tests/torch2/function_hook/graph/test_build_graph_mode.py tests/torch2/function_hook/graph/test_graph_visualisation.py