code: def hybrid_argwhere_3d(output_shape, condition):
      """Find the indices of elements of a 3-D tensor that are non-zero.
  
      Parameters
      ----------
      condition : tvm.te.Tensor
          3-D tensor with boolean values.
  
      Returns
      -------
      out : tvm.te.Tensor
          Indices of non-zero elements.
      """
      a = output_tensor(output_shape, "int32")
      a1 = condition.shape[0]
      a2 = condition.shape[1]
      a3 = condition.shape[2]
      valid_index = 0
      for i1 in range(a1):
          for i2 in range(a2):
              for i3 in range(a3):
                  if condition[i1, i2, i3] != 0:
                      a[valid_index, 0] = i1
                      a[valid_index, 1] = i2
                      a[valid_index, 2] = i3
                      valid_index += 1
      return a
  
file_path: tvm/python/tvm/topi/argwhere.py
imports: import tvm
  from tvm.te import hybrid
label_desc: Use image node_id 3 for calling a global function with example usage: hybrid_argwhere_3d(output_shape, condition) and returns: a