Skip to content

Commit

Permalink
[OpCompat] add and update print, split and sync_batch_norm in op_comp…
Browse files Browse the repository at this point in the history
…at.yaml (PaddlePaddle#55417)
  • Loading branch information
GreatV authored and wz1qqx committed Jul 31, 2023
1 parent faaf014 commit c5eef74
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 @@ -2098,6 +2098,12 @@
extra :
attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32", bool is_test = false]

- op : print
inputs :
in : In
outputs :
out : Out

- op : prior_box
inputs :
{input: Input, image: Image}
Expand Down Expand Up @@ -2532,6 +2538,10 @@
out : Out

- op : split
inputs:
x : X
outputs:
out : Out
int_array:
sections :
data_type : int
Expand Down Expand Up @@ -2668,6 +2678,10 @@
attrs : [bool use_mkldnn = false, float beta = 1.0]

- op : sync_batch_norm
inputs :
{x : X, scale : Scale, bias : Bias, mean : Mean, variance : Variance}
outputs :
{y : Y, mean_out : MeanOut, variance_out : VarianceOut, saved_mean : SavedMean, saved_variance : SavedVariance, reserve_space : ReserveSpace}
backward : sync_batch_norm_grad
extra :
attrs : [bool use_mkldnn = false, bool fuse_with_relu = false]
Expand Down

0 comments on commit c5eef74

Please sign in to comment.