From 7eb0457ce24d5ea85cfdd272122e912fbde220bd Mon Sep 17 00:00:00 2001 From: Wang Xin Date: Fri, 14 Jul 2023 14:39:39 +0800 Subject: [PATCH] [OpCompat] add and update print, split and sync_batch_norm in op_compat.yaml (#55417) --- paddle/phi/api/yaml/op_compat.yaml | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/paddle/phi/api/yaml/op_compat.yaml b/paddle/phi/api/yaml/op_compat.yaml index 56eb4eb5c88..0db2774eaaf 100755 --- a/paddle/phi/api/yaml/op_compat.yaml +++ b/paddle/phi/api/yaml/op_compat.yaml @@ -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} @@ -2532,6 +2538,10 @@ out : Out - op : split + inputs: + x : X + outputs: + out : Out int_array: sections : data_type : int @@ -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] -- GitLab