nncf icon indicating copy to clipboard operation
nncf copied to clipboard

[PT] Introduce GraphBuilderMode

Open AlexanderDokuchaev opened this issue 4 months ago • 0 comments

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

AlexanderDokuchaev avatar Oct 21 '24 20:10 AlexanderDokuchaev