o toi@s0ddlmZddlZddZddZddZdS) ) annotationsNcCsntjt|ftjd}t|d}|dkr1|||}|||<||8}|||}|d8}|dks||d<|S)N)dtyper)npemptylenint64)ishapereskmrY/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/_op_common_indices.py _get_indices s   rcCs8t||ddD]\}}|dkrdS||krdSqdS)NTstrictrF)zip)indr r srrr_is_outsrcCs@d}d}tt|t|ddD]\}}|||7}||9}q|S)NrrTr)rreversed)indicesr rmulposshrrr _get_indexs   r) __future__rnumpyrrrrrrrrs