node.h 1.9 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 };
X
better  
Xin Pan 已提交
30 31
  static const char kControlDepVarName[];

X
polish  
Xin Pan 已提交
32 33
  explicit Node(const std::string& name, Type type)
      : name_(name), var_desc_(nullptr), op_desc_(nullptr), type_(type) {}
X
Xin Pan 已提交
34

X
Xin Pan 已提交
35 36 37 38 39
  explicit Node(VarDesc* var_desc)
      : name_(var_desc->Name()),
        var_desc_(var_desc),
        op_desc_(nullptr),
        type_(Type::kVariable) {}
X
tmp  
Xin Pan 已提交
40

X
Xin Pan 已提交
41 42 43 44 45
  explicit Node(OpDesc* op_desc)
      : name_(op_desc->Type()),
        var_desc_(nullptr),
        op_desc_(op_desc),
        type_(Type::kOperation) {}
X
Xin Pan 已提交
46 47 48

  Type NodeType() const { return type_; }

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

X
Xin Pan 已提交
51 52 53 54
  VarDesc* Var() {
    PADDLE_ENFORCE(type_ == Type::kVariable);
    return var_desc_;
  }
X
Xin Pan 已提交
55

X
Xin Pan 已提交
56 57 58
  OpDesc* Op() {
    PADDLE_ENFORCE(type_ == Type::kOperation);
    return op_desc_;
X
tmp  
Xin Pan 已提交
59
  }
60 61 62

  std::vector<Node*> inputs;
  std::vector<Node*> outputs;
X
Xin Pan 已提交
63 64

 protected:
X
Xin Pan 已提交
65
  const std::string name_;
66 67
  VarDesc* var_desc_;
  OpDesc* op_desc_;
X
Xin Pan 已提交
68 69
  Type type_;

X
tmp  
Xin Pan 已提交
70
 private:
X
Xin Pan 已提交
71 72 73
  DISABLE_COPY_AND_ASSIGN(Node);
};

X
tmp  
Xin Pan 已提交
74
}  // namespace ir
X
Xin Pan 已提交
75
}  // namespace framework
X
start  
Xin Pan 已提交
76
}  // namespace paddle