="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">import</span> <span class="nn">torch_tensorrt.fx.tracer.dispatch_tracer.aten_tracer</span> <span class="k">as</span> <span class="nn">aten_tracer</span>
<span class="kn">from</span> <span class="nn">torch.fx.passes.splitter_base</span> <span class="kn">import</span> <span class="n">SplitResult</span>

<span class="kn">from</span> <span class="nn">.fx2trt</span> <span class="kn">import</