message: Shuffle output_sizes when testing NCHW.
  The output shape needs to reflect that the input has been transposed to NCHW.
  Without this fix the test fails with XLA somewhere inside
  compute_gradient_error.
  I have not done the analysis to figure out how this test passes with TF without
  XLA.
diff: @@ -253,6 +253,7 @@ class PoolingTest(test.TestCase):
  ksize = test_util.NHWCToNCHW(ksize)
  strides = test_util.NHWCToNCHW(strides)
  t = test_util.NHWCToNCHW(t)
  + output_sizes = test_util.NHWCToNCHW(output_sizes)
  t = pool_func(
  t,
  