o
    ûpih  ã                   @   s:   d dl Zd dlmZ d dlmZ ejddddd„ ƒZdS )	é    N)Úset_layer_name)Úconverter_registryzpd_op.grid_samplez8.x)Ztrt_versionc                 C   sÔ   |\}}|  ¡  dddg¡}|  ¡  dd¡}|  ¡  dd¡}|  ¡  dd¡}|dkr/tjj}	n|d	kr8tjj}	n|d
kr@tjj}	|dkrItjj}
n|dkrQtjj	}
|  
||¡}|
|_||_|	|_t||ƒ | d¡S )NZpaddingsr   ÚmodeZbilinearÚpadding_modeZzerosÚalign_cornersTÚborderZ
reflectionZnearest)ÚattrsÚgetÚtrtZ
SampleModeZFILLZCLAMPZREFLECTZInterpolationModeZNEARESTZLINEARZadd_grid_sampleÚinterpolation_moder   Úsample_moder   Z
get_output)ÚnetworkZ	paddle_opZinputsZinput_tensorZgrid_tensorÚpaddingr   r   r   r   r   Zgrid_sample_layer© r   úc/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/tensorrt/impls/vision.pyÚgrid_sample_converter   s*   




r   )Ztensorrtr
   Zpaddle.tensorrt.converter_utilsr   Zpaddle.tensorrt.registerr   Úregisterr   r   r   r   r   Ú<module>   s
   