| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | executor_island_coarsening.cc | 140 for (IslandOp island : old_merged_island.islands) in CoarseningAnalysis() local
212 for (IslandOp island : merged_island.islands) { in GetOperandCandidateToMergeWith() local
263 for (IslandOp island : merged_island.islands) { in GetResultCandidateToMergeWith() local
320 for (IslandOp island : merged_island.islands) in GetNewIslandResultsAndForwardResults() local
323 for (IslandOp island : merged_island.islands) { in GetNewIslandResultsAndForwardResults() local
391 auto move_inner_ops = [block, new_yield_op](IslandOp island) { in MoveInnerOpsToNewIsland()
397 for (IslandOp island : merged_island.islands) move_inner_ops(island); in MoveInnerOpsToNewIsland() local
407 for (IslandOp island : merged_island.islands) in MergeIslands() local
410 for (IslandOp island : merged_island.islands) in MergeIslands() local
430 for (IslandOp island : merged_island.islands) in MergeIslands() local
[all …]
|
| D | graph_pruning.cc | 92 if (auto island = llvm::dyn_cast<IslandOp>(op)) { in VisitOp() local
141 bool GraphPruningPass::ShouldPreserveIsland(IslandOp island) { in ShouldPreserveIsland()
164 auto island = llvm::dyn_cast<IslandOp>(op); in PruneGraph() local
|
| D | executor_tpuv1_island_coarsening.cc | 211 IslandOp CreateMergedIsland(IslandOp island, SmallVector<IslandOp, 16>& islands, in CreateMergedIsland() argument
235 for (IslandOp island : islands) { in CreateMergedIsland() local
254 for (IslandOp island : islands) { in CreateMergedIsland() local
292 IslandOp island = dyn_cast<IslandOp>(*op); in MergeIsland() local
503 IslandOp island = dyn_cast<IslandOp>(*op); in CollectSpecialTpuOps() local
586 IslandOp island = dyn_cast<IslandOp>(island_op); in EraseIdentityWithNoReplicationInfo() local
|
| D | convert_control_to_data_outputs.cc | 87 graph_op.walk([&](IslandOp island) { in CollectChainResources()
245 auto island = builder.create<IslandOp>( in CreateIsland() local
|
| D | cluster_formation.cc | 229 func.walk([&](tf_executor::IslandOp island) { in runOnOperation()
|
| D | tfg-to-tfe.cc | 425 auto island = rewriter.create<tf_executor::IslandOp>( in matchAndRewrite() local
|
| D | shape_inference.cc | 553 if (auto island = dyn_cast<tf_executor::IslandOp>(op)) { in ComputeOutputComponent() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | breakup-islands.cc | 89 auto island = dyn_cast<tf_executor::IslandOp>(&op); in AddStatefulControlFlowDependencies() local
122 if (auto island = dyn_cast<tf_executor::IslandOp>(&item)) { in runOnFunction() local
171 void PopulateEmptyIsland(tf_executor::IslandOp island) { in PopulateEmptyIsland()
197 auto island = builder.create<tf_executor::IslandOp>( in CreateIsland() local
234 tf_executor::IslandOp island, in FindSourcesAndSinksInIsland()
|
| D | tf_functional_to_executor.cc | 80 auto island = builder.create<tf_executor::IslandOp>( in runOnOperation() local
|
| D | export_graphdef.cc | 102 auto island = llvm::dyn_cast<mlir::tf_executor::IslandOp>(op); in GetIslandInnerOpOrSelf() local
340 if (auto island = llvm::dyn_cast<mlir::tf_executor::IslandOp>(inst)) { in AddEdge() local
549 } else if (auto island = in Convert() local
|
| D | import_model.cc | 1645 mlir::Operation* island = node_values_.find(arg_node.node->id())->second; in ConvertFunctionArgAndRets() local
1915 auto island = builder_.create<mlir::tf_executor::IslandOp>( in CreateOperation() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | verify_suitable_for_graph_export.cc | 53 auto island = llvm::dyn_cast<mlir::tf_executor::IslandOp>(op); in VerifyExportSuitable() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 196 } else if (auto island = dyn_cast<tf_executor::IslandOp>(op)) { in BacktrackValue() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 278 IslandOp island = *this; in verify() local
|