mirror of https://github.com/llvm/torch-mlir
5a47f92390
This commit adds lowering of `aten.squeeze.dim` op into `linalg.TensorCollapseShape` op. Here, the dim(th) dimension of the input tensor is not supposed to be dynamic. Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com> |
||
---|---|---|
.. | ||
AdjustCallingConventions.cpp | ||
CMakeLists.txt | ||
DecomposeComplexOps.cpp | ||
GlobalizeObjectGraph.cpp | ||
InlineGlobalSlots.cpp | ||
MaximizeValueSemantics.cpp | ||
PassDetail.h | ||
Passes.cpp | ||
PrepareForGlobalizeObjectGraph.cpp | ||
ReduceOpVariants.cpp | ||
RefinePublicReturn.cpp | ||
RefineTypes.cpp |