torch-mlir/lib/Dialect/Torch/IR
Quinn Dawkins 11a8901078
[MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097)
- Includes a canonicalizer for `aten.add.t`needed for successfully lowering the shape function
 - Only offers support for statically sized index tensors when there is more than one
 - Dynamic shape support remains for single indexing tensors
2022-07-28 19:00:02 -04:00
..
CMakeLists.txt Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00
TorchDialect.cpp Bump LLVM to 889c6f3996769a991a24da957f597e7500d158e7 2022-07-15 13:36:04 -07:00
TorchOps.cpp [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -04:00
TorchOpsODSGenerated.cpp Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00
TorchTypes.cpp Add support for ScalarType::QUInt8 2022-06-29 15:33:28 -07:00
UtilsForODSGenerated.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
UtilsForODSGenerated.h Reduce compilation time for TorchOps.cpp.inc 2022-03-21 14:42:26 -07:00