o Ơi@sPddlZddlZddlmZddlmZmZddZeej j j j dd Z dS) N)tensorrt_converter)extend_mod_attr_to_tuplemark_as_int8_layerc Csvt|d|}t|d|}t|d|}|j|tjj|d}||_||_||_|jr-tj j |_ |j r6t ||j |dS)N kernel_sizestridepadding)inputtype window_sizer)r add_poolingtrt PoolingTypeMAXrr name ceil_mode PaddingModeEXPLICIT_ROUND_UP padding_mode dynamic_ranger get_output) networkmod dimension input_val layer_namerrr layerrX/home/ubuntu/.local/lib/python3.10/site-packages/torch_tensorrt/fx/converters/maxpool.pycommon_maxpool s       r cCsHt|dksJ|d}t|tjjstd|dt||d||dS)Nrr zMaxPool2d received input z) that is not part of the TensorRT region!r)rrr)len isinstancertensorrtITensor RuntimeErrorr )rsubmodargskwargsrrrrr maxpool2d s  r))r#rtorchconverter_registryrconverter_utilsrrr nnmodulespooling MaxPool2dr)rrrrs