[PATCH] D77360: [mlir][Linalg] Add a linalg.tensor_reshape to operate on tensors
Lei Zhang via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Mon Apr 6 12:31:06 PDT 2020
antiagainst added inline comments.
================
Comment at: mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.td:140
+ let description = [{
+ The `linalg.reshape` op produces a new tensor whose sizes are a
+ reassociation of the original `src`.
----------------
s/sizes/dimensions/ ?
================
Comment at: mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.td:143
- let extraClassDeclaration = [{
- static StringRef getReassociationAttrName() { return "reassociation"; }
- MemRefType getViewType() { return view().getType().cast<MemRefType>(); }
+ A reassociation is defined as a continuous grouping of dimensions and is
+ represented with an affine map array attribute. In the future,
----------------
Nit: I think a reassociation is represented as an affine map attribute and all the reassociations as a whole are represented as the array attribute?
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D77360/new/
https://reviews.llvm.org/D77360
More information about the llvm-commits
mailing list