operator.cc 5.1 KB
Newer Older
Q
Qiao Longfei 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/* 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. */

#include "paddle/framework/operator.h"
16 17
#include <algorithm>
#include "paddle/framework/op_registry.h"
Q
Qiao Longfei 已提交
18 19 20 21

namespace paddle {
namespace framework {

Q
qijun 已提交
22
template <>
23
Eigen::DefaultDevice& ExecutionContext::GetEigenDevice<
Q
qijun 已提交
24
    platform::CPUPlace, Eigen::DefaultDevice>() const {
D
dongzhihong 已提交
25
  return *device_context_->get_eigen_device<Eigen::DefaultDevice>();
Q
qijun 已提交
26 27 28 29
}

#ifndef PADDLE_ONLY_CPU
template <>
30
Eigen::GpuDevice&
31
ExecutionContext::GetEigenDevice<platform::GPUPlace, Eigen::GpuDevice>() const {
D
dongzhihong 已提交
32
  return *device_context_->get_eigen_device<Eigen::GpuDevice>();
Q
qijun 已提交
33 34 35
}
#endif

Y
Yan Chunwei 已提交
36
const std::string& OperatorBase::Input(const std::string& name) const {
Y
Yu Yang 已提交
37 38
  auto& ins = Inputs(name);
  PADDLE_ENFORCE_EQ(ins.size(), 1UL,
Y
Yu Yang 已提交
39 40
                    "Op %s input %s should contain only one variable", type_,
                    name);
Y
Yu Yang 已提交
41
  return ins[0];
Y
Yan Chunwei 已提交
42 43
}

Y
Yu Yang 已提交
44 45
const std::vector<std::string>& OperatorBase::Inputs(
    const std::string& name) const {
Y
Yu Yang 已提交
46 47 48 49
  auto it = inputs_.find(name);
  PADDLE_ENFORCE(it != inputs_.end(), "Op %s do not have input %s", type_,
                 name);
  return it->second;
Y
Yan Chunwei 已提交
50 51 52
}

const std::string& OperatorBase::Output(const std::string& name) const {
Y
Yu Yang 已提交
53 54 55
  auto& outs = Outputs(name);
  PADDLE_ENFORCE_EQ(outs.size(), 1UL,
                    "Op %s output %s should contain only one variable", type_,
Y
Yu Yang 已提交
56
                    name);
Y
Yu Yang 已提交
57
  return outs[0];
Y
Yan Chunwei 已提交
58 59
}

Y
Yu Yang 已提交
60 61
const std::vector<std::string>& OperatorBase::Outputs(
    const std::string& name) const {
Y
Yu Yang 已提交
62 63 64 65
  auto it = outputs_.find(name);
  PADDLE_ENFORCE(it != outputs_.end(), "Op %s does not have output %s", type_,
                 name);
  return it->second;
Y
Yan Chunwei 已提交
66 67
}

Q
Qiao Longfei 已提交
68 69
std::string OperatorBase::DebugString() const {
  std::stringstream ss;
Y
Yu Yang 已提交
70
  ss << "Op(" << type_ << "), inputs:{";
Y
Yu Yang 已提交
71 72
  for (auto it = inputs_.begin(); it != inputs_.end();) {
    auto& input = *it;
Y
Yu Yang 已提交
73 74 75 76 77 78
    ss << input.first << "[";
    for (size_t i = 0; i < input.second.size(); ++i) {
      ss << input.second[i];
      if (i != input.second.size() - 1) {
        ss << ", ";
      }
79
    }
Y
Yu Yang 已提交
80
    ss << "]";
Y
Yu Yang 已提交
81 82
    ++it;
    if (it != inputs_.end()) {
83 84
      ss << ", ";
    }
Q
Qiao Longfei 已提交
85
  }
Y
Yu Yang 已提交
86
  ss << "}, outputs:{";
Y
Yu Yang 已提交
87 88
  for (auto it = outputs_.begin(); it != outputs_.end();) {
    auto& output = *it;
Y
Yu Yang 已提交
89 90 91 92 93 94
    ss << output.first << "[";
    for (size_t i = 0; i < output.second.size(); ++i) {
      ss << output.second[i];
      if (i != output.second.size() - 1) {
        ss << ", ";
      }
95
    }
Y
Yu Yang 已提交
96
    ss << "]";
Y
Yu Yang 已提交
97 98
    ++it;
    if (it != outputs_.end()) {
99 100
      ss << ", ";
    }
Q
Qiao Longfei 已提交
101
  }
Y
Yu Yang 已提交
102
  ss << "}.";
Q
Qiao Longfei 已提交
103 104 105
  return ss.str();
}

D
dongzhihong 已提交
106 107
void OperatorBase::Rename(const std::string& old_name,
                          const std::string& new_name) {
Y
Yu Yang 已提交
108 109 110 111 112 113 114
  for (auto& input : inputs_) {
    std::replace(input.second.begin(), input.second.end(), old_name, new_name);
  }
  for (auto& output : outputs_) {
    std::replace(output.second.begin(), output.second.end(), old_name,
                 new_name);
  }
D
dongzhihong 已提交
115 116
}

Y
Yu Yang 已提交
117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
OperatorBase::OperatorBase(const std::string& type,
                           const OperatorBase::VarNameMap& inputs,
                           const OperatorBase::VarNameMap& outputs,
                           const AttributeMap& attrs)
    : type_(type), inputs_(inputs), outputs_(outputs), attrs_(attrs) {
  static std::atomic<size_t> gUniqId(0UL);
  for (auto& output : outputs_) {
    for (auto& output_name : output.second) {
      if (output_name == kTempVarName) {
        output_name += type_;
        output_name += "@";
        output_name += std::to_string(gUniqId.fetch_add(1));
      }
    }
  }
}
133

Y
Yu Yang 已提交
134 135 136 137 138 139 140 141 142 143
std::vector<std::string> OperatorBase::OutputVars(bool has_intermediate) const {
  std::vector<std::string> ret_val;
  if (has_intermediate) {
    // push all outputs into ret_val
    for (auto& o : outputs_) {
      ret_val.reserve(ret_val.size() + o.second.size());
      ret_val.insert(ret_val.end(), o.second.begin(), o.second.end());
    }
    return ret_val;
  }
144
  auto it = OpRegistry::op_info_map().find(type_);
Y
Yu Yang 已提交
145
  PADDLE_ENFORCE(
146
      it != OpRegistry::op_info_map().end(),
Y
Yu Yang 已提交
147 148
      "Operator %s not registered, cannot figure out intermediate outputs",
      type_);
149 150 151 152
  PADDLE_ENFORCE(
      it->second.proto_ != nullptr,
      "Operator %s has no OpProto, cannot figure out intermediate outputs",
      type_);
Y
Yu Yang 已提交
153 154

  // get all OpProto::Var for outputs
155
  for (auto& o : it->second.proto_.outputs()) {
Y
Yu Yang 已提交
156 157 158 159 160 161 162 163 164
    // ignore all intermediate output
    if (o.intermediate()) continue;
    auto out = outputs_.find(o.name());
    if (out != outputs_.end()) {
      ret_val.reserve(ret_val.size() + out->second.size());
      ret_val.insert(ret_val.end(), out->second.begin(), out->second.end());
    }
  }
  return ret_val;
D
dongzhihong 已提交
165 166
}

Q
Qiao Longfei 已提交
167
}  // namespace framework
L
liaogang 已提交
168
}  // namespace paddle