torch-mlir/lib/Conversion
Vivek Khandelwal 17c3c15131
[ONNX] Add OnnxToTorch lowering for SoftmaxCrossEntropyLoss op (#3278)
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-05-06 17:26:32 +05:30
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [ONNX] Add OnnxToTorch lowering for SoftmaxCrossEntropyLoss op (#3278) 2024-05-06 17:26:32 +05:30
TorchToArith Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
TorchToLinalg [TorchToLinalg] Adds Quantization Support for ConvTranspose (#3240) 2024-04-30 09:23:09 -07:00
TorchToSCF Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3130) 2024-04-11 06:47:35 -07:00
TorchToStablehlo [Stablehlo] Improve the lowering of pool op in stablehlo (#3259) 2024-05-01 00:06:13 +08:00
TorchToTMTensor Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
TorchToTensor Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
TorchToTosa [NFC reformat] Run pre-commit on all files and format misc. 2024-04-27 14:08:09 -07:00
Utils [linalg] Fix bug for conversion of complex dtype (#3269) 2024-05-01 12:06:53 +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