提交 1f827c0f 编写于 作者: C Channingss

fix bug

上级 b53bba9e
...@@ -44,7 +44,7 @@ class ONNXOpMapper(OpMapper): ...@@ -44,7 +44,7 @@ class ONNXOpMapper(OpMapper):
elif op in self.opset.default_op_mapping: elif op in self.opset.default_op_mapping:
self.opset.directly_map(node) self.opset.directly_map(node)
elif op in custom_layers: elif op in custom_layers:
self.opser.deal_custom_layer(node) self.opset.deal_custom_layer(node)
elif op in self.opset.elementwise_ops: elif op in self.opset.elementwise_ops:
self.opset.elementwise_map(node) self.opset.elementwise_map(node)
print("Nodes converted.") print("Nodes converted.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册