Lines Matching defs:memref

604   /// Build a list of memref.load operations indexed at `(row, col)` indices
608                                       OpFoldResult laneId, Value memref,
618                                       OpFoldResult laneId, Value memref,
622   /// Build a list of memref.store operations indexed at `(row, col)` indices
627                                              OpFoldResult laneId, Value memref,
638       Value memref, IndexCalculator indexFn, ArrayRef<int64_t> vectorShape);
665                                  OpFoldResult laneId, Value memref,
675     auto load = b.create<memref::LoadOp>(loc, memref, ValueRange{row, col});
682     OpBuilder &b, Location loc, OpFoldResult laneId, Value memref,
684   auto loads = buildMemRefLoads(b, loc, laneId, memref, std::move(indexFn));
686   Type elementType = getElementTypeOrSelf(memref.getType());
705     Value memref, const IndexCalculator &indexFn) {
715         b.create<memref::StoreOp>(loc, val, memref, ValueRange{row, col});
723     Value memref, IndexCalculator indexFn, ArrayRef<int64_t> vectorShape) {
736   return buildMemRefStores(b, loc, toStore, laneId, memref, std::move(indexFn));
783          "expected lhs to be a 2D memref");
785          "expected rhs to be a 2D memref");
787          "expected res to be a 2D memref");
865   buildGlobalMemRefDescriptor(TypedValue<MemRefType> memref,
951 HopperBuilder::buildGlobalMemRefDescriptor(TypedValue<MemRefType> memref,
955   Value unrankedMemRef = rewriter.create<memref::CastOp>(
957       UnrankedMemRefType::get(memref.getType().getElementType(),
958                               memref.getType().getMemorySpace()),
959       memref);
961       memref::getMixedSizes(rewriter, loc, memref);
970           MemRefType::Builder(memref.getType())
990   auto mixedSizes = memref::getMixedSizes(rewriter, loc, sharedMemref);
1072            "expected in to be a 2D memref");