diff --git a/paddle/operators/sequence_slice_op.cc b/paddle/operators/sequence_slice_op.cc index 2fae37e0bf643d59b458cd2bae67a53537e62ad5..990053b8af9f6bfb982eaf994374daa45ce6cbfb 100755 --- a/paddle/operators/sequence_slice_op.cc +++ b/paddle/operators/sequence_slice_op.cc @@ -35,11 +35,14 @@ class SequenceSliceOp : public framework::OperatorWithKernel { auto offset_dim = ctx->GetInputDim("Offset"); auto length_dim = ctx->GetInputDim("Length"); - PADDLE_ENFORCE_EQ(offset_dim.size(), 2UL, - "Only support one level sequence now."); - PADDLE_ENFORCE_EQ(length_dim.size(), 2UL, - "Only support one level sequence now."); - + PADDLE_ENFORCE_EQ( + offset_dim.size(), 2UL, + "Only support one level sequence now, The rank of offset must be 2."); + PADDLE_ENFORCE_EQ( + length_dim.size(), 2UL, + "Only support one level sequence now, The rank of Length must be 2."); + + // Initialize the output's dims to maximum ctx->SetOutputDim("Out", input_dims); } diff --git a/paddle/operators/sequence_slice_op.h b/paddle/operators/sequence_slice_op.h index cbb950b152a69d6d74dd32fd5afc02ee38603f1c..c7d7ef4916756397c3d5911d7ec25305efc815c0 100755 --- a/paddle/operators/sequence_slice_op.h +++ b/paddle/operators/sequence_slice_op.h @@ -54,10 +54,10 @@ class SequenceSliceOpKernel : public framework::OpKernel { PADDLE_ENFORCE_EQ(lod.size(), 1UL, "Only support one level sequence now."); PADDLE_ENFORCE_EQ( - n, length->dims()[0], + n, static_cast(length->dims()[0]), "The size of input-sequence and length-array should be the same") PADDLE_ENFORCE_EQ( - n, offset->dims()[0], + n, static_cast(offset->dims()[0]), "The size of input-sequence and offset-array should be the same") const int64_t* offset_data = offset->data();