未验证 提交 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 @@ ...@@ -594,6 +594,14 @@
outputs : outputs :
out : Out 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 - op : flip
inputs : inputs :
x : X x : X
...@@ -648,6 +656,14 @@ ...@@ -648,6 +656,14 @@
extra : extra :
attrs : [bool use_mkldnn = false] 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 - op : fused_conv2d
extra : extra :
attrs : [bool use_cudnn = false, float fuse_alpha = 0.0f, float fuse_beta = 0.0f, float Scale_in = 1.0f, 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 ...@@ -59,6 +59,7 @@ from paddle.tensor import zeros # noqa: F401
from paddle.tensor.creation import assign # noqa: F401 from paddle.tensor.creation import assign # noqa: F401
from paddle.tensor.manipulation import cast # noqa: F401 from paddle.tensor.manipulation import cast # noqa: F401
"""
math_op = [ math_op = [
'add', 'add',
'subtract', 'subtract',
...@@ -116,11 +117,4 @@ others = [ ...@@ -116,11 +117,4 @@ others = [
'reshape', 'reshape',
'full', '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.
先完成此消息的编辑!
想要评论请 注册