code: def argwhere_4d(output_shape, condition):
      """Compute for argwhere 4D
  
      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, a2, a3 = condition.shape
          s1 = a2 * a3
          s2 = a1 * s1
          out[write_index * 4] = fdiv(idx, s2)
          out[write_index * 4 + 1] = fdiv(fmod(idx, s2), s1)
          out[write_index * 4 + 2] = fdiv(fmod(idx, s1), a3)
          out[write_index * 4 + 3] = fmod(idx, a3)
  
      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 6 for calling a global function with example usage: argwhere_4d(output_shape, condition) and returns: argwhere_common