| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | parallel_execute_to_islands.cc | 91 tf_executor::IslandOp island_op, in ExpandParallelExecuteToIslands() argument
108 builder->setInsertionPoint(island_op); in ExpandParallelExecuteToIslands()
110 island_op.getLoc(), yield.getOperandTypes(), in ExpandParallelExecuteToIslands()
111 island_op.control().getType(), island_op.controlInputs()); in ExpandParallelExecuteToIslands()
121 tf_executor::IslandOp island_op, in CreateIslandsFromParallelExecute() argument
123 OpBuilder builder(island_op); in CreateIslandsFromParallelExecute()
127 ExpandParallelExecuteToIslands(island_op, parallel_execute_op, &builder, in CreateIslandsFromParallelExecute()
140 for (auto result : llvm::zip(island_op.outputs(), parallel_execute_outputs)) in CreateIslandsFromParallelExecute()
145 if (!island_op.control().use_empty()) { in CreateIslandsFromParallelExecute()
149 builder.setInsertionPoint(island_op); in CreateIslandsFromParallelExecute()
[all …]
|
| D | replicate_to_island.cc | 140 tf_executor::IslandOp island_op, tf_device::ReplicateOp replicate_op, in ExpandReplicateIntoReplicas() argument
148 auto control_type = tf_executor::ControlType::get(island_op.getContext()); in ExpandReplicateIntoReplicas()
149 llvm::SmallVector<Value, 8> replica_inputs(island_op.controlInputs()); in ExpandReplicateIntoReplicas()
157 builder.setInsertionPoint(island_op); in ExpandReplicateIntoReplicas()
162 island_op.getLoc(), output_types, control_type, replica_inputs); in ExpandReplicateIntoReplicas()
236 tf_executor::IslandOp island_op, in CreateIslandsFromReplicate() argument
238 OpBuilder builder(island_op); in CreateIslandsFromReplicate()
243 if (failed(ExpandReplicateIntoReplicas(tf_dialect, builder, island_op, in CreateIslandsFromReplicate()
258 for (auto result : llvm::zip(island_op.outputs(), replicas_outputs)) in CreateIslandsFromReplicate()
263 if (!island_op.control().use_empty()) { in CreateIslandsFromReplicate()
[all …]
|
| D | executor_tpuv1_outline_tpu_island.cc | 81 auto island_op = cast<IslandOp>(replicate_op->getParentOp()); in runOnOperation() local
82 if (!island_op || island_op.WrapsSingleOp()) return; in runOnOperation()
83 islands_to_outline.push_back(island_op); in runOnOperation()
86 for (IslandOp island_op : islands_to_outline) { in runOnOperation() local
91 getUsedValuesDefinedAbove(island_op.body(), operands); in runOnOperation()
100 for (Value operand : island_op.GetYield().getOperands()) in runOnOperation()
109 OpBuilder(ctx).create<FuncOp>(island_op.getLoc(), name, func_type); in runOnOperation()
116 YieldOp yield_op = island_op.GetYield(); in runOnOperation()
117 outlined_func.getBody().takeBody(island_op.body()); in runOnOperation()
121 island_op.body().push_back(new Block); in runOnOperation()
[all …]
|
| D | tpu_device_propagation.cc | 168 if (auto island_op = in PropagateDevicesInGraph() local
170 if (!island_op.WrapsSingleOp()) continue; in PropagateDevicesInGraph()
171 op_to_update = &island_op.GetBody().front(); in PropagateDevicesInGraph()
|
| D | shape_inference.cc | 934 if (auto island_op = dyn_cast<tf_executor::IslandOp>(op)) { in InferShapeForNonTFDialectOperation() local
936 island_op.GetYield(), island_op.GetYield().fetches(), op->getResults()); in InferShapeForNonTFDialectOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | breakup-islands.cc | 54 void BreakUpIsland(tf_executor::IslandOp island_op,
211 tf_executor::IslandOp island_op, in BreakUpIsland() argument
215 auto island_body = island_op.GetBody().without_terminator(); in BreakUpIsland()
218 PopulateEmptyIsland(island_op); in BreakUpIsland()
223 if (island_op.WrapsSingleOp()) return; in BreakUpIsland()
226 auto island_control_inputs = llvm::to_vector<4>(island_op.controlInputs()); in BreakUpIsland()
229 for (auto fetch : island_op.GetYield().fetches()) { in BreakUpIsland()
233 } else if (fetch.getDefiningOp()->getParentOp() == island_op) { in BreakUpIsland()
249 island_op.getLoc(), nullptr, island_op); in BreakUpIsland()
255 FindSourcesAndSinksInIsland(island_op, side_effect_analysis); in BreakUpIsland()
[all …]
|
| D | tf_executor_to_functional.cc | 45 auto island_op = llvm::dyn_cast<tf_executor::IslandOp>(op); in LiftIslandOpInnerOpsFromGraph() local
46 if (!island_op) in LiftIslandOpInnerOpsFromGraph()
52 auto& island_body = island_op.GetBody().getOperations(); in LiftIslandOpInnerOpsFromGraph()
59 llvm::zip(island_op.outputs(), island_op.GetYield().fetches())) in LiftIslandOpInnerOpsFromGraph()
|
| D | import_model.cc | 1562 auto island_op = llvm::cast<mlir::tf_executor::IslandOp>(inst); in ConvertFunctionArgAndRets() local
1563 mlir::Operation* inner_op = &island_op.GetBody().front(); in ConvertFunctionArgAndRets()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 986 auto island_op = llvm::cast<IslandOp>(block.front()); in matchAndRewrite() local
987 YieldOp yield_op = island_op.GetYield(); in matchAndRewrite()
995 if (operand.getDefiningOp() != island_op) { in matchAndRewrite()
1006 auto &island_body = island_op.GetBody().getOperations(); in matchAndRewrite()
|