提交 3671c286 编写于 作者: C chengtbf 提交者: Will Zhang

add op activation improve for in/out diff (#548)



Former-commit-id: 7a8ce23c
上级 a87953f0
......@@ -11,6 +11,8 @@ class CloneOp final : public Operator {
CloneOp() = default;
~CloneOp() = default;
bool NeedExtraInDiffMemWhenBackward() const override { return false; }
bool NeedOutWhenBackward() const override { return false; }
void InitFromOpConf() override;
const PbMessage& GetSpecialConf() const override;
void InferBlobDescs(
......
......@@ -11,6 +11,8 @@ class ConcatOp final : public Operator {
ConcatOp() = default;
~ConcatOp() = default;
bool NeedExtraInDiffMemWhenBackward() const override { return false; }
bool NeedOutWhenBackward() const override { return false; }
void InitFromOpConf() override;
const PbMessage& GetSpecialConf() const override;
......
......@@ -11,6 +11,7 @@ class SoftmaxOp final : public Operator {
SoftmaxOp() = default;
~SoftmaxOp() = default;
bool NeedExtraInDiffMemWhenBackward() const override { return false; }
void InitFromOpConf() override;
const PbMessage& GetSpecialConf() const override;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册