未验证 提交 99177b42 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #14693 from kbinias/fix-for-accuracy-problem-for-inlplace-operators

Fix for accuracy problem for inplace operators when MKL-DNN mode is enabled
......@@ -754,7 +754,7 @@ void OperatorWithKernel::RunImpl(const Scope& scope,
kernel_iter->second(ExecutionContext(*this, exec_scope, *dev_ctx));
if (!transfered_inplace_vars.empty()) {
if (run_by_executor_ && !transfered_inplace_vars.empty()) {
// there is inplace variable has been transfered.
TransferInplaceVarsBack(scope, transfered_inplace_vars, *transfer_scope);
}
......@@ -776,6 +776,7 @@ void OperatorWithKernel::RunImpl(const Scope& scope,
}
}
}
void OperatorWithKernel::TransferInplaceVarsBack(
const Scope& scope, const std::vector<std::string>& inplace_vars,
const Scope& transfer_scope) const {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册