Searched defs:innerTiles (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 3809 SmallVector<OpFoldResult> innerTiles; global() member
3861 build(OpBuilder & builder,OperationState & state,Value source,Value dest,ArrayRef<int64_t> innerDimsPos,ArrayRef<OpFoldResult> innerTiles,std::optional<Value> paddingValue,ArrayRef<int64_t> outerDimsPerm) build() argument
3900 requirePaddingValue(ArrayRef<int64_t> inputShape,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outputShape,ArrayRef<int64_t> outerDimsPerm,ArrayRef<OpFoldResult> innerTiles) requirePaddingValue() argument
4334 build(OpBuilder & builder,OperationState & state,Value source,Value dest,ArrayRef<int64_t> innerDimsPos,ArrayRef<OpFoldResult> innerTiles,ArrayRef<int64_t> outerDimsPerm) build() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 1523 auto innerTiles = packOp.getStaticInnerTiles(); vectorizeAsTensorPackOp() local
1577 ArrayRef<int64_t> innerTiles = unpackOp.getStaticInnerTiles(); vectorizeAsTensorUnpackOp() local
|