Searched refs:side_effect_analysis (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | merge_control_flow.cc | 67 const TF::SideEffectAnalysis::Info& side_effect_analysis);
73 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in SafeToMerge() argument
97 for (auto* successor : side_effect_analysis.DirectControlSuccessors(&op)) { in SafeToMerge()
102 for (auto* successor : side_effect_analysis.DirectControlSuccessors(&op)) { in SafeToMerge()
124 side_effect_analysis.DirectControlSuccessors(dependency)) { in SafeToMerge()
175 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in MoveResultsAfter() argument
184 side_effect_analysis.DirectControlSuccessors(walked_op)) { in MoveResultsAfter()
207 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in CreateMergedIf() argument
239 side_effect_analysis); in CreateMergedIf()
287 Block* block, const TF::SideEffectAnalysis::Info& side_effect_analysis) { in OptimizeIfRegions() argument
[all …]
|
| D | tpu_extract_head_tail_outside_compilation.cc | 131 const TF::SideEffectAnalysis& side_effect_analysis, in FindOutsideCompiledOpsAtHead() argument
133 const auto& analysis = side_effect_analysis.GetAnalysisForFunc( in FindOutsideCompiledOpsAtHead()
202 OpBuilder* builder, const TF::SideEffectAnalysis& side_effect_analysis, in LiftHeadOutsideCompiledOps() argument
206 FindOutsideCompiledOpsAtHead(side_effect_analysis, cluster); in LiftHeadOutsideCompiledOps()
225 const TF::SideEffectAnalysis& side_effect_analysis, in FindOutsideCompiledOpsAtTailAndClusterResults() argument
229 const auto& analysis = side_effect_analysis.GetAnalysisForFunc( in FindOutsideCompiledOpsAtTailAndClusterResults()
348 OpBuilder* builder, const TF::SideEffectAnalysis& side_effect_analysis, in LiftTailOutsideCompiledOps() argument
353 FindOutsideCompiledOpsAtTailAndClusterResults(side_effect_analysis, *cluster, in LiftTailOutsideCompiledOps()
421 auto& side_effect_analysis = getAnalysis<TF::SideEffectAnalysis>(); in runOnOperation() local
436 if (failed(LiftHeadOutsideCompiledOps(&builder, side_effect_analysis, in runOnOperation()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | breakup-islands.cc | 52 const TF::SideEffectAnalysis::Info& side_effect_analysis);
55 const TF::SideEffectAnalysis::Info& side_effect_analysis,
61 FuncOp func, const TF::SideEffectAnalysis::Info& side_effect_analysis) { in runOnFunction() argument
82 BreakUpIsland(island, side_effect_analysis, &new_control_inputs); in runOnFunction()
185 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in FindSourcesAndSinksInIsland() argument
189 auto predecessors = side_effect_analysis.DirectControlPredecessors(&sub_op); in FindSourcesAndSinksInIsland()
212 const TF::SideEffectAnalysis::Info& side_effect_analysis, in BreakUpIsland() argument
255 FindSourcesAndSinksInIsland(island_op, side_effect_analysis); in BreakUpIsland()
265 side_effect_analysis.DirectControlPredecessors(&sub_op); in BreakUpIsland()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ |
| D | BUILD | 822 "analysis/side_effect_analysis.cc",
826 "analysis/side_effect_analysis.h",
|