torch-mlir/projects/pt1/python/torch_mlir_e2e_test
Rob Suderman 0723584936
[torch] Add folder for torch.aten.*.Scalar comparisons (#3000)
This folds small version of the tensor-scalar comparison operators as
they are commonly used for shape computations. This includes le, lt, ge,
gt, eq, and ne.
2024-03-08 13:44:00 -08:00
..
configs [onnx] Add testing using the `onnx` compilation using torch tests (#2795) 2024-02-15 10:17:13 -08:00
debug Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
linalg_on_tensors_backends [torch-mlir][sparse] inline sparse helper methods (#2918) 2024-02-16 20:56:42 -08:00
onnx_backends [torch] Make torch.aten.unflatten lower directly to linalg (#2971) 2024-03-04 10:17:42 -08:00
stablehlo_backends [bazel] fix bazel with stablehlo refbackend and fix some typo (#2911) 2024-02-16 01:38:13 +08:00
test_suite [torch] Add folder for torch.aten.*.Scalar comparisons (#3000) 2024-03-08 13:44:00 -08:00
tosa_backends Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
__init__.py Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
annotations.py Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
framework.py [torch-mlir][NFC] remove trailing whitespace (#2936) 2024-02-20 11:23:14 -08:00
registry.py Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
reporting.py Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
utils.py Rename torch_mlir.compile APIs and introduce FX based analogs (#2842) 2024-02-06 19:07:59 -08:00