未验证 提交 d1e84f3e 编写于 作者: C cc 提交者: GitHub

Add some ops for cacluating output scale, test=develop (#28644)

上级 b2f7ab66
...@@ -69,6 +69,10 @@ _out_scale_op_list = [ ...@@ -69,6 +69,10 @@ _out_scale_op_list = [
"hard_swish", "hard_swish",
"hard_sigmoid", "hard_sigmoid",
"conv2d_transpose", "conv2d_transpose",
"gru",
"bilinear_interp",
"nearest_interp",
"trilinear_interp",
] ]
# list op real input and output names, to avoid processing input such as AxisTensor. # list op real input and output names, to avoid processing input such as AxisTensor.
...@@ -114,6 +118,7 @@ _op_real_in_out_name = { ...@@ -114,6 +118,7 @@ _op_real_in_out_name = {
"scale": [["X"], ["Out"]], "scale": [["X"], ["Out"]],
"hard_swish": [["X"], ["Out"]], "hard_swish": [["X"], ["Out"]],
"hard_sigmoid": [["X"], ["Out"]], "hard_sigmoid": [["X"], ["Out"]],
"gru": [["Input", "Weight"], ["Hidden"]],
} }
_conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose'] _conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册