torch-mlir/lib/Dialect/Torch
Prashant Kumar ed9bd556b3 Fix bug for aten_nll_loss op in the refine types pass
The check for `self.hasSizes` was missing before performing `.size()`
operation.
2022-02-17 19:02:12 +05:30
..
IR LLVM bump 2022-02-16 15:28:13 -05:00
Transforms Fix bug for aten_nll_loss op in the refine types pass 2022-02-17 19:02:12 +05:30
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