Skip to content

Commit

Permalink
[TorchFX] Graph tests extended with edge shapes (#2918)
Browse files Browse the repository at this point in the history 
### Changes

Graph tests extended with edges shape

### Reason for changes

To test edge shapes by graph tests

### Related tickets

#2904
  • Loading branch information
@daniil-lyakhov
daniil-lyakhov authored Sep 30, 2024
1 parent b94e1df commit 2c8b70c
Show file tree
Hide file tree
Showing 27 changed files with 10,836 additions and 10,828 deletions.
8 changes: 4 additions & 4 deletions ...data/reference_graphs/fx/extracted/ConvolutionWithAllConstantInputsModelconv2d_conv2d.dot
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ strict digraph {
"2 conv2d_input" [id=2, type=input];
"3 conv2d" [id=3, type=conv2d];
"4 output" [id=4, type=output];
"0 _param_constant0" -> "1 add";
"1 add" -> "3 conv2d";
"2 conv2d_input" -> "3 conv2d";
"3 conv2d" -> "4 output";
"0 _param_constant0" -> "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, 1, 1)", style=solid];
}
8 changes: 4 additions & 4 deletions ...rch/data/reference_graphs/fx/extracted/ConvolutionWithNotTensorBiasModelconv2d_conv2d.dot
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ strict digraph {
"2 conv2d_input" [id=2, type=input];
"3 conv2d" [id=3, type=conv2d];
"4 output" [id=4, type=output];
"0 _param_constant0" -> "1 add";
"1 add" -> "3 conv2d";
"2 conv2d_input" -> "3 conv2d";
"3 conv2d" -> "4 output";
"0 _param_constant0" -> "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];
}
12 changes: 6 additions & 6 deletions ...s/torch/data/reference_graphs/fx/extracted/MultiBranchesConnectedModelconv2d_1_add__1.dot
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,10 @@ strict digraph {
"4 _tensor_constant0_1" [id=4, type=get_attr];
"5 add__1" [id=5, type=add_];
"6 output" [id=6, type=output];
"0 _param_constant2" -> "3 conv2d_1";
"1 _param_constant3" -> "3 conv2d_1";
"2 conv2d_1_input" -> "3 conv2d_1";
"3 conv2d_1" -> "5 add__1";
"4 _tensor_constant0_1" -> "5 add__1";
"5 add__1" -> "6 output";
"0 _param_constant2" -> "3 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"1 _param_constant3" -> "3 conv2d_1" [label="(3,)", style=solid];
"2 conv2d_1_input" -> "3 conv2d_1" [label=None, style=solid];
"3 conv2d_1" -> "5 add__1" [label="(1, 3, 3, 3)", style=solid];
"4 _tensor_constant0_1" -> "5 add__1" [label="(1,)", style=solid];
"5 add__1" -> "6 output" [label="(1, 3, 3, 3)", style=solid];
}
36 changes: 18 additions & 18 deletions tests/torch/data/reference_graphs/fx/extracted/MultiBranchesConnectedModelconv2d_add__1.dot
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,22 +17,22 @@ strict digraph {
"15 _tensor_constant0_2" [id=15, type=get_attr];
"16 add_1" [id=16, type=add];
"17 output" [id=17, type=output];
"0 _param_constant0" -> "3 conv2d";
"1 _param_constant1" -> "3 conv2d";
"2 conv2d_input" -> "3 conv2d";
"3 conv2d" -> "6 conv2d_1";
"3 conv2d" -> "8 add_";
"4 _param_constant2" -> "6 conv2d_1";
"5 _param_constant3" -> "6 conv2d_1";
"6 conv2d_1" -> "10 add__1";
"7 _tensor_constant0" -> "8 add_";
"8 add_" -> "11 add";
"9 _tensor_constant0_1" -> "10 add__1";
"10 add__1" -> "11 add";
"10 add__1" -> "17 output";
"11 add" -> "14 conv2d_2";
"12 _param_constant4" -> "14 conv2d_2";
"13 _param_constant5" -> "14 conv2d_2";
"14 conv2d_2" -> "16 add_1";
"15 _tensor_constant0_2" -> "16 add_1";
"0 _param_constant0" -> "3 conv2d" [label="(3, 3, 1, 1)", style=solid];
"1 _param_constant1" -> "3 conv2d" [label="(3,)", style=solid];
"2 conv2d_input" -> "3 conv2d" [label=None, style=solid];
"3 conv2d" -> "6 conv2d_1" [label="(1, 3, 3, 3)", style=solid];
"3 conv2d" -> "8 add_" [label="(1, 3, 3, 3)", style=solid];
"4 _param_constant2" -> "6 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"5 _param_constant3" -> "6 conv2d_1" [label="(3,)", style=solid];
"6 conv2d_1" -> "10 add__1" [label="(1, 3, 3, 3)", style=solid];
"7 _tensor_constant0" -> "8 add_" [label="(1,)", style=solid];
"8 add_" -> "11 add" [label="(1, 3, 3, 3)", style=solid];
"9 _tensor_constant0_1" -> "10 add__1" [label="(1,)", style=solid];
"10 add__1" -> "11 add" [label="(1, 3, 3, 3)", style=solid];
"10 add__1" -> "17 output" [label="(1, 3, 3, 3)", style=solid];
"11 add" -> "14 conv2d_2" [label="(1, 3, 3, 3)", style=solid];
"12 _param_constant4" -> "14 conv2d_2" [label="(3, 3, 1, 1)", style=solid];
"13 _param_constant5" -> "14 conv2d_2" [label="(3,)", style=solid];
"14 conv2d_2" -> "16 add_1" [label="(1, 3, 3, 3)", style=solid];
"15 _tensor_constant0_2" -> "16 add_1" [label="(1,)", style=solid];
}
28 changes: 14 additions & 14 deletions ...s/torch/data/reference_graphs/fx/extracted/MultiBranchesConnectedModelconv2d_add__add.dot
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,18 +12,18 @@ strict digraph {
"10 add__1" [id=10, type=add_];
"11 add" [id=11, type=add];
"12 output" [id=12, type=output];
"0 _param_constant0" -> "3 conv2d";
"1 _param_constant1" -> "3 conv2d";
"2 conv2d_input" -> "3 conv2d";
"3 conv2d" -> "6 conv2d_1";
"3 conv2d" -> "8 add_";
"4 _param_constant2" -> "6 conv2d_1";
"5 _param_constant3" -> "6 conv2d_1";
"6 conv2d_1" -> "10 add__1";
"7 _tensor_constant0" -> "8 add_";
"8 add_" -> "11 add";
"8 add_" -> "12 output";
"9 _tensor_constant0_1" -> "10 add__1";
"10 add__1" -> "11 add";
"11 add" -> "12 output";
"0 _param_constant0" -> "3 conv2d" [label="(3, 3, 1, 1)", style=solid];
"1 _param_constant1" -> "3 conv2d" [label="(3,)", style=solid];
"2 conv2d_input" -> "3 conv2d" [label=None, style=solid];
"3 conv2d" -> "6 conv2d_1" [label="(1, 3, 3, 3)", style=solid];
"3 conv2d" -> "8 add_" [label="(1, 3, 3, 3)", style=solid];
"4 _param_constant2" -> "6 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"5 _param_constant3" -> "6 conv2d_1" [label="(3,)", style=solid];
"6 conv2d_1" -> "10 add__1" [label="(1, 3, 3, 3)", style=solid];
"7 _tensor_constant0" -> "8 add_" [label="(1,)", style=solid];
"8 add_" -> "11 add" [label="(1, 3, 3, 3)", style=solid];
"8 add_" -> "12 output" [label="(1, 3, 3, 3)", style=solid];
"9 _tensor_constant0_1" -> "10 add__1" [label="(1,)", style=solid];
"10 add__1" -> "11 add" [label="(1, 3, 3, 3)", style=solid];
"11 add" -> "12 output" [label="(1, 3, 3, 3)", style=solid];
}
24 changes: 12 additions & 12 deletions .../reference_graphs/fx/extracted/MultiBranchesConnectedModelconv2d_conv2d_1_add__add__1.dot
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,16 +12,16 @@ strict digraph {
"10 _tensor_constant0_1" [id=10, type=get_attr];
"11 add__1" [id=11, type=add_];
"12 output" [id=12, type=output];
"0 _param_constant0" -> "3 conv2d";
"1 _param_constant1" -> "3 conv2d";
"2 conv2d_input" -> "3 conv2d";
"3 conv2d" -> "9 add_";
"4 _param_constant2" -> "7 conv2d_1";
"5 _param_constant3" -> "7 conv2d_1";
"6 conv2d_1_input" -> "7 conv2d_1";
"7 conv2d_1" -> "11 add__1";
"8 _tensor_constant0" -> "9 add_";
"9 add_" -> "12 output";
"10 _tensor_constant0_1" -> "11 add__1";
"11 add__1" -> "12 output";
"0 _param_constant0" -> "3 conv2d" [label="(3, 3, 1, 1)", style=solid];
"1 _param_constant1" -> "3 conv2d" [label="(3,)", style=solid];
"2 conv2d_input" -> "3 conv2d" [label=None, style=solid];
"3 conv2d" -> "9 add_" [label="(1, 3, 3, 3)", style=solid];
"4 _param_constant2" -> "7 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"5 _param_constant3" -> "7 conv2d_1" [label="(3,)", style=solid];
"6 conv2d_1_input" -> "7 conv2d_1" [label=None, style=solid];
"7 conv2d_1" -> "11 add__1" [label="(1, 3, 3, 3)", style=solid];
"8 _tensor_constant0" -> "9 add_" [label="(1,)", style=solid];
"9 add_" -> "12 output" [label="(1, 3, 3, 3)", style=solid];
"10 _tensor_constant0_1" -> "11 add__1" [label="(1,)", style=solid];
"11 add__1" -> "12 output" [label="(1, 3, 3, 3)", style=solid];
}
32 changes: 16 additions & 16 deletions ...s/torch/data/reference_graphs/fx/extracted/MultiBranchesConnectedModelconv2d_conv2d_2.dot
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,20 +15,20 @@ strict digraph {
"13 _param_constant5" [id=13, type=get_attr];
"14 conv2d_2" [id=14, type=conv2d];
"15 output" [id=15, type=output];
"0 _param_constant0" -> "3 conv2d";
"1 _param_constant1" -> "3 conv2d";
"2 conv2d_input" -> "3 conv2d";
"3 conv2d" -> "6 conv2d_1";
"3 conv2d" -> "8 add_";
"4 _param_constant2" -> "6 conv2d_1";
"5 _param_constant3" -> "6 conv2d_1";
"6 conv2d_1" -> "10 add__1";
"7 _tensor_constant0" -> "8 add_";
"8 add_" -> "11 add";
"9 _tensor_constant0_1" -> "10 add__1";
"10 add__1" -> "11 add";
"11 add" -> "14 conv2d_2";
"12 _param_constant4" -> "14 conv2d_2";
"13 _param_constant5" -> "14 conv2d_2";
"14 conv2d_2" -> "15 output";
"0 _param_constant0" -> "3 conv2d" [label="(3, 3, 1, 1)", style=solid];
"1 _param_constant1" -> "3 conv2d" [label="(3,)", style=solid];
"2 conv2d_input" -> "3 conv2d" [label=None, style=solid];
"3 conv2d" -> "6 conv2d_1" [label="(1, 3, 3, 3)", style=solid];
"3 conv2d" -> "8 add_" [label="(1, 3, 3, 3)", style=solid];
"4 _param_constant2" -> "6 conv2d_1" [label="(3, 3, 1, 1)", style=solid];
"5 _param_constant3" -> "6 conv2d_1" [label="(3,)", style=solid];
"6 conv2d_1" -> "10 add__1" [label="(1, 3, 3, 3)", style=solid];
"7 _tensor_constant0" -> "8 add_" [label="(1,)", style=solid];
"8 add_" -> "11 add" [label="(1, 3, 3, 3)", style=solid];
"9 _tensor_constant0_1" -> "10 add__1" [label="(1,)", style=solid];
"10 add__1" -> "11 add" [label="(1, 3, 3, 3)", style=solid];
"11 add" -> "14 conv2d_2" [label="(1, 3, 3, 3)", style=solid];
"12 _param_constant4" -> "14 conv2d_2" [label="(3, 3, 1, 1)", style=solid];
"13 _param_constant5" -> "14 conv2d_2" [label="(3,)", style=solid];
"14 conv2d_2" -> "15 output" [label="(1, 3, 3, 3)", style=solid];
}
Loading

0 comments on commit 2c8b70c

Please sign in to comment.