torch-mlir/projects/pt1/python/torch_mlir
Rob Suderman 61f0a5facf
[torch] Add an `aten.cat` length-0 canonicalization (#2966)
If an input is length-0 along the dimension of canonicalization we can
remove the tensor from the list
2024-03-01 21:41:12 -08:00
..
_torch_mlir_custom_op_example Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
csrc Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
jit_ir_importer [torch] Add an `aten.cat` length-0 canonicalization (#2966) 2024-03-01 21:41:12 -08:00
_dynamo_fx_importer.py Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
_version.py Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
compiler_utils.py [torch-mlir][NFC] remove trailing whitespace (#2936) 2024-02-20 11:23:14 -08:00
dynamo.py [torch-mlir][NFC] remove trailing whitespace (#2936) 2024-02-20 11:23:14 -08:00
torchscript.py Rename torch_mlir.compile APIs and introduce FX based analogs (#2842) 2024-02-06 19:07:59 -08:00