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