torch-mlir/lib/Dialect/TorchConversion/IR
Yi Zhang 603e068e45 E2e implementation for `aten.cat`,`aten.gather`, `aten.bmm`
Also contains the following changes:
- Remove derefineOp canonicalizer because it's not safe.
- Support for optional tensor and list tensors in reduceOpVariant. This
only works for some special detected and easy to handle cases. For list,
it covers the case list is got from a `ListConstruct`. For optional, it
covers the case optional is constructed from a `DerefineOp`.
- Remove the `inferReturnTypes` for `FromBuiltinTensorOp` because it's
not safe to deduce types from the input. For example, a built-in tensor
of i8 could be converted to si8 or ui8. It's better to let the user
specify the return type explicitly.
2021-09-22 19:15:01 -04:00
..
CMakeLists.txt Add TorchToIREE and factor out TorchConversion dialect. 2021-08-16 15:01:58 -07:00
TorchConversionDialect.cpp Add TorchToIREE and factor out TorchConversion dialect. 2021-08-16 15:01:58 -07:00
TorchConversionOps.cpp E2e implementation for `aten.cat`,`aten.gather`, `aten.bmm` 2021-09-22 19:15:01 -04:00