Home
last modified time | relevance | path

Searched refs:OpResult (Results 1 – 25 of 47) sorted by relevance

12

/external/llvm-project/mlir/lib/IR/
DValue.cpp29 unsigned trailingNo = resultNo - OpResult::getMaxInlineResults(); in Value()
39 OpResult result = cast<OpResult>(); in getType()
50 OpResult result = cast<OpResult>(); in setType()
72 if (auto result = dyn_cast<OpResult>()) in getDefiningOp()
110 OpResult result = cast<OpResult>(); in getUseList()
173 Operation *OpResult::getOwner() const { in getOwner()
184 unsigned OpResult::getResultNumber() const { in getResultNumber()
195 unsigned OpResult::getNumInline(unsigned numResults) { in getNumInline()
201 unsigned OpResult::getNumTrailing(unsigned numResults) { in getNumTrailing()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_op_lifting_cleanup.cc80 auto can_eliminate = [&](OpResult &result) -> bool { in EliminateUnusedResults()
88 for (OpResult result : op->getResults()) { in EliminateUnusedResults()
109 for (OpResult result : op->getResults()) { in EliminateUnusedResults()
155 for (OpResult result : llvm::reverse(op->getResults())) { in EliminateUnusedResultsForIfCase()
201 for (OpResult result : llvm::reverse(op.getResults())) { in EliminateUnusedResultsForWhile()
253 for (OpResult result : op->getResults()) { in ForwardCommonArgToOutput()
336 for (OpResult result : op->getResults()) { in CanonicalizeRegionIfCaseCluster()
358 for (OpResult result : llvm::reverse(op->getResults())) { in CanonicalizeRegionIfCaseCluster()
377 for (OpResult result : llvm::reverse(op.getResults())) { in CanonicalizeWhileRegion()
Dtf_device_replication_pass.cc90 OpResult result = v.dyn_cast<OpResult>(); in runOnOperation()
Dlayout_optimization.cc183 OpResult result = op->getResult(res); in runOnFunction()
205 for (OpResult result : op->getResults()) { in MoveTransposeBefore()
262 for (OpResult result : op->getResults()) { in MoveTransposeBefore()
393 OpResult result = op->getResult(idx); in MoveTransposeAfter()
Dremove_vars_in_session_initializer.cc55 if (auto op_result = use.get().dyn_cast<mlir::OpResult>()) { in RecursiveRemove()
/external/llvm-project/mlir/include/mlir/IR/
DOperationSupport.h586 return trailingResultNumber + OpResult::getMaxInlineResults(); in getResultNumber()
767 : public llvm::indexed_accessor_range<ResultRange, Operation *, OpResult,
768 OpResult, OpResult> {
770 using indexed_accessor_range<ResultRange, Operation *, OpResult, OpResult,
771 OpResult>::indexed_accessor_range;
782 static OpResult dereference(Operation *op, ptrdiff_t index);
785 friend llvm::indexed_accessor_range<ResultRange, Operation *, OpResult,
786 OpResult, OpResult>;
DOperation.h278 OpResult getResult(unsigned idx) { return OpResult(this, idx); } in getResult()
289 OpResult getOpResult(unsigned idx) { return getResult(idx); } in getOpResult()
535 for (OpResult result : getOpResults()) in dropAllUses()
583 [](OpResult result) { return result.use_empty(); }); in use_empty()
589 return llvm::any_of(getOpResults(), [block](OpResult result) { in isUsedOutsideOfBlock()
666 unsigned numTrailingResults = OpResult::getNumTrailing(numResults); in prefixAllocSize()
667 unsigned numInlineResults = OpResult::getNumInline(numResults); in prefixAllocSize()
682 getInlineResult(OpResult::getMaxInlineResults() - 1)) - in getTrailingResult()
DValue.h25 class OpResult; variable
313 class OpResult : public Value {
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dshape_inference_utils.h41 tensorflow::shape_inference::InferenceContext&, OpResult)>;
Dattribute_utils.h59 while (auto result = val.dyn_cast<OpResult>()) { in GetValueAsConstant()
Dshape_inference_utils.cc280 auto op_result = op->getOperand(input).dyn_cast<OpResult>(); in InferReturnTypeComponentsForTFOp()
385 OpResult op_result) -> ShapeHandle { in InferReturnTypeComponentsForTFOp()
/external/llvm-project/mlir/include/mlir/Dialect/GPU/
DGPUBase.td97 "OpResult", "getAsyncToken", (ins), [{}], [{
99 return op.asyncToken().template dyn_cast_or_null<OpResult>();
/external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/
Dbasic_classes.cc47 py::class_<mlir::OpResult, mlir::Value>(m, "OpResult"); in init_basic_classes()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.h74 bool PropagateInputToOutput(const Value& operand, const OpResult& result);
Dresource_alias_analysis.cc186 int res_index = value.cast<OpResult>().getResultNumber(); in BacktrackValue()
365 const OpResult& result) { in PropagateInputToOutput()
/external/llvm-project/mlir/lib/Transforms/
DBufferUtils.cpp88 value.isa<OpResult>() && in build()
DNormalizeMemRefs.cpp264 OpResult oldResult = userOp->getResult(resIndex); in updateFunctionSignature()
265 OpResult newResult = newCallOp->getResult(resIndex); in updateFunctionSignature()
DBufferResultsToOutParams.cpp87 for (OpResult result : op.getResults()) { in updateCalls()
/external/tensorflow/tensorflow/compiler/mlir/
Dop_or_arg_name_mapper.cc121 if (auto result = val.dyn_cast<mlir::OpResult>()) { in GetName()
/external/llvm-project/mlir/lib/CAPI/IR/
DIR.cpp501 return unwrap(value).isa<OpResult>(); in mlirValueIsAOpResult()
518 return wrap(unwrap(value).cast<OpResult>().getOwner()); in mlirOpResultGetOwner()
523 unwrap(value).cast<OpResult>().getResultNumber()); in mlirOpResultGetResultNumber()
/external/llvm-project/mlir/lib/Transforms/Utils/
DDialectConversion.cpp399 if (OpResult result = argReplacementValue.dyn_cast<OpResult>()) in materializeLiveConversions()
870 for (OpResult result : repl.first->getResults()) in applyRewrites()
891 Operation *replOp = repl.cast<OpResult>().getOwner(); in applyRewrites()
2100 legalizeErasedResult(Operation *op, OpResult result,
2106 legalizeChangedResultType(Operation *op, OpResult result, Value newValue,
2204 for (OpResult result : repl.first->getResults()) { in finalize()
2265 Operation *op, OpResult result, in legalizeErasedResult()
2284 Operation *op, OpResult result, Value newValue, in legalizeChangedResultType()
DRegionUtils.cpp434 OpResult result = value.cast<OpResult>(); in getOrderOf()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.cc52 mlir::OpResult result = reduction_op->getResult(0); in MatchReductionComputation()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_fuse_linalg.cc140 auto idx = result.dyn_cast<OpResult>().getResultNumber(); in runOnFunction()
/external/llvm-project/mlir/lib/Dialect/SCF/
DSCF.cpp724 void transferBody(Block *source, Block *dest, ArrayRef<OpResult> usedResults, in transferBody()
732 [&](OpResult result) { in transferBody()
742 SmallVector<OpResult, 4> usedResults; in matchAndRewrite()
744 [](OpResult result) { return !result.use_empty(); }); in matchAndRewrite()
753 [](OpResult result) { return result.getType(); }); in matchAndRewrite()

12