mirror of https://github.com/llvm/torch-mlir
9a12a093a6
[onnx] Support `onnx.OneHot` lowering to `torch` Leverage the `aten.onehot` implementation along with `aten.transpose` and `aten.where.scalar`. |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| DefaultDomainAtoF.cpp | ||
| DefaultDomainGtoP.cpp | ||
| DefaultDomainQtoZ.cpp | ||
| OnnxLstmExpander.cpp | ||
| PassDetail.h | ||
| Passes.cpp | ||
| Patterns.cpp | ||
| TorchOnnxToTorch.cpp | ||
| Utils.cpp | ||