| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestDynamicPipeline.cpp | 39 << getOperation()->getName() << "\n"; in runOnOperation()
44 auto symbolOp = dyn_cast<SymbolOpInterface>(getOperation()); in runOnOperation()
46 getOperation()->emitWarning() in runOnOperation()
58 getOperation()->getName().getIdentifier(), in runOnOperation()
65 if (getOperation()->getParentOp()) in runOnOperation()
66 if (!failed(runPipeline(*pm, getOperation()->getParentOp()))) in runOnOperation()
73 getOperation()->walk([&](Operation *op) { in runOnOperation()
74 if (op == getOperation() || !op->isKnownIsolatedFromAbove()) in runOnOperation()
83 if (failed(runPipeline(*pm, getOperation()))) in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | executor_tpuv1_inline_tpu_island.cc | 52 SymbolTable symbol_table(getOperation()); in runOnOperation()
57 auto walk_result = getOperation().walk([&](TF::PartitionedCallOp call_op) { in runOnOperation()
62 << "Found call to inline: " << *call_op.getOperation() << "\n"); in runOnOperation()
64 auto call_interface = cast<CallOpInterface>(call_op.getOperation()); in runOnOperation()
69 cast<CallableOpInterface>(called_func.getOperation()), in runOnOperation()
84 if (!symbol_table.lookupSymbolIn(getOperation(), func_op.getName())) { in runOnOperation()
85 nested_block.getOperations().remove(func_op.getOperation()); in runOnOperation()
86 symbol_table.insert(func_op.getOperation()); in runOnOperation()
|
| D | executor_tpuv1_outline_tpu_island.cc | 57 callee.getOperation()->getBlock()->getOperations().remove( in MoveFuncOp()
58 callee.getOperation()); in MoveFuncOp()
65 SymbolTable symbol_table(getOperation()); in runOnOperation()
70 ModuleOp outlined_module = ModuleOp::create(getOperation().getLoc()); in runOnOperation()
71 outlined_module->setAttrs(getOperation()->getAttrDictionary()); in runOnOperation()
80 getOperation().walk([&](TF::TPUReplicateMetadataOp replicate_op) { in runOnOperation()
|
| D | merge_control_flow.cc | 75 if (source.getOperation()->getBlock() != in SafeToMerge()
76 destination.getOperation()->getBlock()) in SafeToMerge()
78 assert(destination.getOperation()->isBeforeInBlock(source.getOperation())); in SafeToMerge()
93 for (auto* user : destination.getOperation()->getUsers()) { in SafeToMerge()
134 if (dependency->getBlock() != destination.getOperation()->getBlock()) in SafeToMerge()
216 builder.setInsertionPoint(source.getOperation()); in CreateMergedIf()
238 MoveResultsAfter(destination.getOperation(), new_if_op.getOperation(), in CreateMergedIf()
|
| D | device_index_selector.cc | 54 FuncOp func = getOperation(); in runOnOperation()
63 for (auto use : op.getOperation()->getUsers()) { in runOnOperation()
71 op.replaceAllUsesWith(constant.getOperation()); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Hoisting.cpp | 91 << *transferRead.getOperation() << "\n"); in hoistRedundantVectorTransfers()
99 cast<LoopLikeOpInterface>(loop.getOperation())))) in hoistRedundantVectorTransfers()
103 LLVM_DEBUG(DBGS() << "Candidate read: " << *transferRead.getOperation() in hoistRedundantVectorTransfers()
128 LLVM_DEBUG(DBGS() << "Candidate: " << *transferWrite.getOperation() in hoistRedundantVectorTransfers()
143 if (!dom.properlyDominates(transferRead.getOperation(), transferWrite)) in hoistRedundantVectorTransfers()
148 if (use.getOwner() == transferRead.getOperation() || in hoistRedundantVectorTransfers()
149 use.getOwner() == transferWrite.getOperation()) in hoistRedundantVectorTransfers()
154 cast<VectorTransferOpInterface>(transferWrite.getOperation()), in hoistRedundantVectorTransfers()
156 transferWriteUse.getOperation()))) in hoistRedundantVectorTransfers()
161 cast<VectorTransferOpInterface>(transferWrite.getOperation()), in hoistRedundantVectorTransfers()
[all …]
|
| D | FusionOnTensors.cpp | 113 unsigned numProducerIndices = isa<IndexedGenericOp>(producer.getOperation()) in generateFusedTensorOpRegion()
116 unsigned numConsumerIndices = isa<IndexedGenericOp>(consumer.getOperation()) in generateFusedTensorOpRegion()
120 (isa<IndexedGenericOp>(producer.getOperation()) || in generateFusedTensorOpRegion()
121 isa<IndexedGenericOp>(consumer.getOperation())) in generateFusedTensorOpRegion()
218 if (isa<GenericOp>(producer.getOperation()) && in fuseTensorOpsImpl()
219 isa<GenericOp>(consumer.getOperation())) { in fuseTensorOpsImpl()
231 .getOperation(); in fuseTensorOpsImpl()
244 .getOperation(); in fuseTensorOpsImpl()
261 generateFusedTensorOpRegion(rewriter, fusedOp.getOperation(), producer, in fuseTensorOpsImpl()
340 if (isa<GenericOp>(op.getOperation())) in createLinalgOpOfSameType()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransferOpTransforms.cpp | 93 LLVM_DEBUG(DBGS() << "Candidate for dead store: " << *write.getOperation() in deadStoreOp()
98 if (user == write.getOperation()) in deadStoreOp()
117 cast<VectorTransferOpInterface>(write.getOperation()), in deadStoreOp()
118 cast<VectorTransferOpInterface>(read.getOperation()))) in deadStoreOp()
143 LLVM_DEBUG(DBGS() << "Found dead store: " << *write.getOperation() in deadStoreOp()
145 opToErase.push_back(write.getOperation()); in deadStoreOp()
162 LLVM_DEBUG(DBGS() << "Candidate for Forwarding: " << *read.getOperation() in storeToLoadForwarding()
173 cast<VectorTransferOpInterface>(write.getOperation()), in storeToLoadForwarding()
174 cast<VectorTransferOpInterface>(read.getOperation()))) in storeToLoadForwarding()
211 LLVM_DEBUG(DBGS() << "Forward value from " << *lastwrite.getOperation() in storeToLoadForwarding()
[all …]
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | OpDefinition.h | 100 operator bool() { return getOperation() != nullptr; }
109 Operation *getOperation() { return state; } in getOperation() function
112 Dialect *getDialect() { return getOperation()->getDialect(); } in getDialect()
115 Region *getParentRegion() { return getOperation()->getParentRegion(); } in getParentRegion()
119 Operation *getParentOp() { return getOperation()->getParentOp(); } in getParentOp()
123 return getOperation()->getParentOfType<OpTy>(); in getParentOfType()
129 return getOperation()->getParentWithTrait<Trait>(); in getParentWithTrait()
133 MLIRContext *getContext() { return getOperation()->getContext(); } in getContext()
265 return lhs.getOperation() == rhs.getOperation();
268 return lhs.getOperation() != rhs.getOperation();
[all …]
|
| D | FunctionSupport.h | 137 Region &getBody() { return this->getOperation()->getRegion(0); } in getBody()
191 return this->getOperation()->template getAttrOfType<TypeAttr>( in getTypeAttr()
271 ::mlir::impl::eraseFunctionArguments(this->getOperation(), argIndices, in eraseArguments()
283 ::mlir::impl::eraseFunctionResults(this->getOperation(), resultIndices, in eraseResults()
300 return ::mlir::impl::getArgAttrs(this->getOperation(), index); in getArgAttrs()
342 setArgAttr(index, Identifier::get(name, this->getOperation()->getContext()), in setArgAttr()
363 return ::mlir::impl::getResultAttrs(this->getOperation(), index); in getResultAttrs()
406 Identifier::get(name, this->getOperation()->getContext()), in setResultAttr()
426 return ::mlir::impl::getArgAttrDict(this->getOperation(), index); in getArgAttrDict()
441 return ::mlir::impl::getResultAttrDict(this->getOperation(), index); in getResultAttrDict()
[all …]
|
| D | SymbolInterfaces.td | 39 return mlir::SymbolTable::getSymbolName(this->getOperation());
45 this->getOperation()->setAttr(
47 StringAttr::get(name, this->getOperation()->getContext()));
53 return mlir::SymbolTable::getSymbolVisibility(this->getOperation());
77 mlir::SymbolTable::setSymbolVisibility(this->getOperation(), vis);
106 return ::mlir::SymbolTable::getSymbolUses(this->getOperation(), from);
116 return ::mlir::SymbolTable::symbolKnownUseEmpty(this->getOperation(),
128 return ::mlir::SymbolTable::replaceAllSymbolUses(this->getOperation(),
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineMemoryOpInterfaces.td | 32 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
43 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
54 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
67 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
78 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
91 return cast<ConcreteOp>(this->getOperation());
111 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
122 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
133 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
146 ConcreteOp op = cast<ConcreteOp>(this->getOperation());
[all …]
|
| /external/llvm-project/llvm/lib/Transforms/InstCombine/ |
| D | InstCombineAtomicRMW.cpp | 27 switch(RMWI.getOperation()) { in isIdempotentRMW()
40 switch(RMWI.getOperation()) { in isIdempotentRMW()
65 switch(RMWI.getOperation()) { in isSaturating()
77 switch(RMWI.getOperation()) { in isSaturating()
109 RMWI.getOperation() != AtomicRMWInst::Xchg) { in visitAtomicRMWInst()
121 if (RMWI.getOperation() == AtomicRMWInst::Xchg && in visitAtomicRMWInst()
141 RMWI.getOperation() != AtomicRMWInst::Or) { in visitAtomicRMWInst()
145 RMWI.getOperation() != AtomicRMWInst::FAdd) { in visitAtomicRMWInst()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/InstCombine/ |
| D | InstCombineAtomicRMW.cpp | 25 switch(RMWI.getOperation()) { in isIdempotentRMW()
38 switch(RMWI.getOperation()) { in isIdempotentRMW()
63 switch(RMWI.getOperation()) { in isSaturating()
75 switch(RMWI.getOperation()) { in isSaturating()
107 RMWI.getOperation() != AtomicRMWInst::Xchg) { in visitAtomicRMWInst()
119 if (RMWI.getOperation() == AtomicRMWInst::Xchg && in visitAtomicRMWInst()
139 RMWI.getOperation() != AtomicRMWInst::Or) { in visitAtomicRMWInst()
144 RMWI.getOperation() != AtomicRMWInst::FAdd) { in visitAtomicRMWInst()
|
| /external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/ |
| D | ops.cc | 84 .getOperation(); in init_ops()
92 return opb.create<mlir::TF::AddV2Op>(loc, x, y).getOperation(); in init_ops()
103 .getOperation(); in init_ops()
111 return opb.create<mlir::TF::ConstOp>(loc, value).getOperation(); in init_ops()
121 .getOperation(); in init_ops()
130 .getOperation(); in init_ops()
138 return opb.create<mlir::TF::GreaterOp>(loc, x, y).getOperation(); in init_ops()
151 .getOperation(); in init_ops()
159 return opb.create<mlir::TF::LessEqualOp>(loc, x, y).getOperation(); in init_ops()
167 return opb.create<mlir::TF::LessOp>(loc, x, y).getOperation(); in init_ops()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Analysis/ |
| D | DependenceAnalysis.cpp | 102 std::make_pair(en.value().getOperation(), en.index())); in LinalgDependenceGraph()
127 return getDependencesFrom(src.getOperation(), dt); in getDependencesFrom()
142 return getDependencesInto(dst.getOperation(), dt); in getDependencesInto()
165 LinalgOpView{src.getOperation(), srcIndex}, in addDependencesBetween()
166 LinalgOpView{dst.getOperation(), dstIndex}); in addDependencesBetween()
176 LinalgOpView{src.getOperation(), srcIndex}, in addDependencesBetween()
177 LinalgOpView{dst.getOperation(), dstIndex}); in addDependencesBetween()
191 LinalgOpView{src.getOperation(), srcIndex}, in addDependencesBetween()
192 LinalgOpView{dst.getOperation(), dstIndex}); in addDependencesBetween()
202 LinalgOpView{src.getOperation(), srcIndex}, in addDependencesBetween()
[all …]
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopFusionUtils.cpp | 140 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in getFusedLoopNestInsertionPoint()
145 getFirstDependentOpInRange(forOpA.getOperation(), forOpB.getOperation()); in getFusedLoopNestInsertionPoint()
147 getLastDependentOpInRange(forOpA.getOperation(), forOpB.getOperation()); in getFusedLoopNestInsertionPoint()
172 return forOpB.getOperation(); in getFusedLoopNestInsertionPoint()
288 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in canFuseLoops()
322 *srcForOp.getOperation(), *dstForOp.getOperation()); in canFuseLoops()
404 auto *childForOp = forOp.getOperation(); in getLoopNestStats()
463 opCount += getComputeCostHelper(childForOp.getOperation(), stats, in getComputeCostHelper()
524 auto *op = forOp.getOperation(); in buildSliceTripCountMap()
555 return getComputeCostHelper(forOp.getOperation(), stats, in getComputeCost()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Async/Transforms/ |
| D | AsyncRefCountingOptimization.cpp | 125 dropRef->isBeforeInBlock(addRef.getOperation())) in optimizeReferenceCounting()
129 if (cancellable.find(dropRef.getOperation()) != cancellable.end()) in optimizeReferenceCounting()
133 Operation *addRefOp = addRef.getOperation(); in optimizeReferenceCounting()
134 Operation *dropRefOp = dropRef.getOperation(); in optimizeReferenceCounting()
167 cancellable[dropRef.getOperation()] = addRef.getOperation(); in optimizeReferenceCounting()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | trim_functions_tf.cc | 77 for (auto func : getOperation().getOps<FuncOp>()) { in TrimModule()
106 SymbolTable symbol_table = SymbolTable(getOperation()); in Verify()
108 for (auto func : getOperation().getOps<FuncOp>()) { in Verify()
111 return getOperation().emitError() in Verify()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToVulkan/ |
| D | ConvertGPULaunchFuncToVulkanLaunchFunc.cpp | 81 getOperation().walk([this, &done](gpu::LaunchFuncOp op) { in runOnOperation()
92 llvm::make_early_inc_range(getOperation().getOps<gpu::GPUModuleOp>())) in runOnOperation()
96 llvm::make_early_inc_range(getOperation().getOps<spirv::ModuleOp>())) in runOnOperation()
102 OpBuilder builder(getOperation().getBody()->getTerminator()); in declareVulkanLaunchFunc()
151 ModuleOp module = getOperation(); in convertGpuLaunchFunc()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToSPIRV/ |
| D | ConvertGPUToSPIRVPass.cpp | 44 ModuleOp module = getOperation(); in runOnOperation()
52 builder.setInsertionPoint(moduleOp.getOperation()); in runOnOperation()
53 kernelModules.push_back(builder.clone(*moduleOp.getOperation())); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 256 .getOperation(); in ImportInstructionImpl()
372 .getOperation(); in ImportInstructionImpl()
379 .getOperation(); in ImportInstructionImpl()
428 .getOperation(); in ImportInstructionImpl()
446 return scatter_op.getOperation(); in ImportInstructionImpl()
470 return select_scatter_op.getOperation(); in ImportInstructionImpl()
484 .getOperation(); in ImportInstructionImpl()
503 return sort_op.getOperation(); in ImportInstructionImpl()
508 .getOperation(); in ImportInstructionImpl()
526 return op.getOperation(); in ImportInstructionImpl()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOpsInterface.td | 98 return getDimsOfType($_op.getOperation(), getWindowIteratorTypeName(), res);
171 return this->getOperation()->getOperand(i);
212 auto range = this->getOperation()->getOperands();
264 assert(i + this->getOperation()->getNumResults() < $_op.getNumOutputs()
266 return this->getOperation()->getOperand($_op.getNumInputs() + i);
318 for (Type type : this->getOperation()->getResults().getTypes())
331 auto range = this->getOperation()->getOperands();
350 return this->getOperation()->getOperand(i);
362 return $_op.getNumOutputs() - this->getOperation()->getNumResults();
389 this->getOperation()->getNumResults();
[all …]
|
| /external/llvm-project/mlir/test/lib/IR/ |
| D | TestFunc.cpp | 19 auto module = getOperation(); in runOnOperation()
43 auto module = getOperation(); in runOnOperation()
67 auto module = getOperation(); in runOnOperation()
|
| /external/llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| D | LoopSpecialization.cpp | 58 ifOp.getThenBodyBuilder().clone(*op.getOperation(), map); in specializeParallelLoopForUnrolling()
59 ifOp.getElseBodyBuilder().clone(*op.getOperation()); in specializeParallelLoopForUnrolling()
87 ifOp.getThenBodyBuilder().clone(*op.getOperation(), map); in specializeForLoopForUnrolling()
88 ifOp.getElseBodyBuilder().clone(*op.getOperation()); in specializeForLoopForUnrolling()
|