未验证 提交 922d2481 编写于 作者: G gouzil 提交者: GitHub

[OpCompat] add cast and repeat_interleave in op_compat.yaml (#55467)

* add cast and repeat_interleave

* fix
上级 7bd50187
......@@ -420,6 +420,12 @@
out : Out
drop_empty_grad : [input_grad]
- op : cast
inputs :
x : X
outputs :
out : Out
- op : ceil
backward : ceil_grad
inputs :
......@@ -2260,6 +2266,14 @@
extra :
attrs : [bool use_mkldnn = false, bool use_cudnn = false]
- op : repeat_interleave
inputs :
x : X
outputs :
out : Out
attrs :
repeats : Repeats
- op : reshape (reshape2)
backward : reshape_grad (reshape2_grad)
inputs:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册