a1b1f343d1213127518c12be84ad313c75f375ff,sonnet/python/modules/conv_gpu_test.py,Conv3DTestDataFormats,testConv3DDataFormatsBatchNorm,#Conv3DTestDataFormats#Any#,271

Before Change

                           axis=(0, 2, 3, 4))
      return snt.Sequential([conv, functools.partial(bn, is_training=True)])
    x = tf.constant(np.random.random(self.INPUT_SHAPE).astype(np.float32))
    self.helperDataFormats(func, x)

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

After Change

                           axis=(0, 2, 3, 4))
      return snt.Sequential([conv, functools.partial(bn, is_training=True)])

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

    custom_getter = {"w": create_custom_field_getter(conv1.layers[0], "w"),
                     "b": create_custom_field_getter(conv1.layers[0], "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)

Instances