File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -46,7 +46,7 @@ def test_dequantize(self):
4646 {"X" : [5 , 2 ]},
4747 debug = True ,
4848 adjust_batch = True ,
49- black_list = [0 ],
49+ block_list = [0 ],
5050 use_onnx = False ,
5151 )
5252 num_onnxified_ops = sum (
Original file line number Diff line number Diff line change @@ -25,7 +25,7 @@ def onnxifi_caffe2_net(
2525 use_onnx = True ,
2626 merge_fp32_inputs_into_fp16 = False ,
2727 adjust_batch = True ,
28- black_list = None ,
28+ block_list = None ,
2929 weight_names = None ,
3030 net_ssa_rewritten = False ,
3131 timeout = 0 ):
@@ -47,7 +47,7 @@ def onnxifi_caffe2_net(
4747 shape_hints .max_feature_len = max_seq_size
4848 pred_net_str = C .onnxifi (pred_net .SerializeToString (),
4949 shape_hints .SerializeToString (),
50- black_list if black_list else [],
50+ block_list if block_list else [],
5151 weight_names if weight_names is not None else [],
5252 max_batch_size ,
5353 max_seq_size ,
You can’t perform that action at this time.
0 commit comments