code: def _where(
      condition, x=None, y=None, name="where"
  ):  # pylint: disable=unused-argument
      if x is None and y is None:
          return np.stack(np.asarray(condition).nonzero(), axis=-1)
      return np.where(condition, x, y)
  
imports: import functools
  import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import ops
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import einsum
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import norm
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import tensordot
label_desc: Use image node_id 22 for calling a global function with example usage: _where(condition, x, y, name) and returns: np, np