Skip to content

Commit 2ded073

Browse files
authored
fix split node conversion (#1124)
1 parent 5557c73 commit 2ded073

File tree

1 file changed

+1
-1
lines changed
  • neural_compressor/adaptor/ox_utils/operators

1 file changed

+1
-1
lines changed

neural_compressor/adaptor/ox_utils/operators/split.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def convert(self):
5151
quantized_input_names = []
5252
quantized_input_names.append(parent.input[0])
5353
if len(node.input) > 1:
54-
quantized_input_names = quantized_input_names.extend(node.input[1:])
54+
quantized_input_names.extend(node.input[1:])
5555
outputs = []
5656
for output in node.output:
5757
if output in self.quantizer.model.input_name_to_nodes:

0 commit comments

Comments
 (0)