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

Support onnx roberta

上级 448fb19d
......@@ -457,6 +457,18 @@ class OpSet9():
outputs=[node.name],
**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
def HardSigmoid(self, node):
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.
先完成此消息的编辑!
想要评论请 注册