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

musl/fix_failed_unittests_in_musl
cc 5 years ago committed by GitHub
parent b2f7ab6636
commit d1e84f3e9e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -69,6 +69,10 @@ _out_scale_op_list = [
"hard_swish",
"hard_sigmoid",
"conv2d_transpose",
"gru",
"bilinear_interp",
"nearest_interp",
"trilinear_interp",
]
# list op real input and output names, to avoid processing input such as AxisTensor.
@ -114,6 +118,7 @@ _op_real_in_out_name = {
"scale": [["X"], ["Out"]],
"hard_swish": [["X"], ["Out"]],
"hard_sigmoid": [["X"], ["Out"]],
"gru": [["Input", "Weight"], ["Hidden"]],
}
_conv_ops = ['conv2d', 'depthwise_conv2d', 'conv2d_transpose']

Loading…
Cancel
Save