Fix failing test in group_conv_gpu.

PiperOrigin-RevId: 314178390
Change-Id: Icbd3300c0ede37b712c69661b5eb7fbbbaa4e7db
This commit is contained in:
Zhenyu Tan 2020-06-01 12:17:42 -07:00 committed by TensorFlower Gardener
parent 5498a3cb12
commit bbe22cf4dd

View File

@ -26,11 +26,11 @@ from tensorflow.python.eager import context
from tensorflow.python.eager import def_function
from tensorflow.python.framework import tensor_spec
from tensorflow.python.framework import test_util
from tensorflow.python.keras import keras_parameterized
from tensorflow.python.keras import testing_utils
from tensorflow.python.ops import array_ops
from tensorflow.python.ops import nn
from tensorflow.python.ops import random_ops
from tensorflow.python.keras import keras_parameterized
from tensorflow.python.keras import testing_utils
from tensorflow.python.platform import test
@ -359,6 +359,7 @@ class Conv3DTest(keras_parameterized.TestCase):
input_data=input_data)
@keras_parameterized.run_all_keras_modes(always_skip_v1=True)
class GroupedConvTest(keras_parameterized.TestCase):
@parameterized.named_parameters(