Skip to content

Commit 18e9512

Browse files
XrekiXreki
committed
Update input constraints and json.
1 parent 82d514e commit 18e9512

File tree

161 files changed

+591
-169
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

161 files changed

+591
-169
lines changed

‎graph_net/tools/init_input_tensor_constraints.sh‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ python3 -m graph_net.model_path_handler \
1515
"handler_path": "$GRAPH_NET_ROOT/constraint_util.py",
1616
"handler_class_name": "UpdateInputTensorConstraints",
1717
"handler_config": {
18-
"resume": true,
18+
"resume": false,
1919
"model_path_prefix": "$GRAPH_NET_ROOT/../",
2020
"data_input_predicator_filepath": "$GRAPH_NET_ROOT/torch/constraint_util.py",
2121
"data_input_predicator_class_name": "NaiveDataInputPredicator",
@@ -30,6 +30,7 @@ python3 -m graph_net.model_path_handler \
3030
"naive_call_method_reshape_pass",
3131
"naive_call_method_expand_pass",
3232
"non_batch_call_method_expand_pass",
33+
"non_batch_call_method_view_pass",
3334
"non_batch_call_function_arange_pass",
3435
"non_batch_call_function_getitem_slice_pass",
3536
"non_batch_call_function_full_pass",

‎samples/mmpose/S-ViPNAS-Res50/graph_net.json‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,13 @@
66
"source": "mmpose",
77
"heuristic_tag": "computer_vision",
88
"dimension_generalization_passes": [
9+
"batch_call_method_view_pass",
10+
"tuple_arg_call_method_view_pass",
911
"naive_call_method_view_pass",
1012
"naive_call_method_reshape_pass",
1113
"naive_call_method_expand_pass",
1214
"non_batch_call_method_expand_pass",
15+
"non_batch_call_method_view_pass",
1316
"non_batch_call_function_arange_pass",
1417
"non_batch_call_function_getitem_slice_pass",
1518
"non_batch_call_function_full_pass",

‎samples/mmpose/S-ViPNAS-Res50/input_tensor_constraints.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from sympy import Symbol
1+
from sympy import Symbol, Expr, Rel, Eq
22

33
S0 = Symbol("S0")
44

‎samples/mmpose/ViTPose-B/graph_net.json‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,13 @@
66
"source": "mmpose",
77
"heuristic_tag": "computer_vision",
88
"dimension_generalization_passes": [
9+
"batch_call_method_view_pass",
10+
"tuple_arg_call_method_view_pass",
911
"naive_call_method_view_pass",
1012
"naive_call_method_reshape_pass",
1113
"naive_call_method_expand_pass",
1214
"non_batch_call_method_expand_pass",
15+
"non_batch_call_method_view_pass",
1316
"non_batch_call_function_arange_pass",
1417
"non_batch_call_function_getitem_slice_pass",
1518
"non_batch_call_function_full_pass",

‎samples/mmpose/ViTPose-B/input_tensor_constraints.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from sympy import Symbol
1+
from sympy import Symbol, Expr, Rel, Eq
22

33
S0 = Symbol("S0")
44
S1 = Symbol("S1")

‎samples/mmseg/CCNet_R101/input_tensor_constraints.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from sympy import Symbol
1+
from sympy import Symbol, Expr, Rel, Eq
22

33
S0 = Symbol("S0")
44

‎samples/mmseg/DeepLabV3+_R101/input_tensor_constraints.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from sympy import Symbol
1+
from sympy import Symbol, Expr, Rel, Eq
22

33
S0 = Symbol("S0")
44

‎samples/mmseg/PoolFormer_m36/graph_net.json‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,6 @@
55
"dynamic": false,
66
"source": "mmseg",
77
"heuristic_tag": "computer_vision",
8-
"dimension_generalization_passes": [],
9-
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
8+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
9+
"dimension_generalization_passes": []
1010
}

‎samples/mmseg/PoolFormer_m36/input_tensor_constraints.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from sympy import Symbol
1+
from sympy import Symbol, Expr, Rel, Eq
22

33
S0 = Symbol("S0")
44
S1 = Symbol("S1")

‎samples/mmseg/Twins_pcpvt-b_uperhead/graph_net.json‎

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,17 +5,20 @@
55
"dynamic": false,
66
"source": "mmseg",
77
"heuristic_tag": "computer_vision",
8+
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier",
89
"dimension_generalization_passes": [
10+
"batch_call_method_view_pass",
11+
"tuple_arg_call_method_view_pass",
912
"naive_call_method_view_pass",
1013
"naive_call_method_reshape_pass",
1114
"naive_call_method_expand_pass",
1215
"non_batch_call_method_expand_pass",
16+
"non_batch_call_method_view_pass",
1317
"non_batch_call_function_arange_pass",
1418
"non_batch_call_function_getitem_slice_pass",
1519
"non_batch_call_function_full_pass",
1620
"non_batch_call_function_full_plus_one_pass",
1721
"non_batch_call_function_zeros_pass",
1822
"non_batch_call_function_arange_plus_one_pass"
19-
],
20-
"symbolic_dimension_reifier": "naive_cv_sym_dim_reifier"
23+
]
2124
}

0 commit comments

Comments
 (0)