Jiawei Wu f0f59d0f5b [stablehlo] fix: enhance torch's index-like op lowering to stablehlo's gather/scatter (#3829) ... In torch.index_put like ops, `values` is only required to be broadcastable to `input[indices]`, rather than exact dimension match. This patch fixes the problem by add additional stablehlo.dynamic_broadcast_in_dim before creating stablehlo.scatter op. BTW, this patch also enhance the `getBroadcastResultShape` utility in hlo namespace.		2024-11-06 11:46:48 +08:00
..
TorchConversionToMLProgram	[NFC] Remove unused header files (#3386)	2024-05-30 14:30:36 +08:00
TorchOnnxToTorch	[ONNX] Add training mode support for BatchNormalization op (#3597)	2024-08-14 10:46:38 +05:30
TorchToArith	[TorchToArith] Add a lowering for `torch.add.float_int` (#3594)	2024-08-07 11:55:27 -05:00
TorchToLinalg	[torch] Add `torch.aten.view.dtype` to op list (#3664)	2024-10-10 14:27:15 +08:00
TorchToSCF	[NFC] Change to *cast instead of .*cast variants (#3405)	2024-05-30 23:45:13 -07:00
TorchToStablehlo	[stablehlo] fix: enhance torch's index-like op lowering to stablehlo's gather/scatter (#3829)	2024-11-06 11:46:48 +08:00
TorchToTMTensor	[Torch] add support for aten.scatter_add (#3534)	2024-07-12 09:15:42 +08:00
TorchToTensor	[NFC] Remove unused header files (#3386)	2024-05-30 14:30:36 +08:00
TorchToTosa	[TOSA] Fix Tensor.hacked_twin to support diff size indexes (#3547)	2024-07-30 14:32:05 -07:00
Utils	Added support for integer to complex conversion (#3604)	2024-08-14 18:13:00 +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