4e6f863acd500dc02e7c2a8e5a01566ac58930d9,sonnet/python/modules/conv_test.py,,,#,4324

Before Change

    with self.assertRaisesRegexp(snt.NotConnectedError, err):
      self.assertEqual(conv3_transpose.output_shape, conv3.input_shape)

if __name__ == "__main__":
  tf.test.main()

After Change

// These functions compute the expected output shape of a convolution of each
// padding type, for a given input shape and kernel size.
_PADDINGS_EXPECTED_SHAPE_TRANSFORMS = {
    snt.SAME: lambda in_length, kernel_size: in_length,
    snt.VALID: lambda in_length, kernel_size: in_length - kernel_size + 1,
    snt.FULL: lambda in_length, kernel_size: in_length + kernel_size - 1,
    snt.CAUSAL: lambda in_length, kernel_size: in_length,
    snt.REVERSE_CAUSAL: lambda in_length, kernel_size: in_length,
}


class Conv2DTest(parameterized.TestCase, tf.test.TestCase):

Instances