4e6f863acd500dc02e7c2a8e5a01566ac58930d9,sonnet/python/modules/conv_test.py,SharedConvTest,testPartitioners,#SharedConvTest#Any#Any#Any#,246

Before Change

      self.assertEqual(type(getattr(convolution, key)),
                       variables.PartitionedVariable)

    if isinstance(conv, snt.Transposable):
      convolution_t = convolution.transpose()
      self.assertEqual(convolution_t.partitioners, convolution.partitioners)

  @parameterized.parameters(*itertools.product(modules,
                                               (True, False),
                                               (tf.float16, tf.float32)))
  def testVariables(self, module_info, use_bias, dtype):

After Change

      self.assertEqual(type(getattr(convolution, key)),
                       variables.PartitionedVariable)

    try:
      convolution_t = convolution.transpose()
    except (AttributeError, snt.NotSupportedError):
      return

    self.assertEqual(convolution_t.partitioners, convolution.partitioners)

  @parameterized.parameters(*itertools.product(modules,
                                               (True, False),

Instances