Searched defs:innerTileSizes (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | DataLayoutPropagation.cpp | 251 SmallVector<OpFoldResult> innerTileSizes; in getOrCreatePackedViewOfOperand() local
654 SmallVector<int64_t> innerTileSizes = packOp.getStaticTiles(); in bubbleUpPackOpThroughCollapseShape() local
901 SmallVector<int64_t> innerTileSizes = unPackOp.getStaticTiles(); pushDownUnPackOpThroughExpandShape() local
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 3970 getPackOpResultTypeShape(ArrayRef<int64_t> sourceShape,ArrayRef<int64_t> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) getPackOpResultTypeShape() argument
3995 getResultShape(OpBuilder & builder,Location loc,ArrayRef<OpFoldResult> sourceDims,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) getResultShape() argument
4033 inferPackedType(RankedTensorType sourceType,ArrayRef<int64_t> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) inferPackedType() argument
4042 createDestinationTensor(OpBuilder & b,Location loc,Value source,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) createDestinationTensor() argument
4351 createDestinationTensor(OpBuilder & b,Location loc,Value source,ArrayRef<OpFoldResult> innerTileSizes,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outerDimsPerm) createDestinationTensor() argument
[all...] |