@@ -1927,16 +1927,24 @@ def version_11(cls, ctx, node, **kwargs):
1927
1927
# gap_pos_k_graph
1928
1928
gap_pos_k_graph = body_graph .create_new_graph_with_same_config ()
1929
1929
gap_pos_k_graph .parent_graph = body_graph
1930
- gap_pos_k = gap_pos_k_graph .make_node ('Concat' , [const_zero .output [0 ], processed_gap .output [0 ]], attr = {'axis' : 0 }) \
1930
+ gap_pos_k = gap_pos_k_graph .make_node ('Concat' , [const_zero .output [0 ],
1931
+ processed_gap .output [0 ]],
1932
+ attr = {'axis' : 0 }) \
1931
1933
if align .startswith ('LEFT' ) \
1932
- else gap_pos_k_graph .make_node ('Concat' , [processed_gap .output [0 ], const_zero .output [0 ]], attr = {'axis' : 0 })
1934
+ else gap_pos_k_graph .make_node ('Concat' , [processed_gap .output [0 ],
1935
+ const_zero .output [0 ]],
1936
+ attr = {'axis' : 0 })
1933
1937
gap_pos_k_graph .add_graph_output (gap_pos_k .output [0 ], TensorProto .INT64 , [- 1 ])
1934
1938
# gap_neg_k_graph
1935
1939
gap_neg_k_graph = body_graph .create_new_graph_with_same_config ()
1936
1940
gap_neg_k_graph .parent_graph = body_graph
1937
- gap_neg_k = gap_neg_k_graph .make_node ('Concat' , [const_zero .output [0 ], processed_gap .output [0 ]], attr = {'axis' : 0 }) \
1941
+ gap_neg_k = gap_neg_k_graph .make_node ('Concat' , [const_zero .output [0 ],
1942
+ processed_gap .output [0 ]],
1943
+ attr = {'axis' : 0 }) \
1938
1944
if align .endswith ('LEFT' ) \
1939
- else gap_neg_k_graph .make_node ('Concat' , [processed_gap .output [0 ], const_zero .output [0 ]], attr = {'axis' : 0 })
1945
+ else gap_neg_k_graph .make_node ('Concat' , [processed_gap .output [0 ],
1946
+ const_zero .output [0 ]],
1947
+ attr = {'axis' : 0 })
1940
1948
gap_neg_k_graph .add_graph_output (gap_neg_k .output [0 ], TensorProto .INT64 , [- 1 ])
1941
1949
# pad output with gap
1942
1950
gap_k = body_graph .make_node ('If' , [is_k_noneg .output [0 ]])
0 commit comments