o
    wi                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)OpRunc                   @  s   e Zd ZdddZdS )GatherNc                 C  s   |j d s
t|}|j d s| }|jdkr!tjd|jdfS z
tj|||dfW S  ty@   tj||t	|df Y S w )NC_CONTIGUOUSr   )r   )dtype)axis)
flagsnpascontiguousarraysizeemptyr   take	TypeErrorastypeint)selfxindicesr    r   Y/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_gather.py_run   s   



zGather._run)N)__name__
__module____qualname__r   r   r   r   r   r      s    r   )
__future__r   numpyr	   onnx.reference.op_runr   r   r   r   r   r   <module>   s   