Lines Matching full:dimension

24 /// Creates a part of the given `op` split along the iteration space `dimension`
34                 ValueRange resultOperands, unsigned dimension,
40   sizesCopy[dimension] = size;
41   offsetsCopy[dimension] = offset;
67 linalg::splitOp(RewriterBase &rewriter, TilingInterface op, unsigned dimension,
72   // Bail out on dimension overflow.
73   if (dimension >= iterationSpace.size())
89       {splitPoint, offsets[dimension], sizes[dimension]});
95       {iterationSpace[dimension].offset, iterationSpace[dimension].size,
112       rewriter, op.getLoc(), op, offsets, sizes, destinationTensors, dimension,
113       minSplitPoint, iterationSpace[dimension].offset, firstResults);
127       rewriter, op.getLoc(), d0 + d1, {offsets[dimension], minSplitPoint});
131                       dimension, remainingSize, totalOffset, secondResults);