From 1eb151752e3c87d7b516e2db2c24cf92c49e8172 Mon Sep 17 00:00:00 2001 From: mozga-intel Date: Mon, 20 May 2019 23:15:51 -0700 Subject: [PATCH] Enable abs operator for a ngraph test=develop (#17436) --- paddle/fluid/operators/ngraph/ops/binary_unary_op.h | 1 + .../fluid/tests/unittests/ngraph/test_activation_ngraph_op.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/ngraph/ops/binary_unary_op.h b/paddle/fluid/operators/ngraph/ops/binary_unary_op.h index 2d11775849a..36630b0c620 100644 --- a/paddle/fluid/operators/ngraph/ops/binary_unary_op.h +++ b/paddle/fluid/operators/ngraph/ops/binary_unary_op.h @@ -51,6 +51,7 @@ static void BuildUnaryNode( } // namespace operators } // namespace paddle +REGISTER_NG_OP(abs, BuildUnaryNode); REGISTER_NG_OP(relu, BuildUnaryNode); REGISTER_NG_OP(tanh, BuildUnaryNode); REGISTER_NG_OP(sigmoid, BuildUnaryNode); diff --git a/python/paddle/fluid/tests/unittests/ngraph/test_activation_ngraph_op.py b/python/paddle/fluid/tests/unittests/ngraph/test_activation_ngraph_op.py index 034d7792c13..c7d62bd8ae1 100644 --- a/python/paddle/fluid/tests/unittests/ngraph/test_activation_ngraph_op.py +++ b/python/paddle/fluid/tests/unittests/ngraph/test_activation_ngraph_op.py @@ -18,7 +18,7 @@ import unittest import numpy as np import paddle.fluid.core as core from paddle.fluid.tests.unittests.op_test import OpTest -from paddle.fluid.tests.unittests.test_activation_op import TestSigmoid, TestRelu, TestTanh +from paddle.fluid.tests.unittests.test_activation_op import TestAbs, TestSigmoid, TestRelu, TestTanh class TestNGRAPHReluDim4(TestRelu): -- GitLab