node.h 2.2 KB
Newer Older
X
Xin Pan 已提交
1
/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
X
start  
Xin Pan 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */

#pragma once

X
Xin Pan 已提交
17 18
#include <string>
#include <vector>
19 20
#include "paddle/fluid/framework/op_desc.h"
#include "paddle/fluid/framework/var_desc.h"
X
Xin Pan 已提交
21 22
#include "paddle/fluid/platform/macros.h"

X
start  
Xin Pan 已提交
23
namespace paddle {
X
Xin Pan 已提交
24
namespace framework {
X
tmp  
Xin Pan 已提交
25
namespace ir {
X
Xin Pan 已提交
26 27 28

class Node {
 public:
X
polish  
Xin Pan 已提交
29
  enum class Type { kOperation, kVariable };
W
Wu Yi 已提交
30
  static constexpr char kControlDepVarName[] = "__control_var";
X
better  
Xin Pan 已提交
31

32 33 34 35 36 37
  explicit Node(const std::string& name, Type type, int id = -1)
      : name_(name),
        var_desc_(nullptr),
        op_desc_(nullptr),
        type_(type),
        id_(id) {}
X
Xin Pan 已提交
38

39
  explicit Node(VarDesc* var_desc, int id = -1)
X
Xin Pan 已提交
40
      : name_(var_desc->Name()),
41
        var_desc_(new VarDesc(*var_desc)),
X
Xin Pan 已提交
42
        op_desc_(nullptr),
43 44
        type_(Type::kVariable),
        id_(id) {}
X
tmp  
Xin Pan 已提交
45

46
  explicit Node(OpDesc* op_desc, int id = -1)
X
Xin Pan 已提交
47 48
      : name_(op_desc->Type()),
        var_desc_(nullptr),
X
Xin Pan 已提交
49
        op_desc_(new OpDesc(*op_desc, op_desc->Block())),
50 51
        type_(Type::kOperation),
        id_(id) {}
X
Xin Pan 已提交
52 53 54

  Type NodeType() const { return type_; }

X
Xin Pan 已提交
55
  std::string Name() const { return name_; }
X
tmp  
Xin Pan 已提交
56

X
Xin Pan 已提交
57 58
  VarDesc* Var() {
    PADDLE_ENFORCE(type_ == Type::kVariable);
59
    return var_desc_.get();
X
Xin Pan 已提交
60
  }
X
Xin Pan 已提交
61

62
  OpDesc* Op() const {
63 64
    PADDLE_ENFORCE(IsOp());
    return op_desc_.get();
X
tmp  
Xin Pan 已提交
65
  }
66

67 68
  int id() const { return id_; }

69 70 71
  bool IsOp() const { return type_ == Type::kOperation; }
  bool IsVar() const { return type_ == Type::kVariable; }

72 73
  std::vector<Node*> inputs;
  std::vector<Node*> outputs;
X
Xin Pan 已提交
74 75

 protected:
X
Xin Pan 已提交
76
  const std::string name_;
77 78
  std::unique_ptr<VarDesc> var_desc_;
  std::unique_ptr<OpDesc> op_desc_;
X
Xin Pan 已提交
79
  Type type_;
80
  int id_;
X
Xin Pan 已提交
81

X
tmp  
Xin Pan 已提交
82
 private:
X
Xin Pan 已提交
83 84 85
  DISABLE_COPY_AND_ASSIGN(Node);
};

X
tmp  
Xin Pan 已提交
86
}  // namespace ir
X
Xin Pan 已提交
87
}  // namespace framework
X
start  
Xin Pan 已提交
88
}  // namespace paddle