a1b1f343d1213127518c12be84ad313c75f375ff,sonnet/python/modules/conv_gpu_test.py,Conv3DTestDataFormats,testConv3DDataFormats,#Conv3DTestDataFormats#Any#Any#,231

Before Change

        stride=stride,
        initializers=create_constant_initializers(1.0, 1.0, use_bias))
    x = tf.constant(np.random.random(self.INPUT_SHAPE).astype(np.float32))
    self.helperDataFormats(func, x)

  @parameterized.named_parameters(
      ("WithBias_Stride1", True, 1), ("WithoutBias_Stride1", False, 1),
      ("WithBias_Stride2", True, 2), ("WithoutBias_Stride2", False, 2))

After Change

    conv1 = func(name="NDHWC", data_format="NDHWC")
    x = tf.constant(np.random.random(self.INPUT_SHAPE).astype(np.float32))
    o1 = conv1(x)

    // We will force both modules to share the same weights by creating
    // a custom getter that returns the weights from the first conv module when
    // tf.get_variable is called.
    custom_getter = {"w": create_custom_field_getter(conv1, "w"),
                     "b": create_custom_field_getter(conv1, "b")}
    conv2 = func(name="NCDHW", data_format="NCDHW", custom_getter=custom_getter)
    x_transpose = tf.transpose(x, perm=(0, 4, 1, 2, 3))
    o2 = tf.transpose(conv2(x_transpose), perm=(0, 2, 3, 4, 1))

    self.checkEquality(o1, o2)

  @parameterized.named_parameters(("WithBias", True), ("WithoutBias", False))
  def testConv3DDataFormatsBatchNorm(self, use_bias):
    Similar to `testConv3DDataFormats`, but this checks BatchNorm support.

Instances