Home
last modified time | relevance | path

Searched refs:canFoldIntoConsumerOp (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/
H A DTensor.h85 bool canFoldIntoConsumerOp(CastOp castOp);
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp93 bool mlir::tensor::canFoldIntoConsumerOp(CastOp castOp) { in canFoldIntoConsumerOp() function in mlir::tensor
136 if (castOp && tensor::canFoldIntoConsumerOp(castOp)) { in foldTensorCast()
1204 if (!canFoldIntoConsumerOp(castOp)) in matchAndRewrite()
1669 if (!castOp || !canFoldIntoConsumerOp(castOp)) in matchAndRewrite()
1978 if (!tensor::canFoldIntoConsumerOp(castOp)) in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp264 if (castOp && memref::CastOp::canFoldIntoConsumerOp(castOp)) { in foldMemRefCast()
1661 return castOp && canFoldIntoConsumerOp(castOp); in matchAndRewrite()
1673 newOperands.push_back(canFoldIntoConsumerOp(tensorCastOp) in matchAndRewrite()
1680 bool fold = canFoldIntoConsumerOp(tensorCastOp); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp516 bool CastOp::canFoldIntoConsumerOp(CastOp castOp) { in canFoldIntoConsumerOp() function in CastOp
717 if (castOp && memref::CastOp::canFoldIntoConsumerOp(castOp)) { in fold()
1995 if (!CastOp::canFoldIntoConsumerOp(cast)) in matchAndRewrite()
2546 if (!CastOp::canFoldIntoConsumerOp(castOp)) in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp3073 if (castOp && memref::CastOp::canFoldIntoConsumerOp(castOp)) { in foldMemRefCast()
3085 if (castOp && tensor::canFoldIntoConsumerOp(castOp)) { in foldTensorCast()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/
H A DMemRefOps.td374 static bool canFoldIntoConsumerOp(CastOp castOp);