Update cts generator for per channel 1

Signed-off-by: cuiyanx <[email protected]>
cuiyanx · Mar 10, 2020 · cb8e2be · cb8e2be
1 parent bac8ea9
commit cb8e2be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/select_per_channel/cts_generator.py b/src/select_per_channel/cts_generator.py
@@ -552,7 +552,7 @@ def DumpJSTest(model, example, js_fd):
             print ("    model.addOperand(%s);"%op.type, file = js_fd)
 
             if str(op.type) in per_channel_types.keys():
-                print ("    model.setOperandSymmPerChannelQuantParams(%s, %s);"%(op.type, per_channel_types[str(op.type)]), file = js_fd)
+                print ("    model.setOperandSymmPerChannelQuantParams(operandIndex++, %s);"%per_channel_types[str(op.type)], file = js_fd)
         print ("", file = js_fd)
 
         # set other inputs value(support only one input)