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

fix composite mean op map

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