U
    ‰d#  ã                
   @   sn   d Z ddlZddlmZ ddlmZ ddddd	d
ddddg
Zddd„Zddd„ZeD ]Z	e 
e	¡eƒ e	< qTd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 kr|d k	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   ú>/tmp/pip-unpacked-wheel-ua33x9lu/torch/onnx/symbolic_opset7.pyr	   #   s
    ÿr	   c                 C   s*   |d kr|d k	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$   ö

