code: def _tf_where(condition, x, y):
      return ab.instructions.pattern_map2(
          lambda x_elt, y_elt: tf1.where(condition, x_elt, y_elt), x, y
      )
  
imports: import collections
  import functools
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python import random
  from tensorflow_probability.python.experimental import auto_batching
  from tensorflow_probability.python.math import gradient
  from tensorflow_probability.python.mcmc import kernel
  from tensorflow_probability.python.util.seed_stream import SeedStream
label_desc: Use image node_id 15 for calling a global function with example usage: _tf_where(condition, x, y) and returns: ab