o
    r::j
  ã                   @  s4   d dl mZ d dlZd dlmZ G dd„ deƒZdS )é    )ÚannotationsN)ÚOpRunBinaryNumpyc                   @  s   e Zd Zdd„ Zdd„ ZdS )ÚDivc                 C  s   dd„ }t  | |||¡ d S )Nc                 S  sp   t | jjtjƒr2t |jjtjƒsJ ‚t | |¡}t | |¡}|dk| dk |dk A @ }|| |j¡ S t | |¡S ©Nr   )	Ú
issubclassÚdtypeÚtypeÚnpÚintegerÚfloor_divideÚ	remainderÚastypeÚdivide)ÚxÚyÚqÚrÚneeds_adjust© r   ú`/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_div.pyÚfunc   s   zDiv.__init__.<locals>.func)r   Ú__init__)ÚselfÚ	onnx_nodeÚ
run_paramsr   r   r   r   r      s   zDiv.__init__c                 C  s4   t  | ||¡}|d j|jkr|d  |j¡fS |S r   )r   Ú_runr   r   )r   ÚaÚbÚresr   r   r   r      s   zDiv._runN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r      s    r   )Ú
__future__r   Únumpyr	   Úonnx.reference.ops._opr   r   r   r   r   r   Ú<module>   s   