diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/depthwise_conv_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/depthwise_conv_op_test.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/depthwise_conv_op_test.py b/tensorflow/python/kernel_tests/depthwise_conv_op_test.py index 58845552db..5741f2ec64 100644 --- a/tensorflow/python/kernel_tests/depthwise_conv_op_test.py +++ b/tensorflow/python/kernel_tests/depthwise_conv_op_test.py @@ -205,6 +205,19 @@ class DepthwiseConv2DTest(test.TestCase): use_gpu=True, grouped_conv=True) + def testDepthwiseConv2DWithUnknownShape(self): + # GitHub issue 22110. + if not test.is_gpu_available(): + return + with self.test_session(use_gpu=True): + x = array_ops.placeholder(dtypes.float32) + f = np.ones([1, 1, 1, 1], np.float32) + v = nn_impl.depthwise_conv2d( + x, f, [1, 1, 1, 1], "VALID", rate=[2, 1], data_format="NCHW") + self.assertAllEqual( + np.ones([1, 1, 1, 1], np.float32), + v.eval(feed_dict={x: np.ones([1, 1, 1, 1], np.float32)})) + def testDepthwiseConv2DFormat(self): if not test.is_gpu_available(): return |