torch-mlir/lib/Dialect/Torch/Transforms
Ramiro Leal-Cavazos 9bc606c384
Add support for returning more than one copy of the same tensor (#1228)
One of the simplifications made by the pass `RefinePublicReturn`
currently only happens if the tensor in question only has one
user. However, the current method of checking this does not correctly
handle the case of a user having multiple uses of the same
tensor. This commit makes sure only unique users are considered.
2022-08-18 22:41:45 +00:00
..
AdjustCallingConventions.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
CMakeLists.txt Iteratively run the main simplification pipeline. 2022-08-17 14:54:33 -07:00
DecomposeComplexOps.cpp Add a concept of "backend legal ops". 2022-08-18 11:46:50 -07:00
DropShapeCalculations.cpp torch: allow torch dialect ops after running drop-shape pass (#979) 2022-06-25 07:27:47 -07:00
EraseModuleInitializer.cpp Iteratively run the main simplification pipeline. 2022-08-17 14:54:33 -07:00
GlobalizeObjectGraph.cpp llvm: update tag to 061e0189 (#1180) 2022-08-08 20:17:35 -07:00
InlineGlobalSlots.cpp build: update llvm tag to 2dde4ba6 (#1229) 2022-08-15 23:54:45 -07:00
LowerToBackendContract.cpp Add a concept of "backend legal ops". 2022-08-18 11:46:50 -07:00
MaximizeValueSemantics.cpp Add lowering for aten.to.device (#1107) 2022-08-10 19:24:02 -04:00
PassDetail.h llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Passes.cpp Add a concept of "backend legal ops". 2022-08-18 11:46:50 -07:00
PrepareForGlobalizeObjectGraph.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
ReduceOpVariants.cpp Undo shape lib changes + update function signature of sum + zero (#1035) 2022-07-11 10:56:12 -07:00
RefinePublicReturn.cpp Add support for returning more than one copy of the same tensor (#1228) 2022-08-18 22:41:45 +00:00
RefineTypes.cpp Iteratively run the main simplification pipeline. 2022-08-17 14:54:33 -07:00
ReifyShapeCalculations.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
ShapeLibrary.cpp Clean up shape functions that use `sum_mean_dim` (#1217) 2022-08-18 08:23:43 -07:00
SimplifyShapeCalculations.cpp [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -04:00