o
    pi                     @   sL   d dl mZmZ d dlmZ ejddddd Zejdddd	d
 ZdS )    )set_layer_name	trt_shape)converter_registryzpd_op.shapeztrt_version_ge=8.0)Ztrt_versionc                 C   s   t | |d | dgdS )Nr   r   )name)r   r   )network	paddle_opinputs r	   f/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/tensorrt/impls/attribute.pyshape_converter   s   r   zpd_op.shape64c                 C   s&   |d }|  |}t|| |dS )Nr   )Z	add_shaper   Z
get_output)r   r   r   Zinput_tensorZshape_layerr	   r	   r
   shape64_converter   s   


r   N)Zpaddle.tensorrt.converter_utilsr   r   Zpaddle.tensorrt.registerr   registerr   r   r	   r	   r	   r
   <module>   s   
