Lines Matching defs:tileShape
347 static SmallVector<int64_t> padTileShapeToSize(ArrayRef<int64_t> tileShape,
349 assert(tileShape.size() <= paddedSize &&
350 "expected tileShape to <= paddedSize");
351 if (tileShape.size() == paddedSize)
352 return to_vector(tileShape);
353 SmallVector<int64_t> result(paddedSize - tileShape.size(), 1);
354 llvm::append_range(result, tileShape);
359 ArrayRef<int64_t> shape, ArrayRef<int64_t> tileShape,
361 : tileShape(padTileShapeToSize(tileShape, shape.size())),
366 mlir::computeShapeRatio(shape, tileShape);
382 return computeElementwiseMul(tileCoords, tileShape);
392 getAffineConstantExprs(tileShape, ctx));