cond_op modify

8 years ago · c557402855
parent 35cc956aac
commit c557402855
1 changed files with 6 additions and 2 deletions
--- 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<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);
  }