operator.cpp 5.2 KB
Newer Older
Z
zhaojiaying01 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.

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 "framework/operator.h"
J
jameswu2014 已提交
16
#include <memory>
17
#include "operators/op_param.h"
朔-望's avatar
朔-望 已提交
18
namespace paddle_mobile {
朔-望's avatar
朔-望 已提交
19
namespace framework {
朔-望's avatar
朔-望 已提交
20

21 22 23 24 25
template <typename Dtype>
vector<string> OperatorBase<Dtype>::GetOutKeys() const {
  auto it = op_input_output_key.find(type_);
  if (it == op_input_output_key.end()) {
    DLOG << type_ << " has no outputs";
26
    return {};
27 28 29 30
  }
  return it->second.second;
}

31 32 33 34
template <typename Dtype>
vector<string> OperatorBase<Dtype>::GetInputKeys() const {
  auto it = op_input_output_key.find(type_);
  if (it == op_input_output_key.end()) {
35
    DLOG << type_ << " has no inputs";
36 37 38 39 40
    return {};
  }
  return it->second.first;
}

朔-望's avatar
朔-望 已提交
41 42 43 44 45
template <typename Dtype>
OperatorBase<Dtype>::OperatorBase(const std::string &type,
                                  const VariableNameMap &inputs,
                                  const VariableNameMap &outputs,
                                  const AttributeMap &attrs,
46
                                  framework::Scope *scope)
朔-望's avatar
朔-望 已提交
47 48 49 50
    : type_(type),
      inputs_(inputs),
      outputs_(outputs),
      attrs_(attrs),
朔-望's avatar
朔-望 已提交
51
      scope_(scope) {
52
  CheckAllInputOutputSet();
53 54 55
#ifdef PADDLE_MOBILE_FPGA
  InsertTensors();
#endif
朔-望's avatar
朔-望 已提交
56
}
57

朔-望's avatar
朔-望 已提交
58 59
template <typename Dtype>
void OperatorBase<Dtype>::CheckAllInputOutputSet() const {}
W
wangliu 已提交
60

61
template <typename Dtype>
L
liuruilong 已提交
62
void OperatorBase<Dtype>::Run() {
63
  RunImpl();
Y
yangfei 已提交
64 65 66 67 68 69
#ifdef PADDLE_MOBILE_DEBUG
  DLOG << "-------------" << type_ << "----------------------------";
  vector<string> input_keys = GetInputKeys();
  for (const auto key : input_keys) {
    auto var_vec_in = inputs_.at(key);
    for (int i = 0; i < var_vec_in.size(); ++i) {
H
update  
hjchen2 已提交
70 71 72 73
      auto var = this->scope_->FindVar(var_vec_in[i]);
      if (var->IsInitialized() &&
          var->template IsType<framework::LoDTensor>()) {
        const Tensor *tensor = var->template Get<framework::LoDTensor>();
Y
yangfei 已提交
74 75 76 77 78 79 80
        if (tensor) DLOG << type_ << " input- " << key << "=" << *tensor;
      }
    }
  }
  for (const auto key : GetOutKeys()) {
    auto var_vec_out = outputs_.at(key);
    for (int i = 0; i < var_vec_out.size(); ++i) {
H
update  
hjchen2 已提交
81 82 83 84
      auto var = scope_->FindVar(var_vec_out[i]);
      if (var->IsInitialized() &&
          var->template IsType<framework::LoDTensor>()) {
        const Tensor *tensor = var->template Get<framework::LoDTensor>();
Y
yangfei 已提交
85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
        if (tensor) DLOG << type_ << " output- " << key << "=" << *tensor;
      }
    }
  }
#endif
}

#ifdef PADDLE_MOBILE_CL
template <>
void OperatorBase<GPU_CL>::Run() {
  RunImpl();
#ifdef PADDLE_MOBILE_DEBUG
  DLOG << "-------------" << type_ << "----------------------------";
  vector<string> input_keys = GetInputKeys();
  for (const auto key : input_keys) {
    auto var_vec_in = inputs_.at(key);
    for (int i = 0; i < var_vec_in.size(); ++i) {
      auto vari = scope_->FindVar(var_vec_in[i]);
      if (vari->IsInitialized()) {
        if (type_ == "feed") {
105
          const Tensor *tensor = vari->template Get<framework::LoDTensor>();
Y
yangfei 已提交
106 107
          if (tensor) DLOG << type_ << " input- " << key << "=" << *tensor;
        } else {
108
          const CLImage *cl_image = vari->template Get<framework::CLImage>();
Y
yangfei 已提交
109 110 111 112 113 114 115 116 117 118 119 120 121
          if (cl_image) {
            DLOG << type_ << " input- " << key << "=" << *cl_image;
          }
        }
      }
    }
  }
  for (const auto key : GetOutKeys()) {
    auto var_vec_out = outputs_.at(key);
    for (int i = 0; i < var_vec_out.size(); ++i) {
      auto vari = scope_->FindVar(var_vec_out[i]);
      if (vari->IsInitialized()) {
        if (type_ == "fetch") {
122
          const Tensor *tensor = vari->template Get<framework::LoDTensor>();
Y
yangfei 已提交
123 124 125 126
          if (tensor) {
            DLOG << type_ << " output- " << key << "=" << *tensor;
          }
        } else {
127
          const CLImage *cl_image = vari->template Get<framework::CLImage>();
Y
yangfei 已提交
128 129 130 131 132 133 134 135
          if (cl_image) {
            DLOG << type_ << " output- " << key << "=" << *cl_image;
          }
        }
      }
    }
  }
#endif
136
}
Y
yangfei 已提交
137
#endif
138

139 140
#ifdef PADDLE_MOBILE_FPGA
template <typename Dtype>
141 142 143 144 145 146 147 148 149 150 151 152 153
void OperatorBase<Dtype>::InsertTensors() {
  static int feed_num = 0;
  static int fetch_num = 0;
  if (type_ == "feed") {
    auto new_name = string("feed") + std::to_string(feed_num++);
    auto var = scope_->Var(new_name);
    var->template GetMutable<framework::LoDTensor>();
    inputs_.at("X") = {string(new_name)};
  } else if (type_ == "fetch") {
    auto new_name = string("fetch") + std::to_string(fetch_num++);
    auto var = scope_->Var(new_name);
    var->template GetMutable<framework::LoDTensor>();
    outputs_.at("Out") = {string(new_name)};
154 155 156 157
  }
}
#endif

朔-望's avatar
朔-望 已提交
158
template class OperatorBase<CPU>;
L
for  
liuruilong 已提交
159 160
template class OperatorBase<FPGA>;
template class OperatorBase<GPU_MALI>;
L
liuruilong 已提交
161
template class OperatorBase<GPU_CL>;
L
for  
liuruilong 已提交
162

朔-望's avatar
朔-望 已提交
163 164
}  // namespace framework
}  // namespace paddle_mobile