Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Torch] Bump PyTorch Version to 2.6.0 #3253

Merged
merged 48 commits into from
Feb 24, 2025
Merged
Changes from 1 commit
Commits
Show all changes
48 commits
Select commit Hold shift + click to select a range
3c8933e
init
anzr299 Feb 4, 2025
56c2766
fix weights only issue
anzr299 Feb 4, 2025
d4e3942
update graphs
anzr299 Feb 4, 2025
816016e
update graphs
anzr299 Feb 4, 2025
bfa8f62
update model loading instances
anzr299 Feb 4, 2025
f6c338e
update transformers requirement
anzr299 Feb 5, 2025
2d8b9b8
Merge branch 'openvinotoolkit:develop' into an/torch260
anzr299 Feb 5, 2025
1b31e79
load_weight and torch 2 fixes
anzr299 Feb 5, 2025
b14e4aa
big fix: use self attention instead of SDPA while loading hf models
anzr299 Feb 6, 2025
e0fb153
part of big fix
anzr299 Feb 6, 2025
3fbea82
missed out model hf fix
anzr299 Feb 6, 2025
00255ea
Merge branch 'develop' into an/torch260
anzr299 Feb 6, 2025
719b4ba
pre-commit fix
anzr299 Feb 6, 2025
27b93fe
Merge branch 'an/torch260' of https://github.com/anzr299/nncf into an…
anzr299 Feb 6, 2025
221dc0f
fix comments
anzr299 Feb 6, 2025
8e742de
update reference graphs
anzr299 Feb 6, 2025
a230b47
change torch.export for swin model to torch.export.export_for_inference
anzr299 Feb 6, 2025
d230ec5
update optimum intel version
anzr299 Feb 6, 2025
ee50c4f
Merge branch 'develop' into an/torch260
anzr299 Feb 11, 2025
527d301
revert old pt fix
anzr299 Feb 12, 2025
e99e4c2
Merge branch 'openvinotoolkit:develop' into an/torch260
anzr299 Feb 13, 2025
49d6251
change export in conformance test
anzr299 Feb 13, 2025
960e12f
xfail for FX backend conformance test
anzr299 Feb 14, 2025
c72a810
xfail cuda torch fx tests
anzr299 Feb 17, 2025
ac4ceb3
fix model transformer issue
anzr299 Feb 18, 2025
c277ad2
Merge branch 'develop' into an/torch260
anzr299 Feb 18, 2025
2c592a3
delete subgraphs with output_1 in name since it is no longer in use
anzr299 Feb 19, 2025
d2a4811
Merge branch 'an/torch260' of https://github.com/anzr299/nncf into an…
anzr299 Feb 19, 2025
f9d8c56
modify issue references for appropriation
anzr299 Feb 19, 2025
59e1b00
update requirements for conformance test to include accelerate
anzr299 Feb 19, 2025
d036282
add proper xfail for FX backend
anzr299 Feb 19, 2025
5e3476d
fix
anzr299 Feb 19, 2025
844ff75
do not export for OV backend in torhcvision pipeline
anzr299 Feb 19, 2025
a4f1a0f
pre commit fix
anzr299 Feb 19, 2025
1873d94
update xfail condition
anzr299 Feb 19, 2025
a92fad0
remove debug code
AlexanderDokuchaev Feb 19, 2025
e4dfda6
xfail error message fix
anzr299 Feb 20, 2025
55fcb1c
update transformers to 4.48.0
anzr299 Feb 20, 2025
5cb4123
Merge branch 'an/torch260' of https://github.com/anzr299/nncf into an…
anzr299 Feb 20, 2025
ae1466b
fix openvino weights compression test
anzr299 Feb 20, 2025
e1702e7
update reference graphs for PT
anzr299 Feb 20, 2025
7651ce7
fix pre-commit test
anzr299 Feb 20, 2025
748871c
Merge branch 'develop' into an/torch260
anzr299 Feb 21, 2025
fd58603
change attention implementation from old transformers to new torch SDPA
anzr299 Feb 21, 2025
14dfcf5
pre commit fix
anzr299 Feb 21, 2025
83c0375
revert attn_implementation according to develop branch
anzr299 Feb 21, 2025
3d08cac
pre-commit fix
anzr299 Feb 21, 2025
a8cedfc
remove xfail for onnx export test
anzr299 Feb 21, 2025
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
Prev Previous commit
Next Next commit
update graphs
anzr299 committed Feb 4, 2025
commit d4e394228baaf696d5566aa3f22bcc2d3aa132af
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
strict digraph {
"0 _conv_w" [id=0, type=get_attr];
"1 add" [id=1, type=add];
"2 conv2d_input" [id=2, type=input];
"3 conv2d" [id=3, type=conv2d];
"4 output" [id=4, type=output];
"0 _conv_w" -> "1 add" [label="(1, 1, 1, 1)", style=solid];
"1 add" -> "3 conv2d" [label="(1, 1, 1, 1)", style=solid];
"2 conv2d_input" -> "3 conv2d" [label=None, style=solid];
"3 conv2d" -> "4 output" [label="(1, 1, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
strict digraph {
"0 _conv_w" [id=0, type=get_attr];
"1 add" [id=1, type=add];
"2 conv2d_input" [id=2, type=input];
"3 conv2d" [id=3, type=conv2d];
"4 output" [id=4, type=output];
"0 _conv_w" -> "1 add" [label="(1, 1, 1, 1)", style=solid];
"1 add" -> "3 conv2d" [label="(1, 1, 1, 1)", style=solid];
"2 conv2d_input" -> "3 conv2d" [label=None, style=solid];
"3 conv2d" -> "4 output" [label="(1, 1, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
strict digraph {
"0 conv_weight" [id=0, type=get_attr];
"1 conv_bias" [id=1, type=get_attr];
"2 conv2d_input" [id=2, type=input];
"3 conv2d" [id=3, type=conv2d];
"4 add" [id=4, type=add];
"5 output" [id=5, type=output];
"0 conv_weight" -> "3 conv2d" [label="(1, 1, 1, 1)", style=solid];
"1 conv_bias" -> "3 conv2d" [label="(1,)", style=solid];
"2 conv2d_input" -> "3 conv2d" [label=None, style=solid];
"3 conv2d" -> "4 add" [label="(1, 1, 3, 3)", style=solid];
"3 conv2d" -> "5 output" [label="(1, 1, 3, 3)", style=solid];
"4 add" -> "5 output" [label="(1, 1, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
strict digraph {
"0 conv_weight" [id=0, type=get_attr];
"1 conv_bias" [id=1, type=get_attr];
"2 conv2d_input" [id=2, type=input];
"3 conv2d" [id=3, type=conv2d];
"4 add" [id=4, type=add];
"5 output" [id=5, type=output];
"0 conv_weight" -> "3 conv2d" [label="(1, 1, 1, 1)", style=solid];
"1 conv_bias" -> "3 conv2d" [label="(1,)", style=solid];
"2 conv2d_input" -> "3 conv2d" [label=None, style=solid];
"3 conv2d" -> "4 add" [label="(1, 1, 3, 3)", style=solid];
"3 conv2d" -> "5 output" [label="(1, 1, 3, 3)", style=solid];
"4 add" -> "5 output" [label="(1, 1, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -15,7 +15,7 @@ strict digraph {
"13 cat" [id=13, type=cat];
"14 conv2d_2" [id=14, type=conv2d];
"15 add" [id=15, type=add];
"16 output_1" [id=16, type=output];
"16 output" [id=16, type=output];
"0 conv_a_weight" -> "8 conv2d" [label="(3, 3, 1, 1)", style=solid];
"1 conv_a_bias" -> "8 conv2d" [label="(3,)", style=solid];
"2 conv_b_weight" -> "9 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
@@ -34,5 +34,5 @@ strict digraph {
"12 const_updated_constant0" -> "13 cat" [label="(1,)", style=solid];
"13 cat" -> "14 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"14 conv2d_2" -> "15 add" [label="(1, 3, 3, 3)", style=solid];
"15 add" -> "16 output_1" [label="(1, 3, 3, 3)", style=solid];
"15 add" -> "16 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -19,7 +19,7 @@ strict digraph {
"17 add" [id=17, type=add];
"18 conv2d_2" [id=18, type=conv2d];
"19 add_1" [id=19, type=add];
"20 output_1" [id=20, type=output];
"20 output" [id=20, type=output];
"0 conv_a_weight" -> "10 quantize_per_channel_default" [label="(3, 3, 1, 1)", style=solid];
"1 conv_a_bias" -> "13 conv2d" [label="(3,)", style=solid];
"2 conv_b_weight" -> "14 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
@@ -44,5 +44,5 @@ strict digraph {
"16 add__1" -> "17 add" [label="(1, 3, 224, 224)", style=solid];
"17 add" -> "18 conv2d_2" [label="(1, 3, 224, 224)", style=solid];
"18 conv2d_2" -> "19 add_1" [label="(1, 3, 224, 224)", style=solid];
"19 add_1" -> "20 output_1" [label="(1, 3, 224, 224)", style=solid];
"19 add_1" -> "20 output" [label="(1, 3, 224, 224)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -18,7 +18,7 @@ strict digraph {
"16 add" [id=16, type=add];
"17 conv2d_2" [id=17, type=conv2d];
"18 add_1" [id=18, type=add];
"19 output_1" [id=19, type=output];
"19 output" [id=19, type=output];
"0 conv_a_bias" -> "12 conv2d" [label="(3,)", style=solid];
"1 conv_b_weight" -> "13 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"2 conv_b_bias" -> "13 conv2d_1" [label="(3,)", style=solid];
@@ -40,5 +40,5 @@ strict digraph {
"15 add__1" -> "16 add" [label="(1, 3, 224, 224)", style=solid];
"16 add" -> "17 conv2d_2" [label="(1, 3, 224, 224)", style=solid];
"17 conv2d_2" -> "18 add_1" [label="(1, 3, 224, 224)", style=solid];
"18 add_1" -> "19 output_1" [label="(1, 3, 224, 224)", style=solid];
"18 add_1" -> "19 output" [label="(1, 3, 224, 224)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -17,7 +17,7 @@ strict digraph {
"15 add" [id=15, type=add];
"16 conv2d_2" [id=16, type=conv2d];
"17 add_1" [id=17, type=add];
"18 output_1" [id=18, type=output];
"18 output" [id=18, type=output];
"0 conv_a_weight" -> "8 quantize_per_tensor_default" [label="(3, 3, 1, 1)", style=solid];
"1 conv_a_bias" -> "11 conv2d" [label="(3,)", style=solid];
"2 conv_b_weight" -> "12 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
@@ -38,5 +38,5 @@ strict digraph {
"14 add__1" -> "15 add" [label="(1, 3, 224, 224)", style=solid];
"15 add" -> "16 conv2d_2" [label="(1, 3, 224, 224)", style=solid];
"16 conv2d_2" -> "17 add_1" [label="(1, 3, 224, 224)", style=solid];
"17 add_1" -> "18 output_1" [label="(1, 3, 224, 224)", style=solid];
"17 add_1" -> "18 output" [label="(1, 3, 224, 224)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -16,7 +16,7 @@ strict digraph {
"14 add" [id=14, type=add];
"15 conv2d_2" [id=15, type=conv2d];
"16 add_1" [id=16, type=add];
"17 output_1" [id=17, type=output];
"17 output" [id=17, type=output];
"0 conv_a_bias" -> "10 conv2d" [label="(3,)", style=solid];
"1 conv_b_weight" -> "11 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"2 conv_b_bias" -> "11 conv2d_1" [label="(3,)", style=solid];
@@ -36,5 +36,5 @@ strict digraph {
"13 add__1" -> "14 add" [label="(1, 3, 224, 224)", style=solid];
"14 add" -> "15 conv2d_2" [label="(1, 3, 224, 224)", style=solid];
"15 conv2d_2" -> "16 add_1" [label="(1, 3, 224, 224)", style=solid];
"16 add_1" -> "17 output_1" [label="(1, 3, 224, 224)", style=solid];
"16 add_1" -> "17 output" [label="(1, 3, 224, 224)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -15,7 +15,7 @@ strict digraph {
"13 cat" [id=13, type=cat];
"14 conv2d_2" [id=14, type=conv2d];
"15 add" [id=15, type=add];
"16 output_1" [id=16, type=output];
"16 output" [id=16, type=output];
"0 const" -> "13 cat" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "8 conv2d" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_bias" -> "8 conv2d" [label="(3,)", style=solid];
@@ -34,5 +34,5 @@ strict digraph {
"12 add__1" -> "13 cat" [label="(1, 3, 3, 3)", style=solid];
"13 cat" -> "14 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"14 conv2d_2" -> "15 add" [label="(1, 3, 3, 3)", style=solid];
"15 add" -> "16 output_1" [label="(1, 3, 3, 3)", style=solid];
"15 add" -> "16 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -10,7 +10,7 @@ strict digraph {
"8 dequantize_per_channel_default" [id=8, type=dequantize_per_channel];
"9 linear_1" [id=9, type=linear];
"10 add" [id=10, type=add];
"11 output_1" [id=11, type=output];
"11 output" [id=11, type=output];
"0 linear_act_weight" -> "7 quantize_per_channel_default" [label="(3, 3)", style=solid];
"1 linear_act_bias" -> "9 linear_1" [label="(3,)", style=solid];
"2 x" -> "9 linear_1" [label="(1, 3, 3, 3)", style=solid];
@@ -22,5 +22,5 @@ strict digraph {
"7 quantize_per_channel_default" -> "8 dequantize_per_channel_default" [label="(3, 3)", style=solid];
"8 dequantize_per_channel_default" -> "9 linear_1" [label="(3, 3)", style=solid];
"9 linear_1" -> "10 add" [label="(1, 3, 3, 3)", style=solid];
"10 add" -> "11 output_1" [label="(1, 3, 3, 3)", style=solid];
"10 add" -> "11 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -8,13 +8,13 @@ strict digraph {
"6 dequantize_per_tensor_default" [id=6, type=dequantize_per_tensor];
"7 linear_1" [id=7, type=linear];
"8 add" [id=8, type=add];
"9 output_1" [id=9, type=output];
"9 output" [id=9, type=output];
"0 linear_act_weight" -> "5 quantize_per_tensor_default" [label="(3, 3)", style=solid];
"1 linear_act_bias" -> "7 linear_1" [label="(3,)", style=solid];
"2 x" -> "7 linear_1" [label="(1, 3, 3, 3)", style=solid];
"4 _frozen_param0" -> "8 add" [label="(3, 3)", style=solid];
"5 quantize_per_tensor_default" -> "6 dequantize_per_tensor_default" [label="(3, 3)", style=solid];
"6 dequantize_per_tensor_default" -> "7 linear_1" [label="(3, 3)", style=solid];
"7 linear_1" -> "8 add" [label="(1, 3, 3, 3)", style=solid];
"8 add" -> "9 output_1" [label="(1, 3, 3, 3)", style=solid];
"8 add" -> "9 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -21,7 +21,7 @@ strict digraph {
"19 cat" [id=19, type=cat];
"20 conv2d_2" [id=20, type=conv2d];
"21 add" [id=21, type=add];
"22 output_1" [id=22, type=output];
"22 output" [id=22, type=output];
"0 const" -> "18 TEST_MODULE_5" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "10 TEST_MODULE_1" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_bias" -> "11 conv2d" [label="(3,)", style=solid];
@@ -46,5 +46,5 @@ strict digraph {
"18 TEST_MODULE_5" -> "19 cat" [label="(1, 3, 3, 3)", style=solid];
"19 cat" -> "20 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"20 conv2d_2" -> "21 add" [label="(1, 3, 3, 3)", style=solid];
"21 add" -> "22 output_1" [label="(1, 3, 3, 3)", style=solid];
"21 add" -> "22 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -21,7 +21,7 @@ strict digraph {
"19 cat" [id=19, type=cat];
"20 conv2d_2" [id=20, type=conv2d];
"21 add" [id=21, type=add];
"22 output_1" [id=22, type=output];
"22 output" [id=22, type=output];
"0 const" -> "18 TEST_MODULE_5" [label="(1, 3, 3, 3)", style=solid];
"0 const" -> "19 cat" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "10 TEST_MODULE_1" [label="(3, 3, 1, 1)", style=solid];
@@ -46,5 +46,5 @@ strict digraph {
"16 add__1" -> "19 cat" [label="(1, 3, 3, 3)", style=solid];
"19 cat" -> "20 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"20 conv2d_2" -> "21 add" [label="(1, 3, 3, 3)", style=solid];
"21 add" -> "22 output_1" [label="(1, 3, 3, 3)", style=solid];
"21 add" -> "22 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -11,7 +11,7 @@ strict digraph {
"9 add" [id=9, type=add];
"10 conv2d_2" [id=10, type=conv2d];
"11 add_1" [id=11, type=add];
"12 output_1" [id=12, type=output];
"12 output" [id=12, type=output];
"0 conv_b_weight" -> "6 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"1 conv_b_bias" -> "6 conv2d_1" [label="(3,)", style=solid];
"2 conv_c_weight" -> "10 conv2d_2" [label="(3, 3, 1, 1)", style=solid];
@@ -26,5 +26,5 @@ strict digraph {
"8 add__1" -> "9 add" [label="(1, 3, 3, 3)", style=solid];
"9 add" -> "10 conv2d_2" [label="(1, 3, 3, 3)", style=solid];
"10 conv2d_2" -> "11 add_1" [label="(1, 3, 3, 3)", style=solid];
"11 add_1" -> "12 output_1" [label="(1, 3, 3, 3)", style=solid];
"11 add_1" -> "12 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -16,10 +16,10 @@ strict digraph {
"14 cat" [id=14, type=cat];
"15 conv2d_2" [id=15, type=conv2d];
"16 add" [id=16, type=add];
"17 output_1" [id=17, type=output];
"17 output" [id=17, type=output];
"0 const" -> "1 const_cloned" [label="(1, 3, 3, 3)", style=solid];
"0 const" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"1 const_cloned" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"1 const_cloned" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
"2 conv_a_weight" -> "10 conv2d" [label="(3, 3, 1, 1)", style=solid];
"3 conv_a_bias" -> "10 conv2d" [label="(3,)", style=solid];
"4 conv_b_weight" -> "11 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
@@ -37,5 +37,5 @@ strict digraph {
"13 add__1" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"14 cat" -> "15 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"15 conv2d_2" -> "16 add" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -16,7 +16,7 @@ strict digraph {
"14 cat" [id=14, type=cat];
"15 conv2d_2" [id=15, type=conv2d];
"16 add" [id=16, type=add];
"17 output_1" [id=17, type=output];
"17 output" [id=17, type=output];
"0 const" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "9 conv2d" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_bias" -> "9 conv2d" [label="(3,)", style=solid];
@@ -34,8 +34,8 @@ strict digraph {
"11 add_" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"12 add__1" -> "13 add__1_cloned" [label="(1, 3, 3, 3)", style=solid];
"12 add__1" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"13 add__1_cloned" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"13 add__1_cloned" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
"14 cat" -> "15 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"15 conv2d_2" -> "16 add" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -16,13 +16,13 @@ strict digraph {
"14 cat" [id=14, type=cat];
"15 conv2d_2" [id=15, type=conv2d];
"16 add" [id=16, type=add];
"17 output_1" [id=17, type=output];
"17 output" [id=17, type=output];
"0 const" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "10 conv2d" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_bias" -> "10 conv2d" [label="(3,)", style=solid];
"3 conv_b_weight" -> "4 conv_b_weight_cloned" [label="(3, 3, 1, 1)", style=solid];
"3 conv_b_weight" -> "11 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"4 conv_b_weight_cloned" -> "17 output_1" [label="(3, 3, 1, 1)", style=solid];
"4 conv_b_weight_cloned" -> "17 output" [label="(3, 3, 1, 1)", style=solid];
"5 conv_b_bias" -> "11 conv2d_1" [label="(3,)", style=solid];
"6 conv_c_weight" -> "15 conv2d_2" [label="(3, 9, 1, 1)", style=solid];
"7 conv_c_bias" -> "15 conv2d_2" [label="(3,)", style=solid];
@@ -37,5 +37,5 @@ strict digraph {
"13 add__1" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"14 cat" -> "15 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"15 conv2d_2" -> "16 add" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -16,7 +16,7 @@ strict digraph {
"14 cat" [id=14, type=cat];
"15 conv2d_2" [id=15, type=conv2d];
"16 add" [id=16, type=add];
"17 output_1" [id=17, type=output];
"17 output" [id=17, type=output];
"0 const" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "10 conv2d" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_bias" -> "10 conv2d" [label="(3,)", style=solid];
@@ -29,13 +29,13 @@ strict digraph {
"7 bias" -> "16 add" [label="(1,)", style=solid];
"8 x" -> "9 x_cloned" [label="(1, 3, 3, 3)", style=solid];
"8 x" -> "10 conv2d" [label="(1, 3, 3, 3)", style=solid];
"9 x_cloned" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"9 x_cloned" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
"10 conv2d" -> "11 conv2d_1" [label="(1, 3, 3, 3)", style=solid];
"10 conv2d" -> "12 add_" [label="(1, 3, 3, 3)", style=solid];
"11 conv2d_1" -> "13 add__1" [label="(1, 3, 3, 3)", style=solid];
"12 add_" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"13 add__1" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"14 cat" -> "15 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"15 conv2d_2" -> "16 add" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -16,11 +16,11 @@ strict digraph {
"14 cat" [id=14, type=cat];
"15 conv2d_2" [id=15, type=conv2d];
"16 add" [id=16, type=add];
"17 output_1" [id=17, type=output];
"17 output" [id=17, type=output];
"0 const" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "2 conv_a_weight_cloned" [label="(3, 3, 1, 1)", style=solid];
"1 conv_a_weight" -> "10 conv2d" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_weight_cloned" -> "17 output_1" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_weight_cloned" -> "17 output" [label="(3, 3, 1, 1)", style=solid];
"3 conv_a_bias" -> "10 conv2d" [label="(3,)", style=solid];
"4 conv_b_weight" -> "11 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"5 conv_b_bias" -> "11 conv2d_1" [label="(3,)", style=solid];
@@ -37,5 +37,5 @@ strict digraph {
"13 add__1" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"14 cat" -> "15 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"15 conv2d_2" -> "16 add" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -16,7 +16,7 @@ strict digraph {
"14 cat" [id=14, type=cat];
"15 conv2d_2" [id=15, type=conv2d];
"16 add" [id=16, type=add];
"17 output_1" [id=17, type=output];
"17 output" [id=17, type=output];
"0 const" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "9 conv2d" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_bias" -> "9 conv2d" [label="(3,)", style=solid];
@@ -31,11 +31,11 @@ strict digraph {
"9 conv2d" -> "10 conv2d_cloned" [label="(1, 3, 3, 3)", style=solid];
"9 conv2d" -> "11 conv2d_1" [label="(1, 3, 3, 3)", style=solid];
"9 conv2d" -> "12 add_" [label="(1, 3, 3, 3)", style=solid];
"10 conv2d_cloned" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"10 conv2d_cloned" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
"11 conv2d_1" -> "13 add__1" [label="(1, 3, 3, 3)", style=solid];
"12 add_" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"13 add__1" -> "14 cat" [label="(1, 3, 3, 3)", style=solid];
"14 cat" -> "15 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"15 conv2d_2" -> "16 add" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output_1" [label="(1, 3, 3, 3)", style=solid];
"16 add" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -19,7 +19,7 @@ strict digraph {
"17 cat" [id=17, type=cat];
"18 conv2d_2" [id=18, type=conv2d];
"19 add" [id=19, type=add];
"20 output_1" [id=20, type=output];
"20 output" [id=20, type=output];
"0 const" -> "15 quantize_per_channel_default" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "9 conv2d" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_bias" -> "9 conv2d" [label="(3,)", style=solid];
@@ -44,5 +44,5 @@ strict digraph {
"16 dequantize_per_channel_default" -> "17 cat" [label="(1, 3, 3, 3)", style=solid];
"17 cat" -> "18 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"18 conv2d_2" -> "19 add" [label="(1, 3, 3, 3)", style=solid];
"19 add" -> "20 output_1" [label="(1, 3, 3, 3)", style=solid];
"19 add" -> "20 output" [label="(1, 3, 3, 3)", style=solid];
}
Original file line number Diff line number Diff line change
@@ -17,7 +17,7 @@ strict digraph {
"15 cat" [id=15, type=cat];
"16 conv2d_2" [id=16, type=conv2d];
"17 add" [id=17, type=add];
"18 output_1" [id=18, type=output];
"18 output" [id=18, type=output];
"0 const" -> "13 quantize_per_tensor_default" [label="(1, 3, 3, 3)", style=solid];
"1 conv_a_weight" -> "9 conv2d" [label="(3, 3, 1, 1)", style=solid];
"2 conv_a_bias" -> "9 conv2d" [label="(3,)", style=solid];
@@ -38,5 +38,5 @@ strict digraph {
"14 dequantize_per_tensor_default" -> "15 cat" [label="(1, 3, 3, 3)", style=solid];
"15 cat" -> "16 conv2d_2" [label="(1, 9, 3, 3)", style=solid];
"16 conv2d_2" -> "17 add" [label="(1, 3, 3, 3)", style=solid];
"17 add" -> "18 output_1" [label="(1, 3, 3, 3)", style=solid];
"17 add" -> "18 output" [label="(1, 3, 3, 3)", style=solid];
}
Loading