Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/IR/
DLinalgOps.cpp1696 auto tensorCastOp = v.getDefiningOp<TensorCastOp>(); in matchAndRewrite() local
1698 canFoldIntoConsumerOp(tensorCastOp) ? tensorCastOp.source() : v); in matchAndRewrite()
1705 auto tensorCastOp = v.getDefiningOp<TensorCastOp>(); in matchAndRewrite() local
1706 bool fold = canFoldIntoConsumerOp(tensorCastOp); in matchAndRewrite()
1707 newOperands.push_back(fold ? tensorCastOp.getOperand() : v); in matchAndRewrite()