o Ơiy @sUddlZddlmZmZmZddlmZiZeeefed<iZ iZ e e Z   ddeded ed ed eegeff d d ZdS)N)AnyCallableDict)Target CONVERTERSFTkeyno_implicit_batch_dimno_explicit_batch_dimenabledreturncs$fdd}dd}|r|S|S)NcsZ|t<r |t<r|t<tddrdnddr$dndd|S)NzConverter for z added to FX Converter Registry withoutwithz Explicit Batch Dim Support + z Implicit Batch Dim Support)rNO_IMPLICIT_BATCH_DIM_SUPPORTNO_EXPLICIT_BATCH_DIM_SUPPORTloggerdebug converterrr rX/home/ubuntu/.local/lib/python3.10/site-packages/torch_tensorrt/fx/converter_registry.pyregister_converters z.tensorrt_converter..register_convertercSs|S)Nrrrrrdisable_converter#sz-tensorrt_converter..disable_converterr)rrr r rrrrrtensorrt_converters r)FFT)loggingtypingrrr torch.fx.noderr__annotations__rr getLogger__name__rboolrrrrrs*