torch-mlir/lib/Conversion
Angel Zhang 52be4bdc18
[ONNX] Fix bugs for the `onnx.OneHot` operator (#3334)
This commit fixes the bugs for the `onnx.OneHot` operator by:

1) Converting negative indices to non-negative indices
2) Handling both `int` and `float` types for `off` and `on` values
3) Using the correct result type

It also includes a new unit test.
2024-05-22 08:32:00 -04:00
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [ONNX] Fix bugs for the `onnx.OneHot` operator (#3334) 2024-05-22 08:32:00 -04:00
TorchToArith Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
TorchToLinalg onnx.Resize and aten._interpolate : allow n spatial dims. (#3368) 2024-05-20 13:35:27 -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] Support AtenTrilOp (#3359) 2024-05-20 15:49:24 +08:00
TorchToTMTensor [Torch Dialect] decompose all index_put-like op to aten.index_put.hacked_twin for stricter semantics (#3071) 2024-05-08 22:44:57 +08: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] Standardize the std::is_same competime expression (#3321) 2024-05-10 17:07:37 +08: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