提交 29baca0d 编写于 作者: B baojun 提交者: Tao Luo

add depthwise_conv2d op to ngraph engine (#17454)

* add depthwise_conv2d test=develop

* use cpu for ngraph test=develop
上级 1735ed32
...@@ -239,3 +239,4 @@ void BuildConv2dGradNode( ...@@ -239,3 +239,4 @@ void BuildConv2dGradNode(
REGISTER_NG_OP(conv2d, BuildConv2dNode); REGISTER_NG_OP(conv2d, BuildConv2dNode);
REGISTER_NG_OP(conv2d_grad, BuildConv2dGradNode); REGISTER_NG_OP(conv2d_grad, BuildConv2dGradNode);
REGISTER_NG_OP(depthwise_conv2d, BuildConv2dNode);
...@@ -15,7 +15,40 @@ ...@@ -15,7 +15,40 @@
from __future__ import print_function from __future__ import print_function
import unittest import unittest
from paddle.fluid.tests.unittests.test_conv2d_op import TestConv2dOp, TestWithPad, TestWithStride, TestWithGroup, TestWith1x1, TestWithInput1x1Filter1x1 from paddle.fluid.tests.unittests.test_conv2d_op import TestConv2dOp, TestWithPad, TestWithStride, TestWithGroup, TestWith1x1, TestWithInput1x1Filter1x1, TestDepthwiseConv, TestDepthwiseConv2, TestDepthwiseConv3, TestDepthwiseConvWithDilation, TestDepthwiseConvWithDilation2
class TestNGRAPHDepthwiseConv(TestDepthwiseConv):
def init_test_case(self):
super(TestNGRAPHDepthwiseConv, self).init_test_case()
self.use_cuda = False
class TestNGRAPHDepthwiseConv2(TestDepthwiseConv2):
def init_test_case(self):
super(TestNGRAPHDepthwiseConv2, self).init_test_case()
self.use_cuda = False
class TestNGRAPHDepthwiseConv3(TestDepthwiseConv3):
def init_test_case(self):
super(TestNGRAPHDepthwiseConv3, self).init_test_case()
self.use_cuda = False
class TestNGRAPHDepthwiseConvWithDilation(TestDepthwiseConvWithDilation):
def init_test_case(self):
super(TestNGRAPHDepthwiseConvWithDilation, self).init_test_case()
self.use_cuda = False
class TestNGRAPHDepthwiseConvWithDilation2(TestDepthwiseConvWithDilation2):
def init_test_case(self):
super(TestNGRAPHDepthwiseConvWithDilation2, self).init_test_case()
self.use_cuda = False
del TestDepthwiseConv, TestDepthwiseConv2, TestDepthwiseConv3, TestDepthwiseConvWithDilation, TestDepthwiseConvWithDilation2
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册