net_op.h 3.3 KB
Newer Older
S
Superjom 已提交
1 2
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

3 4 5
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
S
Superjom 已提交
6

7
    http://www.apache.org/licenses/LICENSE-2.0
S
Superjom 已提交
8

9 10 11 12 13
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. */
S
Superjom 已提交
14 15 16

#pragma once

Y
Yu Yang 已提交
17
#include "paddle/framework/framework.pb.h"
18
#include "paddle/framework/op_registry.h"
S
Superjom 已提交
19 20

namespace paddle {
21 22
namespace operators {

S
Superjom 已提交
23
/**
24 25 26
 * @brief Network is also a type of Operator
 *
 * It will manage the operators it has.
S
Superjom 已提交
27
 *
28
 * Network is the container and controller of a set of operators.
S
Superjom 已提交
29 30 31 32 33

 * A network object knows all Operators belonging to this network. Variables,
 * which are inputs and outputs of these operators, are created and managed by a
 * hierarchy of Scope objects.
 *
34
 * This is the base class of network, all the networks should implement the APIs
S
Superjom 已提交
35 36
 * it defines.
 */
37
class NetOp : public framework::OperatorBase {
38
 public:
39
  static const char kAll[];
40 41 42
  NetOp() : framework::OperatorBase("plain_net", {}, {}, {}) {}
  NetOp(const std::string& type, const VarNameMap& inputs,
        const VarNameMap& outputs, const framework::AttributeMap& attrs);
43

S
Superjom 已提交
44
  /**
45
   * Infer all the operators' input and output variables' shapes, will be called
S
Superjom 已提交
46 47
   * before every mini-batch
   */
48
  void InferShape(const framework::Scope& scope) const override {
49 50 51 52
    for (auto& op : ops_) {
      op->InferShape(scope);
    }
  }
S
Superjom 已提交
53 54 55 56 57 58 59 60

  /**
   * @brief Run the network.
   *
   * Run all the operators with the `scope`, if no scope is provided, default
   * scope will be used instead. If no OpContext is provicded, default context
   * will be used.
   */
61
  void Run(const framework::Scope& scope,
62 63 64 65 66
           const platform::DeviceContext& dev_ctx) const override {
    for (auto& op : ops_) {
      op->Run(scope, dev_ctx);
    }
  }
S
Superjom 已提交
67

68 69 70 71 72 73 74 75
  bool SupportGPU() const override {
    for (auto& op : ops_) {
      if (!op->SupportGPU()) {
        return false;
      }
    }
    return true;
  }
S
Superjom 已提交
76 77

  /**
78
   * @brief Add an operator by ptr
S
Superjom 已提交
79
   */
Y
Yu Yang 已提交
80
  void AddOp(const std::shared_ptr<OperatorBase>& op) {
81
    PADDLE_ENFORCE(!add_op_done_, "Cannot AddOp when this network is sealed");
Y
Yan Chunwei 已提交
82
    PADDLE_ENFORCE_NOT_NULL(op, "Cannot Insert Null op");
83 84
    ops_.push_back(op);
  }
S
Superjom 已提交
85

Y
Yu Yang 已提交
86 87 88
  void InsertOp(size_t pos, const std::shared_ptr<OperatorBase>& op) {
    PADDLE_ENFORCE(!add_op_done_,
                   "Cannot InsertOp when this network is sealed");
Y
Yan Chunwei 已提交
89 90
    PADDLE_ENFORCE_NOT_NULL(op, "Cannot Insert Null op");
    PADDLE_ENFORCE_LE(pos, ops_.size(), "Out of range");
Y
Yu Yang 已提交
91 92 93
    ops_.insert(ops_.begin() + pos, op);
  }

Y
Yu Yang 已提交
94
  void CompleteAddOp(bool calculate = true);
S
Superjom 已提交
95

96 97
  std::string DebugString() const override;

Y
Yu Yang 已提交
98
  bool IsNetOp() const override;
99
  std::vector<std::string> OutputVars(bool has_intermediate) const override;
Y
Yu Yang 已提交
100

101
  std::vector<std::shared_ptr<OperatorBase>> ops_;
S
Superjom 已提交
102

103
 private:
104
  bool add_op_done_{false};
105
  std::set<std::string> intermediate_outputs_;
106 107 108 109 110

  template <typename T, typename KeyType>
  static bool Contains(T container, KeyType key) {
    return container.find(key) != container.end();
  }
S
Superjom 已提交
111 112
};

113
}  // namespace operators
S
Superjom 已提交
114
}  // namespace paddle
新手
引导
客服 返回
顶部