diff --git a/paddle/fluid/operators/ngraph/ops/conv2d_op.h b/paddle/fluid/operators/ngraph/ops/conv2d_op.h index be766ebeb4796be102c917296238b8ab14710131..b8ad7491d57b2d509c8a30b7848590339b13056b 100644 --- a/paddle/fluid/operators/ngraph/ops/conv2d_op.h +++ b/paddle/fluid/operators/ngraph/ops/conv2d_op.h @@ -239,3 +239,4 @@ void BuildConv2dGradNode( REGISTER_NG_OP(conv2d, BuildConv2dNode); REGISTER_NG_OP(conv2d_grad, BuildConv2dGradNode); +REGISTER_NG_OP(depthwise_conv2d, BuildConv2dNode); diff --git a/python/paddle/fluid/tests/unittests/ngraph/test_conv2d_ngraph_op.py b/python/paddle/fluid/tests/unittests/ngraph/test_conv2d_ngraph_op.py index ff2e865b66a5f1166281c267392b0964ca5b3082..764d136ec8d1e4f6772d4d1cdd03b6494aa735d1 100644 --- a/python/paddle/fluid/tests/unittests/ngraph/test_conv2d_ngraph_op.py +++ b/python/paddle/fluid/tests/unittests/ngraph/test_conv2d_ngraph_op.py @@ -15,7 +15,40 @@ from __future__ import print_function 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__': unittest.main()