You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
126 lines
3.8 KiB
126 lines
3.8 KiB
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
|
|
|
|
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
|
|
|
|
#include "paddle/framework/framework.pb.h"
|
|
#include "paddle/framework/op_registry.h"
|
|
|
|
namespace paddle {
|
|
namespace operators {
|
|
|
|
/**
|
|
* @brief Network is also a type of Operator
|
|
*
|
|
* It will manage the operators it has.
|
|
*
|
|
* Network is the container and controller of a set of operators.
|
|
|
|
* 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.
|
|
*
|
|
* This is the base class of network, all the networks should implement the APIs
|
|
* it defines.
|
|
*/
|
|
class NetOp : public framework::OperatorBase {
|
|
public:
|
|
static const char kAll[];
|
|
NetOp() : framework::OperatorBase("plain_net", {}, {}, {}) {}
|
|
|
|
NetOp(const std::string& type, const framework::VariableNameMap& inputs,
|
|
const framework::VariableNameMap& outputs,
|
|
const framework::AttributeMap& attrs);
|
|
|
|
NetOp(const NetOp& o) : framework::OperatorBase(o.type_, {}, {}, o.attrs_) {
|
|
this->ops_.reserve(o.ops_.size());
|
|
std::transform(
|
|
o.ops_.begin(), o.ops_.end(), std::back_inserter(this->ops_),
|
|
[](const std::unique_ptr<framework::OperatorBase>& op) {
|
|
return std::unique_ptr<framework::OperatorBase>(op->Clone());
|
|
});
|
|
this->CompleteAddOp();
|
|
}
|
|
|
|
/**
|
|
* @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.
|
|
*/
|
|
void Run(const framework::Scope& scope,
|
|
const platform::DeviceContext& dev_ctx) const override {
|
|
for (auto& op : ops_) {
|
|
op->Run(scope, dev_ctx);
|
|
}
|
|
}
|
|
|
|
bool SupportGPU() const override {
|
|
for (auto& op : ops_) {
|
|
if (!op->SupportGPU()) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void AppendOp(const framework::OperatorBase& op) { AppendOp(op.Clone()); }
|
|
|
|
/**
|
|
* @brief Add an operator by ptr
|
|
*/
|
|
void AppendOp(std::unique_ptr<framework::OperatorBase> op) {
|
|
PADDLE_ENFORCE(!add_op_done_,
|
|
"Cannot AppendOp when this network is sealed");
|
|
PADDLE_ENFORCE_NOT_NULL(op, "Cannot Insert Null op");
|
|
ops_.push_back(std::move(op));
|
|
}
|
|
|
|
void InsertOp(size_t pos, std::unique_ptr<framework::OperatorBase> op) {
|
|
PADDLE_ENFORCE(!add_op_done_,
|
|
"Cannot InsertOp when this network is sealed");
|
|
PADDLE_ENFORCE_NOT_NULL(op, "Cannot Insert Null op");
|
|
PADDLE_ENFORCE_LE(pos, ops_.size(), "Out of range");
|
|
ops_.insert(ops_.begin() + pos, std::move(op));
|
|
}
|
|
|
|
void InsertOp(size_t pos, const framework::OperatorBase& op) {
|
|
InsertOp(pos, op.Clone());
|
|
}
|
|
|
|
void CompleteAddOp(bool calculate = true);
|
|
|
|
std::string DebugString() const override;
|
|
|
|
bool IsNetOp() const override;
|
|
std::vector<std::string> OutputVars(bool has_intermediate) const override;
|
|
|
|
std::unique_ptr<framework::OperatorBase> Clone() const override;
|
|
|
|
std::vector<std::unique_ptr<framework::OperatorBase>> ops_;
|
|
|
|
private:
|
|
bool add_op_done_{false};
|
|
std::set<std::string> intermediate_outputs_;
|
|
|
|
template <typename T, typename KeyType>
|
|
static bool Contains(T container, KeyType key) {
|
|
return container.find(key) != container.end();
|
|
}
|
|
};
|
|
|
|
} // namespace operators
|
|
} // namespace paddle
|