Angel Zhang 2e194e13d6 [Torch] Fix bugs for `Torch::AtenOneHotOp` (#3350) ... This PR fixes the bugs for `Torch::AtenOneHotOp` by: 1) Using `Torch::kUnknownSize` as the default value for `numClasses` in the pattern matching stage in `DecomposeAtenOneHotOp` 2) Adding `AtenIntScalarOp` to the patterns in `TorchToArith` 3) Handling both `int` and `float` types for `off` and `on` values in `TorchOnnxToTorch` conversion It also includes: 1) A new test in `TorchToArith/basic.mlir`, for `torch.aten.Int.Scalar`, and 2) A new test in `decompose-complex-ops.mlir`, for `torch.aten.one_hot` **Dependencies** This PR is dependent on #3334.		2024-05-22 17:19:08 +00:00
..
TMTensor	[torch] Rework lowering to tm_tensor.scatter to stop serialization (#2940)	2024-02-27 11:46:57 -08:00
Torch	[Torch] Fix bugs for `Torch::AtenOneHotOp` (#3350)	2024-05-22 17:19:08 +00:00
TorchConversion	Integrate llvm-project at dabdec1001dc368373dd581cf72f37a440873ce3 (#3300)	2024-05-08 14:43:06 -04:00