torch-mlir/lib/Dialect/Torch
Ramiro Leal-Cavazos ba29d4f250
Add operand type invariant to `torch.overwrite.tensor.contents` (#606)
This commit adds the invariant to the op `torch.overwrite.tensor.contents` that
both of its operands have the same shape and size. In order to
maintain the invariant, special handling of this op is added to the
`RefineTypes` pass.
2022-02-22 11:41:46 -08:00
..
IR LLVM bump 2022-02-16 15:28:13 -05:00
Transforms Add operand type invariant to `torch.overwrite.tensor.contents` (#606) 2022-02-22 11:41:46 -08:00
Utils [LINALG] Decompose `aten.batch_norm` into `aten.native_batch_norm` 2022-02-16 23:41:38 +05:30
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00