code: def test_get_output_spec_oom(self):
      result = callable_util.get_output_spec(_compute_oom)
      self.assertEqual((int(1e9), int(1e9)), result.shape)
      self.assertEqual(tf.float32, result.dtype)
  
file_path: probability/tensorflow_probability/python/internal/callable_util_test.py
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import callable_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the CallableUtilTest obj's underlying member method code with example usage: obj.test_get_output_spec_oom() without return types