cond_op modify

c5574028 · zchen0211 · 35cc956a · c5574028
显示空白变更内容
内联并排

Showing with 6 addition and 2 deletion

paddle/operators/cond_op.h paddle/operators/cond_op.h +6 -2

未找到文件。
--- 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);
  }