Merge pull request #14400 from jacquesqiao/fix-prelu

fix prelu
panyx0718-patch-1
Qiao Longfei 6 years ago committed by GitHub
commit f8d5775564
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -6822,7 +6822,7 @@ def prelu(x, mode, param_attr=None, name=None):
alpha_shape = x.shape
dtype = helper.input_dtype(input_param_name='x')
alpha = helper.create_parameter(
attr=param_attr,
attr=helper.param_attr,
shape=alpha_shape,
dtype='float32',
is_bias=False,

Loading…
Cancel
Save