torch-mlir/include/npcomp/Dialect/Torch/IR
Bryce Arden 1736ff0253 [prim] Add TupleIndex support
I could not find a corresponding ListIndex in prim, which seems to
translate to a __get_attr__ under the hood. I think the reason a tuple
Index op can exist is because Tuple's are supposed to be frozen, where
List operands can be mutable.
2021-03-02 17:28:32 -08:00
..
CMakeLists.txt Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
OpInterfaces.h Add a number of kernels and new patterns. 2020-11-04 14:36:59 -08:00
OpInterfaces.td Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
TorchBase.td Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
TorchDialect.h Expose signature metadata to ops and implement ATenRecognizeKernelsPass pass. 2020-10-26 20:31:45 -07:00
TorchOps.h Add support for prim::Loop op. 2021-03-02 16:01:34 -08:00
TorchOps.td [prim] Add TupleIndex support 2021-03-02 17:28:32 -08:00
TorchTypes.h Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
TorchTypes.td Add support for prim.NumToTensor 2021-02-26 10:16:56 -08:00