提交 c5574028 编写于 作者: Z zchen0211

cond_op modify

上级 35cc956a
...@@ -61,12 +61,16 @@ class CondOp : public framework::OperatorBase { ...@@ -61,12 +61,16 @@ class CondOp : public framework::OperatorBase {
*/ */
void InferShape(const framework::Scope& scope) const override; void InferShape(const framework::Scope& scope) const override;
// Set True Block /*
* Set True Block
*/
void set_truenet(std::unique_ptr<OperatorBase>&& net) { void set_truenet(std::unique_ptr<OperatorBase>&& net) {
sub_net_op_[0] = std::move(net); sub_net_op_[0] = std::move(net);
} }
// Set False Block /*
* Set False Block
*/
void set_falsenet(std::unique_ptr<OperatorBase>&& net) { void set_falsenet(std::unique_ptr<OperatorBase>&& net) {
sub_net_op_[1] = std::move(net); sub_net_op_[1] = std::move(net);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册