未验证 提交 3c7e94d6 编写于 作者: C cyber-pioneer 提交者: GitHub

[Prim] Add op map (#50673)

* fix flatten op map

* remove prim op all list

* add op map info of full_like

* polish code
上级 457defe7
......@@ -594,6 +594,14 @@
outputs :
out : Out
- op : flatten (flatten_contiguous_range)
inputs :
x : X
outputs :
{out : Out, xshape : XShape}
attrs :
{start_axis : start_axis, stop_axis : stop_axis}
- op : flip
inputs :
x : X
......@@ -648,6 +656,14 @@
extra :
attrs : [bool use_mkldnn = false]
- op : full_like (fill_any_like)
inputs :
x : X
outputs :
out : Out
attrs :
{value: value, dtype: dtype}
- op : fused_conv2d
extra :
attrs : [bool use_cudnn = false, float fuse_alpha = 0.0f, float fuse_beta = 0.0f, float Scale_in = 1.0f,
......
......@@ -59,6 +59,7 @@ from paddle.tensor import zeros # noqa: F401
from paddle.tensor.creation import assign # noqa: F401
from paddle.tensor.manipulation import cast # noqa: F401
"""
math_op = [
'add',
'subtract',
......@@ -116,11 +117,4 @@ others = [
'reshape',
'full',
]
__all__ = []
__all__.extend(math_op)
__all__.extend(trigonometric_op)
__all__.extend(sub_prim)
__all__.extend(others)
__all__.sort()
"""
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册