Skip to content

Commit 2582fec

Browse files
committed
fix dtype
1 parent 215cdaf commit 2582fec

File tree

2 files changed

+19
-20
lines changed

2 files changed

+19
-20
lines changed

paddle/fluid/API.spec

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -267,22 +267,22 @@ paddle.fluid.layers.pow ArgSpec(args=[], varargs='args', keywords='kwargs', defa
267267
paddle.fluid.layers.stanh ArgSpec(args=[], varargs='args', keywords='kwargs', defaults=None)
268268
paddle.fluid.layers.hard_sigmoid ArgSpec(args=[], varargs='args', keywords='kwargs', defaults=None)
269269
paddle.fluid.layers.swish ArgSpec(args=[], varargs='args', keywords='kwargs', defaults=None)
270-
paddle.fluid.layers.sigmoid ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
271-
paddle.fluid.layers.logsigmoid ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
272-
paddle.fluid.layers.exp ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
273-
paddle.fluid.layers.tanh ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
274-
paddle.fluid.layers.tanh_shrink ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
275-
paddle.fluid.layers.sqrt ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
276-
paddle.fluid.layers.abs ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
277-
paddle.fluid.layers.ceil ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
278-
paddle.fluid.layers.floor ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
279-
paddle.fluid.layers.cos ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
280-
paddle.fluid.layers.sin ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
281-
paddle.fluid.layers.round ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
282-
paddle.fluid.layers.reciprocal ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
283-
paddle.fluid.layers.square ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
284-
paddle.fluid.layers.softplus ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
285-
paddle.fluid.layers.softsign ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)
270+
paddle.fluid.layers.sigmoid ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
271+
paddle.fluid.layers.logsigmoid ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
272+
paddle.fluid.layers.exp ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
273+
paddle.fluid.layers.tanh ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
274+
paddle.fluid.layers.tanh_shrink ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
275+
paddle.fluid.layers.sqrt ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
276+
paddle.fluid.layers.abs ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
277+
paddle.fluid.layers.ceil ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
278+
paddle.fluid.layers.floor ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
279+
paddle.fluid.layers.cos ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
280+
paddle.fluid.layers.sin ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
281+
paddle.fluid.layers.round ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
282+
paddle.fluid.layers.reciprocal ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
283+
paddle.fluid.layers.square ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
284+
paddle.fluid.layers.softplus ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
285+
paddle.fluid.layers.softsign ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=None)
286286
paddle.fluid.layers.uniform_random ArgSpec(args=['shape', 'dtype', 'min', 'max', 'seed'], varargs=None, keywords=None, defaults=(None, None, None, None))
287287
paddle.fluid.layers.hard_shrink ArgSpec(args=['x', 'threshold'], varargs=None, keywords=None, defaults=(None,))
288288
paddle.fluid.layers.cumsum ArgSpec(args=['x', 'axis', 'exclusive', 'reverse'], varargs=None, keywords=None, defaults=(None, None, None))

python/paddle/fluid/layers/layer_function_generator.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -220,11 +220,10 @@ def generate_layer_fn_noattr(op_type):
220220
"""
221221
op_proto = OpProtoHolder.instance().get_op_proto(op_type)
222222

223-
def func(x):
223+
def func(x, name=None):
224224
helper = LayerHelper(op_type, **locals())
225-
output = helper.create_tmp_variable(dtype=helper.input_dtype())
226-
helper.append_op(
227-
type=op_type, inputs={"X": [x]}, outputs={"Out": [output]})
225+
output = helper.create_tmp_variable(dtype=x.dtype)
226+
helper.append_op(type=op_type, inputs={"X": x}, outputs={"Out": output})
228227
return output
229228

230229
func.__name__ = op_type

0 commit comments

Comments
 (0)