提交 4309c316 编写于 作者: W wjj19950828

Support onnx roberta

上级 448fb19d
...@@ -457,6 +457,18 @@ class OpSet9(): ...@@ -457,6 +457,18 @@ class OpSet9():
outputs=[node.name], outputs=[node.name],
**attrs) **attrs)
@print_mapping_info
def CumSum(self, node):
val_x = self.graph.get_input_node(node, idx=0, copy=True)
axis = self.graph.get_input_node(node, idx=1, copy=True)
axis_values = _const_weight_or_none(axis)
layer_attrs = {'axis': axis_values}
self.paddle_graph.add_layer(
'paddle.cumsum',
inputs={"x": val_x.name},
outputs=[node.name],
**layer_attrs)
@print_mapping_info @print_mapping_info
def HardSigmoid(self, node): def HardSigmoid(self, node):
val_x = self.graph.get_input_node(node, idx=0, copy=True) val_x = self.graph.get_input_node(node, idx=0, copy=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册