net_op.h 3.0 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

17
#include "paddle/framework/op_registry.h"
S
Superjom 已提交
18 19

namespace paddle {
20 21
namespace operators {

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

 * 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.
 *
33
 * This is the base class of network, all the networks should implement the APIs
S
Superjom 已提交
34 35
 * it defines.
 */
36
class NetOp : public framework::OperatorBase {
37
 public:
S
Superjom 已提交
38
  /**
39
   * Infer all the operators' input and output variables' shapes, will be called
S
Superjom 已提交
40 41
   * before every mini-batch
   */
42
  void InferShape(const framework::Scope& scope) const override {
43 44 45 46
    for (auto& op : ops_) {
      op->InferShape(scope);
    }
  }
S
Superjom 已提交
47 48 49 50 51 52 53 54

  /**
   * @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.
   */
55
  void Run(const framework::Scope& scope,
56 57 58 59 60
           const platform::DeviceContext& dev_ctx) const override {
    for (auto& op : ops_) {
      op->Run(scope, dev_ctx);
    }
  }
61 62 63 64 65 66 67 68 69

  bool SupportGPU() const override {
    for (auto& op : ops_) {
      if (!op->SupportGPU()) {
        return false;
      }
    }
    return true;
  }
S
Superjom 已提交
70 71

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

Y
Yu Yang 已提交
80 81 82
  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 已提交
83 84
    PADDLE_ENFORCE_NOT_NULL(op, "Cannot Insert Null op");
    PADDLE_ENFORCE_LE(pos, ops_.size(), "Out of range");
Y
Yu Yang 已提交
85 86 87
    ops_.insert(ops_.begin() + pos, op);
  }

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

90 91
  std::string DebugString() const override;

Y
Yu Yang 已提交
92 93
  bool IsNetOp() const override;

94
  std::vector<std::shared_ptr<OperatorBase>> ops_;
S
Superjom 已提交
95

96
 private:
97 98 99 100 101 102
  bool add_op_done_{false};

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

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