Searched refs:island_body (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_executor_to_functional.cc | 52 auto& island_body = island_op.GetBody().getOperations(); in LiftIslandOpInnerOpsFromGraph() local
53 parent_block->getOperations().splice(graph_position, island_body, in LiftIslandOpInnerOpsFromGraph()
54 island_body.begin(), in LiftIslandOpInnerOpsFromGraph()
55 std::prev(island_body.end())); in LiftIslandOpInnerOpsFromGraph()
|
| D | breakup-islands.cc | 187 auto island_body = island.GetBody().without_terminator(); in FindSourcesAndSinksInIsland() local
188 for (Operation& sub_op : island_body) { in FindSourcesAndSinksInIsland()
215 auto island_body = island_op.GetBody().without_terminator(); in BreakUpIsland() local
217 if (island_body.empty()) { in BreakUpIsland()
263 for (auto& sub_op : llvm::make_early_inc_range(island_body)) { in BreakUpIsland()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | executor_tpuv1_island_coarsening.cc | 215 Block& island_body = new_island.GetBody(); in MergeIsland() local
219 wrapped_op.moveBefore(&island_body, island_body.end()); in MergeIsland()
230 OpBuilder::atBlockEnd(&island_body) in MergeIsland()
|
| D | executor_island_coarsening.cc | 222 auto& island_body = island.GetBody().getOperations(); in MoveInnerOpsToNewIsland() local
223 block->getOperations().splice(new_yield_op->getIterator(), island_body, in MoveInnerOpsToNewIsland()
224 island_body.begin(), in MoveInnerOpsToNewIsland()
225 std::prev(island_body.end())); in MoveInnerOpsToNewIsland()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 1006 auto &island_body = island_op.GetBody().getOperations(); in matchAndRewrite() local
1009 operation->getIterator(), island_body, island_body.begin(), in matchAndRewrite()
1010 std::prev(island_body.end())); in matchAndRewrite()
|