未验证 提交 65974860 编写于 作者: A Alexandre Eichenberger 提交者: GitHub

replace whitelist by safelist (#3900)

Signed-off-by: NAlexandre Eichenberger <alexe@us.ibm.com>
上级 e2072611
......@@ -45,7 +45,7 @@ class DummyBackend(onnx.backend.base.Backend):
model = onnx.shape_inference.infer_shapes(model)
value_infos = {vi.name: vi for vi in itertools.chain(model.graph.value_info, model.graph.output)}
if do_enforce_test_coverage_whitelist(model):
if do_enforce_test_coverage_safelist(model):
for node in model.graph.node:
for i, output in enumerate(node.output):
if node.op_type == 'Dropout' and i != 0:
......@@ -79,13 +79,13 @@ class DummyBackend(onnx.backend.base.Backend):
return False
test_coverage_whitelist = set(
test_coverage_safelist = set(
['bvlc_alexnet', 'densenet121', 'inception_v1', 'inception_v2',
'resnet50', 'shufflenet', 'SingleRelu', 'squeezenet_old', 'vgg19', 'zfnet'])
def do_enforce_test_coverage_whitelist(model): # type: (ModelProto) -> bool
if model.graph.name not in test_coverage_whitelist:
def do_enforce_test_coverage_safelist(model): # type: (ModelProto) -> bool
if model.graph.name not in test_coverage_safelist:
return False
for node in model.graph.node:
if node.op_type in set(['RNN', 'LSTM', 'GRU']):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册