a
    .=ic#  ã                   @   s^   d Z ddlZddlmZ ddlmZ g d¢Zd
dd„Zddd	„ZeD ]Z	e 
e	¡eƒ e	< qDdS )a'  
Note [ONNX operators that are added/updated from opset 7 to opset 8]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
New operators:
  Expand

Updated operators:
  Min, Max, Sum, Mean: supports multidirectional broadcasting.
  MaxPool: added optional indices output.
  Scan
é    N)Úsymbolic_helper)Úsymbolic_opset9)
ÚscanÚexpandZ	expand_asZmeshgridZadaptive_max_pool1dZadaptive_max_pool2dZadaptive_max_pool3dZmax_pool1d_with_indicesZmax_pool2d_with_indicesZmax_pool3d_with_indicesc                 C   s*   |d u r|d urt  d¡ t | |||¡S )NzœMultidirectional broadcasting is not supported in opset 7. This might cause the onnx model to be incorrect, if inputs to max operators have different shapes)ÚwarningsÚwarnÚopset9Úmax©ÚgÚselfZdim_or_yZkeepdim© r   úk/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/torch/onnx/symbolic_opset7.pyr	   #   s
    ÿr	   c                 C   s*   |d u r|d urt  d¡ t | |||¡S )NzœMultidirectional broadcasting is not supported in opset 7. This might cause the onnx model to be incorrect, if inputs to min operators have different shapes)r   r   r   Úminr
   r   r   r   r   .   s
    ÿr   )NN)NN)Ú__doc__r   Z
torch.onnxr   r   r   Zblock_listed_operatorsr	   r   Zblock_listed_opZ_block_list_in_opsetÚvarsr   r   r   r   Ú<module>   s   

