Skip to content

Commit 9f15d88

Browse files
committed
- Cleanup as suggessted by reviewers
test=develop
1 parent ae8b471 commit 9f15d88

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

paddle/fluid/operators/fused_embedding_fc_lstm_op.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,6 @@ void FusedEmbeddingFCLSTMOp::InferShape(
9393
ctx->SetOutputDim("Cell", out_dims);
9494
ctx->ShareLoD("Ids", "Hidden");
9595
ctx->ShareLoD("Ids", "Cell");
96-
int xx_width = wh_dims[1];
9796
if (!ctx->Attrs().Get<bool>("use_seq")) {
9897
PADDLE_ENFORCE(ctx->HasOutput("BatchedInput"),
9998
"Assert only one Output(BatchedInput) of LSTM.");
@@ -109,7 +108,7 @@ void FusedEmbeddingFCLSTMOp::InferShape(
109108
ctx->SetOutputDim("BatchedHidden", out_dims);
110109
ctx->SetOutputDim("BatchedCell", out_dims);
111110
}
112-
ctx->SetOutputDim("XX", {x_dims[0], xx_width});
111+
ctx->SetOutputDim("XX", {x_dims[0], wh_dims[1]});
113112
ctx->ShareLoD("Ids", "XX");
114113
}
115114

0 commit comments

Comments
 (0)