message: Allocate test tensor on CPU to avoid OOM on GPU
diff: @@ -22,6 +22,7 @@ import numpy as np
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  +from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  @@ -202,6 +203,7 @@ class ReshapeTest(test.TestCase):
  self.assertEqual([100, 1], y.get_shape().as_list())
  def testInt64Shape(self):
  + with ops.device("/device:CPU:0"):
  x = array_ops.zeros([50000, 50000], dtype=dtypes.bool)
  # Provide dimension larger than int32
  y = array_ops.reshape(x, [50000**2])
  