提交 73a9f0f2 编写于 作者: Y Yu Yang

Merge develop

上级 ad8fa77c
...@@ -13,6 +13,7 @@ ...@@ -13,6 +13,7 @@
limitations under the License. */ limitations under the License. */
#pragma once #pragma once
#include <glog/logging.h> #include <glog/logging.h>
#include <paddle/framework/operator.h> #include <paddle/framework/operator.h>
...@@ -22,7 +23,7 @@ namespace operators { ...@@ -22,7 +23,7 @@ namespace operators {
template <typename Place> template <typename Place>
class MulKernel : public framework::OpKernel { class MulKernel : public framework::OpKernel {
public: public:
void Compute(const KernelContext &context) const override { void Compute(const framework::KernelContext &context) const override {
LOG(INFO) << "Mul kernel in " << typeid(Place).name(); LOG(INFO) << "Mul kernel in " << typeid(Place).name();
} }
}; };
......
...@@ -22,7 +22,7 @@ namespace operators { ...@@ -22,7 +22,7 @@ namespace operators {
template <typename Place> template <typename Place>
class RowWiseAddKernel : public framework::OpKernel { class RowWiseAddKernel : public framework::OpKernel {
public: public:
void Compute(const KernelContext &context) const override { void Compute(const framework::KernelContext &context) const override {
LOG(INFO) << "RowWiseAdd kernel in " << typeid(Place).name(); LOG(INFO) << "RowWiseAdd kernel in " << typeid(Place).name();
} }
}; };
......
...@@ -23,7 +23,7 @@ namespace operators { ...@@ -23,7 +23,7 @@ namespace operators {
template <typename Place> template <typename Place>
class SigmoidKernel : public framework::OpKernel { class SigmoidKernel : public framework::OpKernel {
public: public:
void Compute(const KernelContext &context) const override { void Compute(const framework::KernelContext &context) const override {
LOG(INFO) << "Sigmoid kernel in " << typeid(Place).name(); LOG(INFO) << "Sigmoid kernel in " << typeid(Place).name();
} }
}; };
......
...@@ -23,7 +23,7 @@ namespace operators { ...@@ -23,7 +23,7 @@ namespace operators {
template <typename Place> template <typename Place>
class SoftmaxKernel : public framework::OpKernel { class SoftmaxKernel : public framework::OpKernel {
public: public:
void Compute(const KernelContext &context) const override { void Compute(const framework::KernelContext &context) const override {
LOG(INFO) << "Softmax kernel in " << typeid(Place).name(); LOG(INFO) << "Softmax kernel in " << typeid(Place).name();
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册