未验证 提交 7eb0457c 编写于 作者: W Wang Xin 提交者: GitHub

[OpCompat] add and update print, split and sync_batch_norm in op_compat.yaml (#55417)

上级 73e441f9
......@@ -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]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册