code: def _qr(
      input, full_matrices=False, name=None
  ):  # pylint: disable=redefined-builtin,unused-argument
      mode = "complete" if full_matrices else "reduced"
      op = functools.partial(np.linalg.qr, mode=mode)
      if not JAX_MODE:
          op = np.vectorize(op, signature="(m,n)->(m,r),(r,n)")
      return Qr(*op(input))
  
imports: import collections
  import functools
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import ops
label_desc: Use image node_id 17 for calling a global function with example usage: _qr(input, full_matrices, name) and returns: Qr