Lines Matching defs:destinationTensors

390 /// - `destinationTensors` are the init values to use for the outer most loop.
398     ArrayRef<OpFoldResult> tileSizes, ValueRange destinationTensors,
419         rewriter.create<scf::ForOp>(loc, lb, ub, step, destinationTensors,
425     destinationTensors = loop.getRegionIterArgs();
430   if (failed(yieldTiledValuesFn(rewriter, loc, ivs, destinationTensors,
438   assert(tiledResults.size() == destinationTensors.size() &&
444        llvm::zip_equal(tiledResults, destinationTensors, resultOffsets,
469 /// - `destinationTensors` are the init values to use for the outer most loop.
480     ArrayRef<Attribute> mappingVector, ValueRange destinationTensors,
505                                               destinationTensors, mappingAttr);
511                                               destinationTensors, mappingAttr);
516   destinationTensors = forallOp.getRegionOutArgs();
521                          destinationTensors, tiledResults, resultOffsets,
527        llvm::zip_equal(tiledResults, destinationTensors, resultOffsets,
543 /// - `destinationTensors` are the init values to use for the outer most loop.
552     ArrayRef<OpFoldResult> numThreads, ValueRange destinationTensors,
559     return tiledBodyFn(rewriter, loc, ValueRange{}, destinationTensors,
564                                       destinationTensors, tiledBodyFn, loops);
569         destinationTensors, tiledBodyFn, loops);