feat: Add aten.unbind decomposition for VIT (#2430)

gs-olive · gs-olive · commit 2549edba1795 · 2023-11-02T18:01:27.000-07:00
diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py
@@ -153,6 +153,7 @@
     aten.transpose.int,
     aten.tril.default,
     aten.triu.default,
+    aten.unbind,
     aten.unfold,
     aten.unfold_backward,
     aten.unfold_copy,
