`iS@ddlmZddZejddddZdS))_coreNc$t||S)aCounts the number of non-zero values in the array. .. note:: :func:`numpy.count_nonzero` returns `int` value when `axis=None`, but :func:`cupy.count_nonzero` returns zero-dimensional array to reduce CPU-GPU synchronization. Args: a (cupy.ndarray): The array for which to count non-zeros. axis (int or tuple, optional): Axis or tuple of axes along which to count non-zeros. Default is None, meaning that non-zeros will be counted along a flattened version of ``a`` Returns: cupy.ndarray of int: Number of non-zero values in the array along a given axis. Otherwise, the total number of non-zero values in the array is returned. )axis)_count_nonzero)ars g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_sorting/count.py count_nonzeror s( !$ ' ' ''cupy_count_nonzero)z?->lzB->lzh->lzH->lzi->lzI->lzl->lzL->lzq->lzQ->lze->lzf->lzd->lzF->lzD->l)zin0 != type_in0_raw(0)za + bzout0 = aN)N)cupyrr create_reduction_funcrr rrsR((((.-,=91 >>r