code: def _compute_oom():
      x = tf.ones(int(1e9))
      y = tf.einsum("i,j->ij", x, x)
      return y
  
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 a global function with example usage: _compute_oom() and returns: y