node.h 2.0 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 };
30
  static constexpr char kControlDepVarName[] = "__control_var";
X
better  
Xin Pan 已提交
31

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_;
  }
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 63
  bool IsOp() const { return type_ == Type::kOperation; }
  bool IsVar() const { return type_ == Type::kVariable; }

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

 protected:
X
Xin Pan 已提交
68
  const std::string name_;
69 70
  VarDesc* var_desc_;
  OpDesc* op_desc_;
X
Xin Pan 已提交
71 72
  Type type_;

X
tmp  
Xin Pan 已提交
73
 private:
X
Xin Pan 已提交
74 75 76
  DISABLE_COPY_AND_ASSIGN(Node);
};

X
tmp  
Xin Pan 已提交
77
}  // namespace ir
X
Xin Pan 已提交
78
}  // namespace framework
X
start  
Xin Pan 已提交
79
}  // namespace paddle
新手
引导
客服 返回
顶部