| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | parallel_execute_to_islands.cc | 93 tf_executor::IslandOp island_op, in ExpandParallelExecuteToIslands() argument
110 builder->setInsertionPoint(island_op); in ExpandParallelExecuteToIslands()
112 island_op.getLoc(), yield.getOperandTypes(), in ExpandParallelExecuteToIslands()
113 island_op.control().getType(), island_op.controlInputs()); in ExpandParallelExecuteToIslands()
123 tf_executor::IslandOp island_op, in CreateIslandsFromParallelExecute() argument
125 OpBuilder builder(island_op); in CreateIslandsFromParallelExecute()
129 ExpandParallelExecuteToIslands(island_op, parallel_execute_op, &builder, in CreateIslandsFromParallelExecute()
142 for (auto result : llvm::zip(island_op.outputs(), parallel_execute_outputs)) in CreateIslandsFromParallelExecute()
147 if (!island_op.control().use_empty()) { in CreateIslandsFromParallelExecute()
151 builder.setInsertionPoint(island_op); in CreateIslandsFromParallelExecute()
[all …]
|
| D | replicate_to_island.cc | 142 tf_executor::IslandOp island_op, tf_device::ReplicateOp replicate_op, in ExpandReplicateIntoReplicas() argument
150 auto control_type = tf_executor::ControlType::get(island_op.getContext()); in ExpandReplicateIntoReplicas()
151 llvm::SmallVector<Value, 8> replica_inputs(island_op.controlInputs()); in ExpandReplicateIntoReplicas()
159 builder.setInsertionPoint(island_op); in ExpandReplicateIntoReplicas()
164 island_op.getLoc(), output_types, control_type, replica_inputs); in ExpandReplicateIntoReplicas()
238 tf_executor::IslandOp island_op, in CreateIslandsFromReplicate() argument
240 OpBuilder builder(island_op); in CreateIslandsFromReplicate()
245 if (failed(ExpandReplicateIntoReplicas(tf_dialect, builder, island_op, in CreateIslandsFromReplicate()
260 for (auto result : llvm::zip(island_op.outputs(), replicas_outputs)) in CreateIslandsFromReplicate()
265 if (!island_op.control().use_empty()) { in CreateIslandsFromReplicate()
[all …]
|
| D | executor_tpuv1_outline_tpu_island.cc | 82 getOperation().walk([&](IslandOp island_op) { in runOnOperation() argument
83 auto parent_func = island_op->getParentOfType<func::FuncOp>(); in runOnOperation()
90 for (Operation &op : island_op.GetBody().without_terminator()) { in runOnOperation()
93 islands_to_outline.push_back(island_op); in runOnOperation()
101 islands_to_outline.push_back(island_op); in runOnOperation()
108 for (IslandOp island_op : islands_to_outline) { in runOnOperation() local
113 getUsedValuesDefinedAbove(island_op.body(), operands); in runOnOperation()
122 for (Value operand : island_op.GetYield().getOperands()) in runOnOperation()
130 auto outlined_func = OpBuilder(ctx).create<func::FuncOp>(island_op.getLoc(), in runOnOperation()
138 YieldOp yield_op = island_op.GetYield(); in runOnOperation()
[all …]
|
| D | executor_tpuv1_island_coarsening.cc | 113 IslandOp island_op = dyn_cast<IslandOp>(op); in HasControlDependencyWithUnscheduledOp() local
114 if (!island_op) { in HasControlDependencyWithUnscheduledOp()
117 for (Value input : island_op.controlInputs()) { in HasControlDependencyWithUnscheduledOp()
532 auto island_op = llvm::dyn_cast<IslandOp>(*iter); in ExcludeIdentityOp() local
533 if (llvm::dyn_cast_or_null<TF::IdentityOp>(island_op.GetBody().front())) { in ExcludeIdentityOp()
534 if (island_op.outputs().use_empty()) { in ExcludeIdentityOp()
540 collect_output_users_islands(island_op, ops); in ExcludeIdentityOp()
542 collect_input_defining_islands(island_op, ops); in ExcludeIdentityOp()
585 for (Operation& island_op : graph_body) { in EraseIdentityWithNoReplicationInfo()
586 IslandOp island = dyn_cast<IslandOp>(island_op); in EraseIdentityWithNoReplicationInfo()
|
| D | tpu_device_propagation.cc | 171 if (auto island_op = in PropagateDevicesInGraph() local
173 if (!island_op.WrapsSingleOp()) continue; in PropagateDevicesInGraph()
174 op_to_update = &island_op.GetBody().front(); in PropagateDevicesInGraph()
|
| D | shape_inference.cc | 1931 if (auto island_op = dyn_cast<tf_executor::IslandOp>(op)) { in InferShapeForNonTFDialectOperation() local
1933 island_op.GetYield(), island_op.GetYield().fetches(), op->getResults()); in InferShapeForNonTFDialectOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | breakup-islands.cc | 62 void BreakUpIsland(tf_executor::IslandOp island_op,
269 tf_executor::IslandOp island_op, in BreakUpIsland() argument
273 auto island_body = island_op.GetBody().without_terminator(); in BreakUpIsland()
276 PopulateEmptyIsland(island_op); in BreakUpIsland()
281 if (island_op.WrapsSingleOp()) return; in BreakUpIsland()
284 auto island_control_inputs = llvm::to_vector<4>(island_op.controlInputs()); in BreakUpIsland()
287 for (auto fetch : island_op.GetYield().fetches()) { in BreakUpIsland()
291 } else if (fetch.getDefiningOp()->getParentOp() == island_op) { in BreakUpIsland()
307 island_op.getLoc(), nullptr, island_op); in BreakUpIsland()
313 FindSourcesAndSinksInIsland(island_op, side_effect_analysis); in BreakUpIsland()
[all …]
|
| D | tf_executor_to_functional.cc | 46 auto island_op = llvm::dyn_cast<tf_executor::IslandOp>(op); in LiftIslandOpInnerOpsFromGraph() local
47 if (!island_op) in LiftIslandOpInnerOpsFromGraph()
53 auto& island_body = island_op.GetBody().getOperations(); in LiftIslandOpInnerOpsFromGraph()
60 llvm::zip(island_op.outputs(), island_op.GetYield().fetches())) in LiftIslandOpInnerOpsFromGraph()
|
| D | import_model.cc | 1686 auto island_op = llvm::cast<mlir::tf_executor::IslandOp>(inst); in ConvertFunctionArgAndRets() local
1687 mlir::Operation* inner_op = &island_op.GetBody().front(); in ConvertFunctionArgAndRets()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 955 auto island_op = llvm::cast<IslandOp>(block.front()); in matchAndRewrite() local
956 YieldOp yield_op = island_op.GetYield(); in matchAndRewrite()
964 if (operand.getDefiningOp() != island_op) { in matchAndRewrite()
975 auto &island_body = island_op.GetBody().getOperations(); in matchAndRewrite()
|