提交 c5574028 编写于 作者: Z zchen0211

cond_op modify

上级 35cc956a
......@@ -61,12 +61,16 @@ class CondOp : public framework::OperatorBase {
*/
void InferShape(const framework::Scope& scope) const override;
// Set True Block
/*
* Set True Block
*/
void set_truenet(std::unique_ptr<OperatorBase>&& net) {
sub_net_op_[0] = std::move(net);
}
// Set False Block
/*
* Set False Block
*/
void set_falsenet(std::unique_ptr<OperatorBase>&& 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.
先完成此消息的编辑!
想要评论请 注册