code: def argwhere_2d(output_shape, condition):
      """Compute for argwhere 2D
  
      Parameters
      ----------
      condition : list of int or tvm.tir.Any
          The output shape
  
      out : tvm.te.Tensor
          Tensor with boolean values.
  
      Returns
      -------
      stmt : Stmt
          The result IR statement.
      """
  
      def do_write(out, write_index, idx):
          a1 = condition.shape[1]
          out[write_index * 2] = tvm.tir.floordiv(idx, a1)
          out[write_index * 2 + 1] = tvm.tir.floormod(idx, a1)
  
      return argwhere_common(output_shape, condition, do_write)
  
file_path: tvm/python/tvm/topi/cuda/argwhere.py
imports: import logging
  import tvm
  from tvm import te
  from .injective import schedule_injective_from_existing
  from .scan import exclusive_scan
  from ..None import tag
  from ..utils import ceil_div, prod
  from ..transform import reshape
  from ..broadcast import not_equal
  from ..math import cast
label_desc: Use image node_id 4 for calling a global function with example usage: argwhere_2d(output_shape, condition) and returns: argwhere_common