Skip to content

Commit 5cd4bbf

Browse files
authored
cherry-pick from #17935 (#18051)
test=release/1.5 * Fix the GetExpectedKernelType of add_position_encoding_op. * Fix the doc of lstm_unit outputs in nn.py.
1 parent a9da6ea commit 5cd4bbf

File tree

3 files changed

+17
-2
lines changed

3 files changed

+17
-2
lines changed

paddle/fluid/API.spec

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ paddle.fluid.layers.sequence_expand (ArgSpec(args=['x', 'y', 'ref_level', 'name'
104104
paddle.fluid.layers.sequence_expand_as (ArgSpec(args=['x', 'y', 'name'], varargs=None, keywords=None, defaults=(None,)), ('document', '0ecf8d24cf4fd5c035406ee46afccfa0'))
105105
paddle.fluid.layers.sequence_pad (ArgSpec(args=['x', 'pad_value', 'maxlen', 'name'], varargs=None, keywords=None, defaults=(None, None)), ('document', '6a1adf3067b20f6e4bcb354d71c19184'))
106106
paddle.fluid.layers.sequence_unpad (ArgSpec(args=['x', 'length', 'name'], varargs=None, keywords=None, defaults=(None,)), ('document', 'd12803c903c99aa36ec03aaac5f0cc5b'))
107-
paddle.fluid.layers.lstm_unit (ArgSpec(args=['x_t', 'hidden_t_prev', 'cell_t_prev', 'forget_bias', 'param_attr', 'bias_attr', 'name'], varargs=None, keywords=None, defaults=(0.0, None, None, None)), ('document', '409d1c2ca874a22511258175649d2b7f'))
107+
paddle.fluid.layers.lstm_unit (ArgSpec(args=['x_t', 'hidden_t_prev', 'cell_t_prev', 'forget_bias', 'param_attr', 'bias_attr', 'name'], varargs=None, keywords=None, defaults=(0.0, None, None, None)), ('document', 'fe126c58e4339410e875ab1eba246d21'))
108108
paddle.fluid.layers.reduce_sum (ArgSpec(args=['input', 'dim', 'keep_dim', 'name'], varargs=None, keywords=None, defaults=(None, False, None)), ('document', 'dd5f06fb7cf39ca06cbab4abd03e6893'))
109109
paddle.fluid.layers.reduce_mean (ArgSpec(args=['input', 'dim', 'keep_dim', 'name'], varargs=None, keywords=None, defaults=(None, False, None)), ('document', 'a3024789eba11a70c2ef27c358173400'))
110110
paddle.fluid.layers.reduce_max (ArgSpec(args=['input', 'dim', 'keep_dim', 'name'], varargs=None, keywords=None, defaults=(None, False, None)), ('document', '10023caec4d7f78c3b901f023a1feaa7'))

paddle/fluid/operators/add_position_encoding_op.cc

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,13 @@ class AddPositionEncodingOp : public framework::OperatorWithKernel {
3333
ctx->SetOutputDim("Out", x_dims);
3434
ctx->ShareLoD("X", /*->*/ "Out");
3535
}
36+
37+
protected:
38+
framework::OpKernelType GetExpectedKernelType(
39+
const framework::ExecutionContext& ctx) const override {
40+
return framework::OpKernelType(ctx.Input<framework::LoDTensor>("X")->type(),
41+
platform::CPUPlace());
42+
}
3643
};
3744

3845
class AddPositionEncodingOpGrad : public framework::OperatorWithKernel {
@@ -45,6 +52,14 @@ class AddPositionEncodingOpGrad : public framework::OperatorWithKernel {
4552
ctx->SetOutputDim(framework::GradVarName("X"), out_dims);
4653
}
4754
}
55+
56+
protected:
57+
framework::OpKernelType GetExpectedKernelType(
58+
const framework::ExecutionContext& ctx) const override {
59+
return framework::OpKernelType(
60+
ctx.Input<framework::LoDTensor>(framework::GradVarName("Out"))->type(),
61+
platform::CPUPlace());
62+
}
4863
};
4964

5065
class AddPositionEncodingOpMaker : public framework::OpProtoAndCheckerMaker {

python/paddle/fluid/layers/nn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4496,7 +4496,7 @@ def lstm_unit(x_t,
44964496
44974497
i_t = \sigma(L_{i_t})
44984498
4499-
This layer has two outputs including :math:`h_t` and :math:`o_t`.
4499+
This layer has two outputs including :math:`h_t` and :math:`c_t`.
45004500
45014501
Args:
45024502
x_t (Variable): The input value of current step, a 2-D tensor with shape

0 commit comments

Comments
 (0)