提交 e3d2bca8 编写于 作者: S SunAhong1993

fix

上级 264c0c85
...@@ -83,6 +83,10 @@ def shape_convolution(layer, input_shape): ...@@ -83,6 +83,10 @@ def shape_convolution(layer, input_shape):
return get_strided_kernel_output_shape(params, input_shape[0], math.floor) return get_strided_kernel_output_shape(params, input_shape[0], math.floor)
def shape_depthwiseconvolution(layer, input_shape):
return shape_convolution(layer, input_shape)
def shape_deconvolution(layer, input_shape): def shape_deconvolution(layer, input_shape):
h_i = input_shape[0][2] h_i = input_shape[0][2]
......
...@@ -112,7 +112,7 @@ class OpSet9(): ...@@ -112,7 +112,7 @@ class OpSet9():
dict(alpha='negative_slope'), dict(alpha='negative_slope'),
dict(negative_slope=.01)], dict(negative_slope=.01)],
'Elu': ['paddle.nn.functional.elu', 'Elu': ['paddle.nn.functional.elu',
dict(), dict(alpha='alpha'),
dict(alpha=1.)], dict(alpha=1.)],
'ThresholdedRelu': ['paddle.nn.functional.thresholded_relu', 'ThresholdedRelu': ['paddle.nn.functional.thresholded_relu',
dict(alpha='threshold'), dict(alpha='threshold'),
...@@ -121,11 +121,11 @@ class OpSet9(): ...@@ -121,11 +121,11 @@ class OpSet9():
'Sigmoid': ['paddle.nn.Sigmoid'], 'Sigmoid': ['paddle.nn.Sigmoid'],
'Softsign': ['paddle.nn.Softsign'], 'Softsign': ['paddle.nn.Softsign'],
'Softplus': ['paddle.nn.Softplus', 'Softplus': ['paddle.nn.Softplus',
dict(), dict(threshold='threshold'),
dict(threshold=float(sys.maxsize))], dict(threshold=float(sys.maxsize))],
'Exp': ['paddle.exp'], 'Exp': ['paddle.exp'],
'Softmax': ['paddle.nn.Softmax', 'Softmax': ['paddle.nn.Softmax',
dict(), dict(axis='axis'),
dict(axis=1)], dict(axis=1)],
'Sqrt': ['paddle.sqrt'], 'Sqrt': ['paddle.sqrt'],
'Floor': ['paddle.floor'], 'Floor': ['paddle.floor'],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册