add details. test=develop

revert-15661-fix-cpu-broadcast
dzhwinter 6 years ago
parent 488719bac7
commit 94dd50c33f

@ -264,6 +264,20 @@ class ElementwiseOpInplace : public framework::InplaceInToOut {
}
};
class ElementwiseGradOpInplace : public framework::InplaceInToOut {
public:
using framework::InplaceInToOut::InplaceInToOut;
protected:
std::unordered_map<std::string, std::string> Apply(
const framework::OpDesc &op_desc,
framework::BlockDesc *block) const override {
return std::unordered_map<std::string, std::string>{
{framework::GradVarName("Out"), framework::GradVarName("X")},
};
}
};
} // namespace operators
} // namespace paddle
@ -316,4 +330,5 @@ class ElementwiseOpInplace : public framework::InplaceInToOut {
op_type##GradMaker, \
::paddle::operators::ElementwiseOpInplace); \
REGISTER_OPERATOR(op_type##_grad, \
::paddle::operators::ElementwiseOpExplicitGrad)
::paddle::operators::ElementwiseOpExplicitGrad, \
::paddle::operators::ElementwiseGradOpInplace)

Loading…
Cancel
Save