torch-mlir/test/Conversion
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 Bump LLVM and StableHLO (#2598) 2023-11-28 22:12:24 -08:00
TorchOnnxToTorch [ONNX] Add training mode support for BatchNormalization op (#3597) 2024-08-14 10:46:38 +05:30
TorchToArith [torch-mlir] bump stablehlo/llvm version (#3471) 2024-06-18 16:59:53 -07:00
TorchToLinalg [TorchToLinalg] Fix possible OOB access in Interpolate lowering (#3570) 2024-08-02 13:55:37 -05:00
TorchToSCF TorchToLinalg: Try folding shape computations to keep static shapes when possible (#3475) 2024-06-27 08:43:10 +02:00
TorchToStablehlo [Torch] enhance fold of aten.squeeze.dim (#3558) 2024-07-24 14:13:48 +08:00
TorchToTensor [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00
TorchToTosa [Tosa] Conversion from torch.__interpolate to tosa.resize() (#3488) 2024-06-26 09:10:14 -07:00