o p i@s2ddlmZddlmZejdddddZdS) )set_layer_name)converter_registryz pd_op.einsumz8.x)Z trt_versioncCs8|dd}||d|}t|||d}|S)Nequationr)attrsgetZ add_einsumrZ get_output)networkZ paddle_opZinputsrlayerZ output_tensorr c/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/tensorrt/impls/einsum.pyconvert_einsums   r N)Zpaddle.tensorrt.converter_utilsrZpaddle.tensorrt.registerrregisterr r r r r s