message: Make `test_locallyconnected_implementation` parameterized.
diff: @@ -267,9 +267,16 @@ class LocallyConnected2DLayersTest(test.TestCase, parameterized.TestCase):
  @tf_test_util.run_all_in_graph_and_eager_modes
  -class LocallyConnectedImplementationModeTest(test.TestCase):
  -
  - def test_locallyconnected_implementation(self):
  +class LocallyConnectedImplementationModeTest(test.TestCase,
  + parameterized.TestCase):
  +
  + @parameterized.parameters([
  + {'width': 1, 'data_format': 'channels_first'},
  + {'width': 1, 'data_format': 'channels_last'},
  + {'width': 6, 'data_format': 'channels_first'},
  + {'width': 6, 'data_format': 'channels_last'},
  + ])
  + def test_locallyconnected_implementation(self, width, data_format):
  with self.cached_session():
  num_samples = 4
  num_classes = 3
  @@ -278,18 +285,16 @@ class LocallyConnectedImplementationModeTest(test.TestCase):
  np.random.seed(1)
  targets = np.random.randint(0, num_classes, (num_samples,))
  - for width in [1, 6]:
  - for height in [7]:
  - for filters in [2]:
  - for data_format in ['channels_first', 'channels_last']:
  - inputs = get_inputs(data_format, filters, height, num_samples,
  - width)
  -
  - for kernel_x in [(3,)]:
  - for kernel_y in [()] if width == 1 else [(2,)]:
  - for stride_x in [(1,)]:
  - for stride_y in [()] if width == 1 else [(3,)]:
  - for layers in [2]:
  + height = 7
  + filters = 2
  + inputs = get_inputs(data_format, filters, height, num_samples, width)
  +
  + kernel_x = (3,)
  + kernel_y = () if width == 1 else (2,)
  + stride_x = (1,)
  + stride_y = () if width == 1 else (3,)
  + layers = 2
  +
  kwargs = {
  'layers': layers,
  'filters': filters,
  