add_op.h 373 字节
Newer Older
Y
Yu Yang 已提交
1 2 3 4 5
#pragma once
#include <glog/logging.h>
#include <paddle/framework/operator.h>

namespace paddle {
Y
Yu Yang 已提交
6
namespace operators {
Y
Yu Yang 已提交
7 8 9 10 11 12 13 14 15 16 17

template <typename Place>
class AddKernel : public framework::OpKernel {
public:
  void Compute(const KernelContext &context) const override {
    LOG(INFO) << "Add kernel in " << typeid(Place).name();
  }
};

}  // namespace op
}  // namespace paddle