From c557402855dca954aa65b20827d27f258d9106b2 Mon Sep 17 00:00:00 2001 From: zchen0211 Date: Thu, 14 Sep 2017 14:51:45 -0700 Subject: [PATCH] cond_op modify --- paddle/operators/cond_op.h | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/paddle/operators/cond_op.h b/paddle/operators/cond_op.h index 001096d31a..7eeec84996 100644 --- a/paddle/operators/cond_op.h +++ b/paddle/operators/cond_op.h @@ -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&& net) { sub_net_op_[0] = std::move(net); } - // Set False Block + /* + * Set False Block + */ void set_falsenet(std::unique_ptr&& net) { sub_net_op_[1] = std::move(net); } -- GitLab