Lines Matching defs:toTensorOp

254   Value toTensorOp = rewriter.create<bufferization::ToTensorOp>(
256   rewriter.replaceOp(padOp, toTensorOp);
289   maskOp.walk([&](bufferization::ToTensorOp toTensorOp) {
290     if (toTensorOp->getUses().empty())
291       toTensorOps.push_back(toTensorOp.getOperation());
310     auto toTensorOp =
312     assert(toTensorOp && "expected to_tensor op");
313     rewriter.modifyOpInPlace(toTensorOp, [&]() {
314       toTensorOp.setRestrict(true);
315       toTensorOp.setWritable(true);
337   Value toTensorOp = rewriter.create<bufferization::ToTensorOp>(
339   rewriter.replaceOp(allocTensorOp, toTensorOp);
562       auto toTensorOp = rewriter.create<ToTensorOp>(op->getLoc(), alloc);
563       operand->set(toTensorOp);
565         rewriter.modifyOpInPlace(toTensorOp, [&]() {
566           toTensorOp.setRestrict(true);
567           toTensorOp.setWritable(true);
584     auto toTensorOp = resultUse->get().getDefiningOp<ToTensorOp>();
585     assert(toTensorOp && "expected to_tensor op");
586     rewriter.modifyOpInPlace(toTensorOp, [&]() {
587       toTensorOp.setRestrict(true);
588       toTensorOp.setWritable(true);