lrn_op.h 1.5 KB
Newer Older
E
eclipsess 已提交
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. */
L
liuruilong 已提交
14 15 16

#ifdef LRN_OP

17
#pragma once
E
eclipsess 已提交
18

19
#include <string>
E
eclipsess 已提交
20 21 22 23 24 25
#include "framework/operator.h"
#include "operators/kernel/lrn_kernel.h"
#include "operators/op_param.h"

namespace paddle_mobile {
namespace operators {
26
using std::string;
E
eclipsess 已提交
27
template <typename DeviceType, typename T>
L
liuruilong 已提交
28
class LrnOp : public framework::OperatorWithKernel<
N
nhzlx 已提交
29 30
                  DeviceType, LrnParam<DeviceType>,
                  operators::LrnKernel<DeviceType, T>> {
朔-望's avatar
朔-望 已提交
31
 public:
32
  LrnOp(const string &type, const VariableNameMap &inputs,
L
liuruilong 已提交
33
        const VariableNameMap &outputs, const framework::AttributeMap &attrs,
34
        std::shared_ptr<framework::Scope> scope)
N
nhzlx 已提交
35
      : framework::OperatorWithKernel<DeviceType, LrnParam<DeviceType>,
L
liuruilong 已提交
36 37
                                      operators::LrnKernel<DeviceType, T>>(
            type, inputs, outputs, attrs, scope) {}
38
  void InferShape() const override;
L
liuruilong 已提交
39

朔-望's avatar
朔-望 已提交
40
 protected:
E
eclipsess 已提交
41 42
};

朔-望's avatar
朔-望 已提交
43 44
}  // namespace operators
}  // namespace paddle_mobile
L
liuruilong 已提交
45 46

#endif