Merge pull request #15179 from sneaxiy/fix_crf_grad_lod

Fix crf grad lod share
revert-15207-remove_op_handle_lock_and_fix_var
Zeng Jinle 6 years ago committed by GitHub
commit 99e6e8b00f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -230,10 +230,12 @@ class LinearChainCRFGradOp : public framework::OperatorWithKernel {
if (ctx->HasOutput(framework::GradVarName("Emission"))) { if (ctx->HasOutput(framework::GradVarName("Emission"))) {
ctx->SetOutputDim(framework::GradVarName("Emission"), emission_exps_dims); ctx->SetOutputDim(framework::GradVarName("Emission"), emission_exps_dims);
ctx->ShareLoD("Emission", framework::GradVarName("Emission"));
} }
if (ctx->HasOutput(framework::GradVarName("Transition"))) { if (ctx->HasOutput(framework::GradVarName("Transition"))) {
ctx->SetOutputDim(framework::GradVarName("Transition"), ctx->SetOutputDim(framework::GradVarName("Transition"),
transition_exps_dims); transition_exps_dims);
ctx->ShareLoD("Transition", framework::GradVarName("Transition"));
} }
} }

Loading…
Cancel
Save