| /llvm-project/mlir/include/mlir/IR/ |
| H A D | BuiltinTypes.h | 275 scalableDims(other.getScalableDims()) {}
279 ArrayRef<bool> scalableDims = {}) in dropDim()
280 : elementType(elementType), shape(shape), scalableDims(scalableDims) {} in dropDim()
285 scalableDims = newIsScalableDim; in insertDim()
298 if (!scalableDims.empty())
299 scalableDims.erase(pos);
311 return VectorType::get(shape, elementType, scalableDims);
317 CopyOnWriteArrayRef<bool> scalableDims;
352 CopyOnWriteArrayRef<bool> scalableDims; global() variable
|
| H A D | BuiltinDialectBytecode.td | 301 Array<BoolList>:$scalableDims,
307 let cBuilder = "get<$_resultType>(context, shape, elementType, scalableDims)";
|
| H A D | BuiltinTypes.td | 1296 ArrayRefParameter<"bool">:$scalableDims
1301 CArg<"ArrayRef<bool>", "{}">:$scalableDims
1303 // While `scalableDims` is optional, its default value should be
1306 if (scalableDims.empty()) {
1308 scalableDims = isScalableVec;
1310 return $_get(elementType.getContext(), shape, elementType, scalableDims);
|
| /llvm-project/mlir/lib/AsmParser/ |
| H A D | TypeParser.cpp | 485 SmallVector<bool, 4> scalableDims;
486 if (parseVectorDimensionList(dimensions, scalableDims))
494 return getChecked<VectorType>(loc, dimensions, elementType, scalableDims); in parseVectorDimensionList()
498 /// For i-th dimension, `scalableDims[i]` contains either: in parseVectorDimensionList()
508 SmallVectorImpl<bool> &scalableDims) { in parseVectorDimensionList()
520 scalableDims.push_back(scalable);
459 SmallVector<bool, 4> scalableDims; parseVectorType() local
491 parseVectorDimensionList(SmallVectorImpl<int64_t> & dimensions,SmallVectorImpl<bool> & scalableDims) parseVectorDimensionList() argument
|
| H A D | Parser.h | 231 SmallVectorImpl<bool> &scalableDims);
|
| /llvm-project/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | LowerVectorBitCast.cpp | 60 ArrayRef<bool> scalableDims = in matchAndRewrite() local
63 VectorType::get(shape, resultType.getElementType(), scalableDims); in matchAndRewrite()
|
| H A D | LowerVectorMultiReduction.cpp | 233 SmallVector<bool, 2> scalableDims; in matchAndRewrite()
239 scalableDims.push_back(isParallelDimScalable); in matchAndRewrite()
244 scalableDims.push_back(isReductionDimScalable); in matchAndRewrite()
249 std::swap(scalableDims.front(), scalableDims.back()); in matchAndRewrite()
264 scalableDims); in matchAndRewrite()
273 /*scalableDims=*/{isParallelDimScalable}); in matchAndRewrite()
236 SmallVector<bool, 2> scalableDims; matchAndRewrite() local
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | IRTypes.cpp | 625 std::vector<bool> scalableDims; in bindDerived()
627 scalableDims.reserve(rank); in bindDerived()
629 scalableDims.push_back(mlirVectorTypeIsDimScalable(self, i)); in bindDerived()
630 return scalableDims; in bindDerived()
637 std::optional<std::vector<int64_t>> scalableDims, in bindDerived()
639 if (scalable && scalableDims) { in bindDerived()
655 } else if (scalableDims) { in bindDerived()
657 for (int64_t dim : *scalableDims) { in bindDerived()
493 std::vector<bool> scalableDims; bindDerived() local
505 get(std::vector<int64_t> shape,PyType & elementType,std::optional<py::list> scalable,std::optional<std::vector<int64_t>> scalableDims,DefaultingPyLocation loc) get() argument
|
| /llvm-project/mlir/unittests/IR/ |
| H A D | ShapedTypeTest.cpp | 140 SmallVector<bool> scalableDims{true, false, true, false, false}; in TEST()
141 VectorType vectorType = VectorType::get(shape, f32, scalableDims); in TEST()
139 SmallVector<bool> scalableDims{true, false, true, false, false}; TEST() local
|
| /llvm-project/mlir/lib/IR/ |
| H A D | BuiltinTypes.cpp | 219 ArrayRef<bool> scalableDims) { in verify()
230 if (scalableDims.size() != shape.size()) in verify() argument
232 << scalableDims.size() << " and " << shape.size(); in verify()
|
| H A D | AsmPrinter.cpp | 2678 auto scalableDims = vectorTy.getScalableDims(); in printTypeImpl()
2684 if (!scalableDims.empty() && scalableDims[dimIdx]) in printTypeImpl()
2687 if (!scalableDims.empty() && scalableDims[dimIdx])
2611 auto scalableDims = vectorTy.getScalableDims(); printTypeImpl() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Vectorization.cpp | 210 SmallVector<bool> scalableDims; in getCanonicalVecType()
214 scalableDims = in getCanonicalVecType()
218 scalableDims.append(scalableVecDims.begin(), scalableVecDims.end()); in getCanonicalVecType()
221 return VectorType::get(vectorShape, elementType, scalableDims);
3452 lhsEltType, /*scalableDims=*/{false, false, scalableChDim});
3455 /*scalableDims=*/{false, scalableChDim}); in generateNcwConv()
3458 /*scalableDims=*/{false, false, scalableChDim}); in generateNcwConv()
3463 ArrayRef<bool> scalableDims, in generateNcwConv()
3468 VectorType::get(maskShape, rewriter.getI1Type(), scalableDims); in generateNcwConv()
206 SmallVector<bool> scalableDims; getCanonicalVecType() local
3231 __anon729377362e02(ArrayRef<int64_t> maskShape, ArrayRef<bool> scalableDims, Operation *opToMask) depthwiseConv() argument
|
| /llvm-project/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 481 SmallVector<bool> scalableDims; in verify()
491 scalableDims.push_back(sourceScalableDims[dimIdx]);
498 targetShape, getSourceVectorType().getElementType(), scalableDims);
4178 SmallVector<bool> scalableDims = in foldTransferFullMask()
4181 return VectorType::get(maskShape, i1Type, scalableDims); in foldTransferFullMask()
467 SmallVector<bool> scalableDims; verify() local
3995 SmallVector<bool> scalableDims = inferTransferOpMaskType() local
|
| /llvm-project/mlir/include/mlir/Dialect/ArmSME/IR/ |
| H A D | ArmSMEOps.td | 604 "/*scalableDims=*/{true})">,
|