We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ae8b471 commit 9f15d88Copy full SHA for 9f15d88
paddle/fluid/operators/fused_embedding_fc_lstm_op.cc
@@ -93,7 +93,6 @@ void FusedEmbeddingFCLSTMOp::InferShape(
93
ctx->SetOutputDim("Cell", out_dims);
94
ctx->ShareLoD("Ids", "Hidden");
95
ctx->ShareLoD("Ids", "Cell");
96
- int xx_width = wh_dims[1];
97
if (!ctx->Attrs().Get<bool>("use_seq")) {
98
PADDLE_ENFORCE(ctx->HasOutput("BatchedInput"),
99
"Assert only one Output(BatchedInput) of LSTM.");
@@ -109,7 +108,7 @@ void FusedEmbeddingFCLSTMOp::InferShape(
109
108
ctx->SetOutputDim("BatchedHidden", out_dims);
110
ctx->SetOutputDim("BatchedCell", out_dims);
111
}
112
- ctx->SetOutputDim("XX", {x_dims[0], xx_width});
+ ctx->SetOutputDim("XX", {x_dims[0], wh_dims[1]});
113
ctx->ShareLoD("Ids", "XX");
114
115
0 commit comments