Rob Suderman 25738b8c19 [linalg] Broadcast batch for mask on sdpa lowering (#3824) ... Attention often broadcasts a mask across the batch dimension as masking is usually performed the same across attention heads. Added this materialization to the mask dimensions optionally.		2024-10-31 17:59:24 -07:00
..
TorchConversionToMLProgram	[NFC] Remove unused header files (#3386)	2024-05-30 14:30:36 +08:00
TorchOnnxToTorch	Added support for Maxpool (Autopad) (#3774)	2024-10-23 13:04:50 +00:00
TorchToArith	[TorchToArith] Add a lowering for `torch.add.float_int` (#3594)	2024-08-07 11:55:27 -05:00
TorchToLinalg	[linalg] Fix torch.aten.add of `torch.bool` (#3820)	2024-10-31 17:37:25 -07:00
TorchToSCF	[NFC] Change to *cast instead of .*cast variants (#3405)	2024-05-30 23:45:13 -07:00
TorchToStablehlo	[Torch] support AtenExp2Op (#3832)	2024-10-31 19:14:05 +08:00
TorchToTMTensor	[linalg] Broadcast batch for mask on sdpa lowering (#3824)	2024-10-31 17:59:24 -07:00
TorchToTensor	[NFC] Remove unused header files (#3386)	2024-05-30 14:30:36 +08:00
TorchToTosa	[TOSA] Expand Torch to TOSA legalization coverage (#3827)	2024-10-30 16:26:10 -07:00
Utils	[LINALG] Add complex tensor support for `create[Zero|One]InitTensor` utility (#3777)	2024-10-09 16:15:08 +05:30
CMakeLists.txt	[torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055)	2024-03-26 12:41:40 -07:00
PassDetail.h	Minor fixes for `ConvertTorchConversionToMLProgram`. (#1991)	2023-04-04 09:09:58 -07:00
Passes.cpp	Clang format refresh (#2812)	2024-01-29 12:59:33 -05:00