Home
last modified time | relevance | path

Searched refs:tensorCastOp (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1672 auto tensorCastOp = opOperand->get().getDefiningOp<tensor::CastOp>(); in matchAndRewrite() local
1673 newOperands.push_back(canFoldIntoConsumerOp(tensorCastOp) in matchAndRewrite()
1674 ? tensorCastOp.getSource() in matchAndRewrite()
1679 auto tensorCastOp = opOperand->get().getDefiningOp<tensor::CastOp>(); in matchAndRewrite() local
1680 bool fold = canFoldIntoConsumerOp(tensorCastOp); in matchAndRewrite()
1681 newOperands.push_back(fold ? tensorCastOp.getOperand() in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp2016 auto tensorCastOp = in matchAndRewrite() local
2018 if (!tensorCastOp) in matchAndRewrite()
2021 tensorCastOp.getDest().getType())) in matchAndRewrite()
2025 padTensorOp.getLoc(), tensorCastOp.getDest().getType(), in matchAndRewrite()
2032 rewriter.replaceOp(tensorCastOp, replacementOp.getResult()); in matchAndRewrite()