You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
3.6 KiB
83 lines
3.6 KiB
7 years ago
|
/* Copyright (c) 2016 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 "paddle/operators/compare_op.h"
|
||
|
|
#include "paddle/framework/op_registry.h"
|
||
|
|
namespace paddle {
|
||
|
|
namespace operators {
|
||
|
|
template <typename OpComment>
|
||
|
|
class CompareOpProtoMaker : public framework::OpProtoAndCheckerMaker {
|
||
|
|
public:
|
||
|
|
CompareOpProtoMaker(framework::OpProto *proto,
|
||
|
|
framework::OpAttrChecker *op_checker)
|
||
|
|
: OpProtoAndCheckerMaker(proto, op_checker) {
|
||
|
|
OpComment comment;
|
||
|
|
AddInput("X",
|
||
|
|
string::Sprintf("(LoDTensor) the left hand operand of %s operator",
|
||
|
|
comment.type));
|
||
|
|
AddInput("Y", string::Sprintf(
|
||
|
|
"(LoDTensor) the right hand operand of %s operator",
|
||
|
|
comment.type));
|
||
|
|
AddOutput("Out", string::Sprintf(
|
||
|
|
"(LoDTensor) n-dim bool tensor. Each element is %s",
|
||
|
|
comment.equation));
|
||
|
|
AddComment(string::Sprintf(R"DOC(%s Operator
|
||
|
|
|
||
|
|
It operates element-wise on X and Y, and returns the Out. Each of them is a
|
||
|
|
N-dim tensor. X and Y could be any type. The each element of the Out tensor is
|
||
|
|
calculated by %s
|
||
|
|
)DOC",
|
||
|
|
comment.type, comment.equation));
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename OpComment>
|
||
|
|
class CompareOpInferShape : public framework::InferShapeBase {
|
||
|
|
public:
|
||
|
|
void operator()(framework::InferShapeContext *context) const override {
|
||
|
|
OpComment comment;
|
||
|
|
PADDLE_ENFORCE(context->HasInput("X"), "%s operator must has input X",
|
||
|
|
comment.type);
|
||
|
|
PADDLE_ENFORCE(context->HasInput("Y"), "%s operator must has input Y",
|
||
|
|
comment.type);
|
||
|
|
auto dim_x = context->GetInputDim("X");
|
||
|
|
auto dim_y = context->GetInputDim("Y");
|
||
|
|
PADDLE_ENFORCE_EQ(framework::product(dim_x), framework::product(dim_y),
|
||
|
|
"The number of elements in X and Y should be same");
|
||
|
|
|
||
|
|
context->SetOutputDim("Out", context->GetInputDim("X"));
|
||
|
|
context->ShareLoD("X", "Out");
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace operators
|
||
|
|
} // namespace paddle
|
||
|
|
|
||
|
|
#define REGISTER_LOGICAL_OP(op_type, _equation) \
|
||
|
|
struct _##op_type##Comment { \
|
||
|
|
static char type[]; \
|
||
|
|
static char equation[]; \
|
||
|
|
}; \
|
||
|
|
char _##op_type##Comment::type[]{#op_type}; \
|
||
|
|
char _##op_type##Comment::equation[]{_equation}; \
|
||
|
|
REGISTER_OP_WITH_KERNEL( \
|
||
|
|
op_type, ::paddle::operators::CompareOpProtoMaker<_##op_type##Comment>, \
|
||
|
|
::paddle::operators::CompareOpInferShape<_##op_type##Comment>, \
|
||
|
|
::paddle::framework::EmptyGradOpMaker);
|
||
|
|
|
||
|
|
REGISTER_LOGICAL_OP(less_than, "Out = X < Y");
|
||
|
|
REGISTER_LOGICAL_KERNEL(less_than, CPU, paddle::operators::LessThanFunctor);
|
||
|
|
REGISTER_LOGICAL_OP(equal, "Out = X == Y");
|
||
|
|
REGISTER_LOGICAL_KERNEL(equal, CPU, paddle::operators::EqualFunctor);
|