Enable abs operator for a ngraph test=develop (#17436)

resnext-opt
mozga-intel 7 years ago committed by Tao Luo
parent 36757ed203
commit 1eb151752e

@ -51,6 +51,7 @@ static void BuildUnaryNode(
} // namespace operators
} // namespace paddle
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>);

@ -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):

Loading…
Cancel
Save