Vivek Khandelwal 4a0bed0ce0 [ONNX] Add training mode support for BatchNormalization op (#3597) ... This commit extends the OnnxToTorch lowering for BatchNormalization op for supporting the case when training=True. Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>		2024-08-14 10:46:38 +05:30
..
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	[TorchToLinalg] Fix torch.aten.remainder for negative operands (#3581)	2024-08-13 21:17:21 +05:30
TorchToSCF	[NFC] Change to *cast instead of .*cast variants (#3405)	2024-05-30 23:45:13 -07:00
TorchToStablehlo	[stablehlo] support dynamic-shaped index in stablehlo conversion for aten.index-like ops (#3322)	2024-08-01 10:41:09 +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	[LINALG] Added support for conversion from float to complex. (#3595)	2024-08-07 12:36:48 +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