|
|
@ -97,7 +97,6 @@ void FusedEmbeddingFCLSTMOp::InferShape(
|
|
|
|
if (ctx->Attrs().Get<bool>("use_seq")) {
|
|
|
|
if (ctx->Attrs().Get<bool>("use_seq")) {
|
|
|
|
xx_width = wh_dims[1];
|
|
|
|
xx_width = wh_dims[1];
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
// xx_width = x_dims[1] > wh_dims[1] ? wh_dims[1] : x_dims[1]; // ?
|
|
|
|
|
|
|
|
xx_width = wh_dims[1]; //
|
|
|
|
xx_width = wh_dims[1]; //
|
|
|
|
PADDLE_ENFORCE(ctx->HasOutput("BatchedInput"),
|
|
|
|
PADDLE_ENFORCE(ctx->HasOutput("BatchedInput"),
|
|
|
|
"Assert only one Output(BatchedInput) of LSTM.");
|
|
|
|
"Assert only one Output(BatchedInput) of LSTM.");
|
|
|
|