Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_fuse_linalg.cc85 auto definingOp = result.getDefiningOp(); in runOnFunction() local
86 if (!definingOp) { in runOnFunction()
90 if (auto viewLike = dyn_cast<ViewLikeOpInterface>(definingOp)) { in runOnFunction()
98 if (auto tensor_load = dyn_cast<TensorLoadOp>(definingOp)) { in runOnFunction()
106 if (auto tensor_to_memref = dyn_cast<TensorToMemrefOp>(definingOp)) { in runOnFunction()
114 if (auto tensor_cast = dyn_cast<tensor::CastOp>(definingOp)) { in runOnFunction()
123 dyn_cast<RegionBranchOpInterface>(definingOp)) { in runOnFunction()
/external/llvm-project/mlir/lib/Transforms/
DLoopInvariantCodeMotion.cpp88 auto definingOp = value.getDefiningOp(); in moveLoopInvariantCode() local
89 return (definingOp && !!willBeMovedSet.count(definingOp)) || in moveLoopInvariantCode()
DBufferOptimizations.cpp133 Operation *definingOp = allocValue.getDefiningOp(); in hoist() local
134 assert(definingOp && "No defining op"); in hoist()
135 auto operands = definingOp->getOperands(); in hoist()
/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/
DKernelOutlining.cpp92 Operation *definingOp = operand.getDefiningOp(); in extractBeneficiaryOps() local
93 if ((!definingOp || in extractBeneficiaryOps()
94 !extractBeneficiaryOps(definingOp, existingDependencies, in extractBeneficiaryOps()
/external/llvm-project/mlir/lib/Analysis/
DLiveness.cpp359 Operation *definingOp = value.getDefiningOp(); in getStartOperation() local
362 if (isLiveIn(value) || !definingOp) in getStartOperation()
364 return definingOp; in getStartOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor.cc814 Operation *definingOp = token.getDefiningOp(); in Verify() local
815 if (!definingOp) in Verify()
818 auto source = dyn_cast<NextIterationSourceOp>(definingOp); in Verify()
/external/llvm-project/mlir/lib/Dialect/StandardOps/IR/
DOps.cpp1390 Operation *definingOp = memrefOrTensor().getDefiningOp(); in fold() local
1392 if (auto tensorLoadOp = dyn_cast_or_null<TensorLoadOp>(definingOp)) { in fold()
1399 dyn_cast_or_null<DynamicTensorFromElementsOp>(definingOp)) { in fold()
1423 if (auto alloc = dyn_cast_or_null<AllocOp>(definingOp)) in fold()
1427 if (auto view = dyn_cast_or_null<ViewOp>(definingOp)) in fold()
1431 if (auto subview = dyn_cast_or_null<SubViewOp>(definingOp)) { in fold()