Lines Matching defs:dimension

112                                                unsigned dimension,
118   assert(dimension < op.getNumLoops() && "dimension overflow");
121   int64_t loopRange = op.getStaticLoopRanges()[dimension];
163                                          unsigned dimension,
170   // Bail out on dimension overflow.
171   if (dimension >= numLoops)
183   // Find the trip count of the iteration space dimension for which the tile
186                                                     loopRanges[dimension].size);
242 mlir::linalg::computeStaticMultiTileSizes(LinalgOp op, unsigned dimension,
248   assert(dimension < op.getNumLoops() && "dimension overflow");
251   int64_t tripCount = op.getStaticLoopRanges()[dimension];
269                                     unsigned dimension, OpFoldResult targetSize,
271   // Bail out on dimension overflow.
272   if (dimension >= op.getNumLoops())
286   // Find the trip count of the iteration space dimension for which the tile
295       getValueOrCreateConstantIndexOp(b, op.getLoc(), loopRanges[dimension]);
318   // For example, for iteration dimension size of 15 and the target size 8 it is
320   // original space dimension.
631         op, "only support ops with one reduction dimension.");
639         op, "reduction dimension must be mapped to threads");
807   // dimension. This convention is significantly simpler to handle instead of