Searched refs:segmentSizes (Results 1 – 4 of 4) sorted by relevance
| /llvm-project/mlir/lib/Dialect/IRDL/ |
| H A D | IRDLLoading.cpp | 58 SmallVectorImpl<int> &segmentSizes) { in getSegmentSizesFromAttr() argument
94 segmentSizes.push_back(segmentSize); in getSegmentSizesFromAttr()
117 SmallVectorImpl<int> &segmentSizes) { in getSegmentSizes() argument
124 variadicities, segmentSizes); in getSegmentSizes()
133 segmentSizes.push_back(1); in getSegmentSizes()
152 segmentSizes.push_back(1); in getSegmentSizes()
162 segmentSizes.push_back(nonSingleSegmentSize); in getSegmentSizes()
174 SmallVectorImpl<int> &segmentSizes) { in getOperandSegmentSizes() argument
176 op->getNumOperands(), variadicities, segmentSizes); in getOperandSegmentSizes()
185 SmallVectorImpl<int> &segmentSizes) { in getResultSegmentSizes() argument
[all …]
|
| /llvm-project/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 735 SmallVector<int32_t, 11> segmentSizes(11, 1); in getBlockSize()
736 segmentSizes.front() = asyncDependencies.size(); in getBlockSize()
737 segmentSizes.back() = dynamicSharedMemorySize ? 1 : 0; in getBlockSize()
738 segmentSizes[7] = clusterSizeX ? 1 : 0; in getBlockSize()
739 segmentSizes[8] = clusterSizeY ? 1 : 0;
740 segmentSizes[9] = clusterSizeZ ? 1 : 0; in getClusterIds()
742 builder.getDenseI32ArrayAttr(segmentSizes)); in getClusterIds()
1048 SmallVector<int32_t, 11> segmentSizes(11, 1); in matchAndRewrite()
1049 segmentSizes.front() = asyncDependencies.size(); in matchAndRewrite()
1052 segmentSizes[ in matchAndRewrite()
706 SmallVector<int32_t, 11> segmentSizes(11, 1); build() local
1019 SmallVector<int32_t, 11> segmentSizes(11, 1); parse() local
[all...] |
| /llvm-project/mlir/include/mlir/IR/ |
| H A D | ValueRange.h | 58 OperandRangeRange split(DenseI32ArrayAttr segmentSizes) const;
172 MutableOperandRangeRange split(NamedAttribute segmentSizes) const;
|
| /llvm-project/mlir/lib/IR/ |
| H A D | OperationSupport.cpp | 392 OperandRangeRange OperandRange::split(DenseI32ArrayAttr segmentSizes) const { in getBeginOperandIndex()
393 return OperandRangeRange(*this, segmentSizes); in getBeginOperandIndex()
510 MutableOperandRange::split(NamedAttribute segmentSizes) const { in operator MutableArrayRef()
511 return MutableOperandRangeRange(*this, segmentSizes); in operator MutableArrayRef()
|