code: def testNaNs(self):
      # Test that relu(nan) = nan for various sizes.
      for i in range(18):
          x = np.zeros(i) + np.nan
          # TODO(b/178335491): This is broken on GPU today.
          with self.cached_session(use_gpu=False):
              z = nn_ops.relu(_get_weak_tensor(x)).eval()
              self.assertTrue(np.isnan(z).all())
  
imports: from absl.testing import parameterized
  import numpy
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.framework import weak_tensor
  from tensorflow.python.ops import gradient_checker_v2
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_impl
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import weak_tensor_ops
  from tensorflow.python.ops import weak_tensor_test_util
  import tensorflow.python.ops.nn_grad
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the ReluTest obj's underlying member method code with example usage: obj.testNaNs() without return types