m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Enable SQRT operator for the nGraph Bridge (#17549)

Browse Source 
* Enable sqrt operator for the nGraph Bridge.

test=develop

* Update activation_op.h
dependabot/pip/python/requests-2.20.0
Krzysztof Binias 6 years ago committed by Tao Luo
parent 96845d2168
commit f34830e2aa
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File

1
paddle/fluid/operators/ngraph/ops/activation_op.h
Unescape View File

@ -113,5 +113,6 @@ void BuildTanhGradNode(
REGISTER_NG_OP(gelu, BuildGeluNode);
REGISTER_NG_OP(gelu_grad, BuildGeluGradNode);
REGISTER_NG_OP(relu_grad, BuildReluGradNode);
REGISTER_NG_OP(sqrt, BuildUnaryNode<ngraph::op::Sqrt>);
REGISTER_NG_OP(square, BuildSquareNode);
REGISTER_NG_OP(tanh_grad, BuildTanhGradNode);

2
python/paddle/fluid/tests/unittests/ngraph/test_activation_ngraph_op.py
Unescape View File

@ -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 TestAbs, TestGelu, TestSigmoid, TestSquare, TestRelu, TestTanh
from paddle.fluid.tests.unittests.test_activation_op import TestAbs, TestGelu, TestSigmoid, TestSqrt, TestSquare, TestRelu, TestTanh
class TestNGRAPHReluDim4(TestRelu):

Write Preview
Loading…
Cancel
Save