fix crf grad lod share

test=develop
6 years ago · be425461a1
parent e1679b8847
commit be425461a1
1 changed files with 2 additions and 0 deletions
--- a/paddle/fluid/operators/linear_chain_crf_op.cc
+++ b/paddle/fluid/operators/linear_chain_crf_op.cc
@ -230,10 +230,12 @@ class LinearChainCRFGradOp : public framework::OperatorWithKernel {

    if (ctx->HasOutput(framework::GradVarName("Emission"))) {
      ctx->SetOutputDim(framework::GradVarName("Emission"), emission_exps_dims);
+      ctx->ShareLoD("Emission", framework::GradVarName("Emission"));
    }
    if (ctx->HasOutput(framework::GradVarName("Transition"))) {
      ctx->SetOutputDim(framework::GradVarName("Transition"),
                        transition_exps_dims);
+      ctx->ShareLoD("Transition", framework::GradVarName("Transition"));
    }
  }