From e9216d0602d29cfecaa9f122dd1e5c5aa1f9fae0 Mon Sep 17 00:00:00 2001
From: Krzysztof Binias <krzysztof.binias@intel.com>
Date: Fri, 24 May 2019 11:57:59 +0200
Subject: [PATCH] Enable logical operators for the nGraph Bridge. (#17543)

test=develop
---
 .../operators/ngraph/ops/binary_unary_op.h    |  4 +++
 .../ngraph/test_logical_ngraph_op.py          | 28 +++++++++++++++++++
 2 files changed, 32 insertions(+)
 create mode 100644 python/paddle/fluid/tests/unittests/ngraph/test_logical_ngraph_op.py

diff --git a/paddle/fluid/operators/ngraph/ops/binary_unary_op.h b/paddle/fluid/operators/ngraph/ops/binary_unary_op.h
index 36630b0c6..b8e9f3d85 100644
--- a/paddle/fluid/operators/ngraph/ops/binary_unary_op.h
+++ b/paddle/fluid/operators/ngraph/ops/binary_unary_op.h
@@ -55,3 +55,7 @@ REGISTER_NG_OP(abs, BuildUnaryNode<ngraph::op::Abs>);
 REGISTER_NG_OP(relu, BuildUnaryNode<ngraph::op::Relu>);
 REGISTER_NG_OP(tanh, BuildUnaryNode<ngraph::op::Tanh>);
 REGISTER_NG_OP(sigmoid, BuildUnaryNode<ngraph::op::Sigmoid>);
+
+REGISTER_NG_OP(logical_and, BuildBinaryNode<ngraph::op::And>);
+REGISTER_NG_OP(logical_or, BuildBinaryNode<ngraph::op::Or>);
+REGISTER_NG_OP(logical_not, BuildUnaryNode<ngraph::op::Not>);
diff --git a/python/paddle/fluid/tests/unittests/ngraph/test_logical_ngraph_op.py b/python/paddle/fluid/tests/unittests/ngraph/test_logical_ngraph_op.py
new file mode 100644
index 000000000..2f227ce87
--- /dev/null
+++ b/python/paddle/fluid/tests/unittests/ngraph/test_logical_ngraph_op.py
@@ -0,0 +1,28 @@
+#   Copyright (c) 2019 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.
+
+from __future__ import print_function
+
+import unittest, sys
+sys.path.append("../")
+import numpy as np
+
+from test_logical_op import create_test_class
+
+create_test_class('logical_and', lambda _a, _b: np.logical_and(_a, _b))
+create_test_class('logical_or', lambda _a, _b: np.logical_or(_a, _b))
+create_test_class('logical_not', lambda _a: np.logical_not(_a), False)
+
+if __name__ == '__main__':
+    unittest.main()
-- 
GitLab