conv_op.cpp 2.2 KB
Newer Older
W
wangliu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
/* 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. */
朔-望's avatar
朔-望 已提交
14

L
liuruilong 已提交
15 16
#ifdef CONV_OP

Z
zhaojiaying01 已提交
17
#include "operators/conv_op.h"
Z
zhaojiaying01 已提交
18
#include <vector>
朔-望's avatar
朔-望 已提交
19
#include "framework/op_proto_maker.h"
Z
zhaojiaying01 已提交
20
#include "framework/op_registry.h"
L
liuruilong 已提交
21
#include "operators/math/conv_func.h"
朔-望's avatar
朔-望 已提交
22 23

namespace paddle_mobile {
朔-望's avatar
朔-望 已提交
24
namespace operators {
朔-望's avatar
朔-望 已提交
25

朔-望's avatar
朔-望 已提交
26
template <typename Dtype, typename T>
朔-望's avatar
朔-望 已提交
27
void ConvOp<Dtype, T>::InferShape() const {
L
liuruilong 已提交
28 29 30 31 32 33
  auto in_dims = this->param_.Input()->dims();
  auto filter_dims = this->param_.Filter()->dims();
  const std::vector<int> &strides = this->param_.Strides();
  std::vector<int> paddings = this->param_.Paddings();
  int groups = this->param_.Groups();
  std::vector<int> dilations = this->param_.Dilations();
朔-望's avatar
朔-望 已提交
34

W
wangliu 已提交
35 36 37 38 39
  PADDLE_MOBILE_ENFORCE((in_dims.size() == filter_dims.size() &&
                         dilations.size() == paddings.size() &&
                         paddings.size() == strides.size()),
                        "ConvParam is not suitable");

40 41
  std::vector<int64_t> output_shape({in_dims[0], filter_dims[0]});
  for (size_t i = 0; i < strides.size(); ++i) {
L
liuruilong 已提交
42 43 44
    output_shape.push_back(
        math::ConvOutputSize(in_dims[i + 2], filter_dims[i + 2], dilations[i],
                             paddings[i], strides[i]));
45
  }
朔-望's avatar
朔-望 已提交
46

47
  framework::DDim ddim = framework::make_ddim(output_shape);
L
liuruilong 已提交
48
  this->param_.Output()->Resize(ddim);
朔-望's avatar
朔-望 已提交
49
}
朔-望's avatar
朔-望 已提交
50

朔-望's avatar
朔-望 已提交
51
template class ConvOp<CPU, float>;
朔-望's avatar
朔-望 已提交
52

朔-望's avatar
朔-望 已提交
53 54
}  // namespace operators
}  // namespace paddle_mobile
Z
zhaojiaying01 已提交
55 56

namespace ops = paddle_mobile::operators;
L
for  
liuruilong 已提交
57 58 59 60 61 62 63 64 65 66
#ifdef PADDLE_MOBILE_CPU
REGISTER_OPERATOR_CPU(conv2d, ops::ConvOp);
#endif
#ifdef PADDLE_MOBILE_MALI_GPU
REGISTER_OPERATOR_MALI_GPU(conv2d, ops::ConvOp);
#endif
#ifdef PADDLE_MOBILE_FPGA
REGISTER_OPERATOR_FPGA(conv2d, ops::ConvOp);
#endif

L
liuruilong 已提交
67
#endif