Skip to content

Commit

Permalink
[OpCompat] add cast and repeat_interleave in op_compat.yaml (PaddlePa…
Browse files Browse the repository at this point in the history
…ddle#55467)

* add cast and repeat_interleave

* fix
  • Loading branch information
gouzil authored and wz1qqx committed Jul 31, 2023
1 parent ec5a8c4 commit 4bf4c89
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions paddle/phi/api/yaml/op_compat.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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 :
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit 4bf4c89

Please sign in to comment.