提交 2590527f 编写于 作者: C cyber-pioneer

fix composite mean op map

上级 14a92c8c
...@@ -1138,8 +1138,14 @@ ...@@ -1138,8 +1138,14 @@
extra : extra :
attrs : [bool use_mkldnn = false] attrs : [bool use_mkldnn = false]
- op : reduce_mean - op : mean (reduce_mean)
backward : reduce_mean_grad backward : reduce_mean_grad
inputs :
x : X
outputs :
out : Out
attrs :
{axis : dim, keepdim : keep_dim}
extra : extra :
attrs : [bool use_mkldnn = false] attrs : [bool use_mkldnn = false]
......
...@@ -183,7 +183,8 @@ def _get_args_values(op, phi_name): ...@@ -183,7 +183,8 @@ def _get_args_values(op, phi_name):
and arg_name in op_content["attrs"].keys() and arg_name in op_content["attrs"].keys()
): ):
attrs.append(op.attr(op_content["attrs"][arg_name])) attrs.append(op.attr(op_content["attrs"][arg_name]))
attrs.append(op.attr(arg_name)) else:
attrs.append(op.attr(arg_name))
return inputs, attrs return inputs, attrs
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册