update the folder name with argmaxmin

Author: bowang007

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

@@ -2267,7 +2267,7 @@ def aten_ops_argmax(
     kwargs: Dict[str, Argument],
     name: str,
 ) -> Union[TRTTensor, Sequence[TRTTensor]]:
-    return impl.topk.argmax(
+    return impl.argmaxmin.argmax(
         ctx,
         target,
         SourceIR.ATEN,
@@ -2287,7 +2287,7 @@ def aten_ops_argmin(
     kwargs: Dict[str, Argument],
     name: str,
 ) -> Union[TRTTensor, Sequence[TRTTensor]]:
-    return impl.topk.argmin(
+    return impl.argmaxmin.argmin(
         ctx,
         target,
         SourceIR.ATEN,

py/torch_tensorrt/dynamo/conversion/impl/__init__.py

@@ -3,6 +3,7 @@
 from . import (
     activation,
     addmm,
+    argmaxmin,
     attention,
     cast,
     cat,
@@ -24,7 +25,6 @@
     slice,
     split,
     squeeze,
-    topk,
     unary,
     unsqueeze,
 )

py/torch_tensorrt/dynamo/conversion/impl/topk/__init__.py renamed to py/torch_tensorrt/dynamo/conversion/impl/argmaxmin/__init__.py

@@ -3,7 +3,7 @@
 from torch.fx.node import Target
 from torch_tensorrt.dynamo._SourceIR import SourceIR
 from torch_tensorrt.dynamo.conversion._ConversionContext import ConversionContext
-from torch_tensorrt.dynamo.conversion.impl.topk.base import argmax_argmin
+from torch_tensorrt.dynamo.conversion.impl.argmaxmin.base import argmax_argmin
 from torch_tensorrt.fx.types import TRTTensor