torch-mlir/lib/Dialect/Torch
zjgarvey 40e762ca42
Adds result types to a prelu decomp (#3098)
This adds explicit result types instead of relying on shape/dtype
computations.

Solves a regression issue with IREE: #3092
2024-04-02 11:41:56 -07:00
..
IR [ONNX] Preliminary Work Towards Supporting QuantizedMLP_basic onnx e2e test (#3089) 2024-04-01 16:21:05 -07:00
Transforms Adds result types to a prelu decomp (#3098) 2024-04-02 11:41:56 -07:00
Utils Adds accumulator types in TorchToLinalg for `AtenMmOp` and `AtenConvolutionOp` (#3027) 2024-03-14 16:40:40 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00