32 changes: 32 additions & 0 deletions torch2trt/converters/sigmoid.py
50 changes: 50 additions & 0 deletions torch2trt/converters/softmax.py
80 changes: 80 additions & 0 deletions torch2trt/converters/split.py
40 changes: 40 additions & 0 deletions torch2trt/converters/stack.py
118 changes: 118 additions & 0 deletions torch2trt/converters/sub.py
52 changes: 52 additions & 0 deletions torch2trt/converters/sum.py
18 changes: 18 additions & 0 deletions torch2trt/converters/tanh.py
22 changes: 22 additions & 0 deletions torch2trt/converters/tensor.py
51 changes: 51 additions & 0 deletions torch2trt/converters/transpose.py
281 changes: 281 additions & 0 deletions torch2trt/converters/unary.py
76 changes: 76 additions & 0 deletions torch2trt/converters/view.py
35 changes: 35 additions & 0 deletions torch2trt/module_test.py
296 changes: 296 additions & 0 deletions torch2trt/plugins/group_norm.cpp
284 changes: 284 additions & 0 deletions torch2trt/plugins/interpolate.cpp
30 changes: 30 additions & 0 deletions torch2trt/plugins/plugins.cpp
152 changes: 152 additions & 0 deletions torch2trt/test.py
Empty file added torch2trt/tests/__init__.py
Empty file.
20 changes: 20 additions & 0 deletions torch2trt/tests/test_contiguous.py
Empty file.
148 changes: 148 additions & 0 deletions torch2trt/tests/torchvision/classification.py
24 changes: 24 additions & 0 deletions torch2trt/tests/torchvision/save_load.py
39 changes: 39 additions & 0 deletions torch2trt/tests/torchvision/segmentation.py
727 changes: 727 additions & 0 deletions torch2trt/torch2trt.py
63 changes: 63 additions & 0 deletions torch2trt/utils.py