| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 191 setInsertionPoint(®ion->front(), region->front().begin());
200 setInsertionPoint(op);
206 setInsertionPoint(block, insertPoint);
308 setInsertionPoint(ip.getBlock(), ip.getPoint()); in restoreInsertionPoint()
314 void setInsertionPoint(Block *block, Block::iterator insertPoint) { in setInsertionPoint() function
322 void setInsertionPoint(Operation *op) { in setInsertionPoint() function
323 setInsertionPoint(op->getBlock(), Block::iterator(op)); in setInsertionPoint()
329 setInsertionPoint(op->getBlock(), ++Block::iterator(op)); in setInsertionPointAfter()
348 setInsertionPoint(block, block->begin()); in setInsertionPointToStart()
353 setInsertionPoint(block, block->end()); in setInsertionPointToEnd()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 158 builder.setInsertionPoint(op); in LowerWhile()
243 builder->setInsertionPoint(terminator); in ReplaceTerminator()
261 builder.setInsertionPoint(op); in LowerIfRegion()
265 builder.setInsertionPoint(op); in LowerIfRegion()
271 builder.setInsertionPoint(op); in LowerIfRegion()
291 builder.setInsertionPoint(op); in LowerCaseRegion()
300 builder.setInsertionPoint(op); in LowerCaseRegion()
323 builder.setInsertionPoint(op); in LowerWhileRegion()
363 builder.setInsertionPoint(op); in LowerWhileRegion()
|
| D | legalize_tf_communication.cc | 329 builder.setInsertionPoint(host_compute); in RewriteHostComputeOp()
358 builder.setInsertionPoint(send_to_host); in RewriteSendToHostOp()
370 builder.setInsertionPoint(recv_from_host); in RewriteRecvFromHostOp()
384 builder.setInsertionPoint(call); in RewriteCallOp()
538 builder.setInsertionPoint(terminator); in RewriteControlFlowTerminator()
625 builder.setInsertionPoint(region_if); in ProcessRegionIfOp()
692 builder.setInsertionPoint(region_while); in ProcessRegionWhileOp()
725 builder.setInsertionPoint(terminator); in RewriteFunctionTerminator()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | parallel_execute_to_islands.cc | 102 builder->setInsertionPoint(terminator); in ExpandParallelExecuteToIslands()
108 builder->setInsertionPoint(island_op); in ExpandParallelExecuteToIslands()
149 builder.setInsertionPoint(island_op); in CreateIslandsFromParallelExecute()
173 builder.setInsertionPoint(fetch); in CreateIslandsFromParallelExecute()
|
| D | replicate_to_island.cc | 152 builder.setInsertionPoint(&terminator); in ExpandReplicateIntoReplicas()
157 builder.setInsertionPoint(island_op); in ExpandReplicateIntoReplicas()
267 builder.setInsertionPoint(island_op); in CreateIslandsFromReplicate()
290 builder.setInsertionPoint(fetch); in CreateIslandsFromReplicate()
|
| D | cluster_outlining.cc | 92 builder->setInsertionPoint(cluster_return_op); in BuildFunction()
112 builder->setInsertionPoint(cluster_op); in OutlineCluster()
|
| D | readonly_references_to_resources.cc | 137 builder.setInsertionPoint(variable_v2_op); in runOnFunction()
161 builder.setInsertionPoint(user); in runOnFunction()
|
| D | tpu_resource_partitioning.cc | 98 builder.setInsertionPoint(assign_var); in PartitionResourceReadsWrites()
125 builder.setInsertionPoint(partitioned_input); in PartitionResourceReadsWrites()
|
| D | tpu_extract_head_tail_outside_compilation.cc | 107 before ? builder->setInsertionPoint(op) : builder->setInsertionPointAfter(op); in CreateLaunchForBlock()
316 builder->setInsertionPoint(old_terminator); in UpdateClusterResults()
321 builder->setInsertionPoint(cluster); in UpdateClusterResults()
400 builder->setInsertionPoint(cluster); in RemoveClusterAliasedOutputs()
|
| D | tpu_extract_outside_compilation.cc | 278 builder.setInsertionPoint(&op); in MoveOpsToHost()
296 builder.setInsertionPoint(insertion_point); in MoveOpsToHost()
386 builder.setInsertionPoint(while_op.cond().front().getTerminator()); in DecomposeControlFlow()
438 builder.setInsertionPoint(host_launch_op.GetBody().getTerminator()); in CreateParallelExecuteForOutsideCompilation()
|
| D | layout_optimization.cc | 170 builder.setInsertionPoint(op); in runOnFunction()
271 builder.setInsertionPoint(op); in MoveTransposeBefore()
390 builder.setInsertionPoint(op); in MoveTransposeAfter()
|
| D | rewrite_tpu_embedding_ops.cc | 38 builder->setInsertionPoint(op); in AddOperandAndRewriteAs()
|
| D | tpu_dynamic_layout_pass.cc | 173 builder.setInsertionPoint(execute_launch); in HandleInput()
200 builder.setInsertionPoint(replicate); in HandleReplicatedInputs()
|
| /external/llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| D | Utils.cpp | 34 b.setInsertionPoint(loop); in cloneWithNewYields()
47 b.setInsertionPoint(yield); in cloneWithNewYields()
92 b.setInsertionPoint(ifOp->getParentOfType<FuncOp>()); in outlineIfOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/ |
| D | tfl_to_std.cc | 28 b.setInsertionPoint(op); in ConvertTFLQuantOpsToMlirQuantOps()
46 b.setInsertionPoint(op); in ConvertMlirQuantOpsToTFLQuantOps()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToSPIRV/ |
| D | ConvertGPUToSPIRVPass.cpp | 52 builder.setInsertionPoint(moduleOp.getOperation()); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | raise_custom_ops.cc | 54 builder.setInsertionPoint(op); in runOnFunction()
|
| D | split_merged_operands.cc | 95 builder->setInsertionPoint(op); in DuplicateValueIfNeeded()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | FoldUtils.cpp | 155 builder.setInsertionPoint(&entry, entry.begin()); in getOrCreateConstant()
200 builder.setInsertionPoint(&entry, entry.begin()); in tryToFold()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | AsyncRegionRewriter.cpp | 60 builder.setInsertionPoint(op); in operator ()()
245 builder.setInsertionPoint(it->getBlock(), it); in addAsyncDependencyAfter()
|
| /external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/ |
| D | builders.cc | 39 &mlir::OpBuilder::setInsertionPoint)) in init_builders()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Fusion.cpp | 455 b.setInsertionPoint(consumer.getOperation()); in fuseProducerOfBuffer()
511 b.setInsertionPoint(consumer.getOperation()); in fuseProducerOfTensor()
817 builder.setInsertionPoint(tiledOp); in fuseOperations()
829 builder.setInsertionPoint(fusedOp); in fuseOperations()
859 builder.setInsertionPoint(rootOp); in tileAndFuseLinalgOpsImpl()
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/ |
| D | LowerABIAttributesPass.cpp | 35 builder.setInsertionPoint(funcOp.getOperation()); in createGlobalVarForEntryPointArgument()
118 builder.setInsertionPoint(spirvModule.body().front().getTerminator()); in lowerEntryPointABIAttr()
|
| /external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/ |
| D | ConvertLaunchFuncToLLVMCalls.cpp | 189 rewriter.setInsertionPoint(launchOp); in matchAndRewrite()
240 rewriter.setInsertionPoint(launchOp); in matchAndRewrite()
|
| /external/llvm-project/flang/lib/Lower/ |
| D | FIRBuilder.cpp | 113 setInsertionPoint(module.getBody()->getTerminator()); in createGlobal()
126 setInsertionPoint(module.getBody()->getTerminator()); in createGlobal()
|