Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Transforms/
DMemRefDataFlowOpt.cpp205 Operation *defOp = memref.getDefiningOp(); in runOnFunction() local
206 if (!defOp || !isa<AllocOp>(defOp)) in runOnFunction()
218 defOp->erase(); in runOnFunction()
/external/llvm-project/mlir/lib/Dialect/Linalg/Analysis/
DDependenceAnalysis.cpp42 Operation *defOp = v.getDefiningOp(); in find() local
43 if (!defOp) in find()
46 if (isa<TensorToMemrefOp>(defOp)) in find()
49 if (auto memEffect = dyn_cast<MemoryEffectOpInterface>(defOp)) { in find()
63 if (auto viewLikeOp = dyn_cast<ViewLikeOpInterface>(defOp)) { in find()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dpasses.cc102 mlir::Operation* defOp = memref.getDefiningOp(); in SearchAllocOp() local
103 while (auto subviewOp = mlir::dyn_cast_or_null<mlir::SubViewOp>(defOp)) { in SearchAllocOp()
104 defOp = subviewOp.source().getDefiningOp(); in SearchAllocOp()
106 return HasEffectsOnValue<mlir::MemoryEffects::Allocate>(memref, defOp) in SearchAllocOp()
107 ? defOp in SearchAllocOp()
/external/llvm-project/mlir/include/mlir/IR/
DMatchers.h158 if (auto defOp = op->getOperand(idx).getDefiningOp()) in matchOperandOrValueAtIndex() local
159 return matcher.match(defOp); in matchOperandOrValueAtIndex()
/external/llvm-project/mlir/lib/Analysis/
DLiveness.cpp189 if (Operation *defOp = value.getDefiningOp()) in resolveLiveness() local
190 currentBlock = defOp->getBlock(); in resolveLiveness()
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DConvertLaunchFuncToRuntimeCalls.cpp448 if (auto *defOp = token.getDefiningOp()) { in matchAndRewrite() local
449 rewriter.setInsertionPointAfter(defOp); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Affine/IR/
DAffineOps.cpp137 if (auto *defOp = value.getDefiningOp()) in isValidDim() local
138 return isValidDim(value, getAffineScope(defOp)); in isValidDim()
236 if (auto *defOp = value.getDefiningOp()) in isValidSymbol() local
237 return isValidSymbol(value, getAffineScope(defOp)); in isValidSymbol()
262 auto *defOp = value.getDefiningOp(); in isValidSymbol() local
263 if (!defOp) { in isValidSymbol()
274 if (matchPattern(defOp, m_Constant(&operandCst))) in isValidSymbol()
278 if (auto applyOp = dyn_cast<AffineApplyOp>(defOp)) in isValidSymbol()
282 if (auto dimOp = dyn_cast<DimOp>(defOp)) in isValidSymbol()
/external/llvm-project/mlir/test/lib/Dialect/Test/
DTestPatterns.cpp368 auto *defOp = operands[0].getDefiningOp(); in matchAndRewrite() local
369 if (auto packerOp = llvm::dyn_cast_or_null<TestCastOp>(defOp)) { in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorOps.cpp1781 auto defOp = extractStridedSliceOp.vector().getDefiningOp(); in matchAndRewrite() local
1782 auto constantMaskOp = dyn_cast_or_null<ConstantMaskOp>(defOp); in matchAndRewrite()
3008 auto defOp = operand.getDefiningOp(); in matchAndRewrite() local
3009 maskDimSizes.push_back(cast<ConstantIndexOp>(defOp).getValue()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/StandardOps/IR/
DOps.cpp328 auto *defOp = alloc.getOperand(dynamicDimPos).getDefiningOp(); in matchAndRewrite() local
329 if (auto constantIndexOp = dyn_cast_or_null<ConstantIndexOp>(defOp)) { in matchAndRewrite()
4273 auto *defOp = viewOp.sizes()[dynamicDimPos].getDefiningOp(); in matchAndRewrite() local
4274 if (auto constantIndexOp = dyn_cast_or_null<ConstantIndexOp>(defOp)) { in matchAndRewrite()