未验证 提交 f66a705f 编写于 作者: R RedContritio 提交者: GitHub

update transpose in op_compat.yaml (#55458)

上级 8f8fa38a
...@@ -904,7 +904,7 @@ ...@@ -904,7 +904,7 @@
- op : transpose - op : transpose
args : (Tensor x, int[] perm) args : (Tensor x, int[] perm)
output : Tensor output : Tensor(out)
infer_meta : infer_meta :
func : TransposeInferMeta func : TransposeInferMeta
kernel : kernel :
......
...@@ -2791,6 +2791,10 @@ ...@@ -2791,6 +2791,10 @@
- op : transpose (transpose2) - op : transpose (transpose2)
backward : transpose_grad (transpose2_grad) backward : transpose_grad (transpose2_grad)
inputs :
x : X
outputs :
out : Out
attrs: attrs:
perm : axis perm : axis
extra : extra :
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册