Searched defs:side_effect_analysis (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | tpu_extract_head_tail_outside_compilation.cc | 134 const TF::SideEffectAnalysis& side_effect_analysis, in FindOutsideCompiledOpsAtHead() 207 OpBuilder* builder, const TF::SideEffectAnalysis& side_effect_analysis, in LiftHeadOutsideCompiledOps() 230 const TF::SideEffectAnalysis& side_effect_analysis, in FindOutsideCompiledOpsAtTailAndClusterResults() 357 OpBuilder* builder, const TF::SideEffectAnalysis& side_effect_analysis, in LiftTailOutsideCompiledOps() 463 auto& side_effect_analysis = getAnalysis<TF::SideEffectAnalysis>(); in runOnOperation() local
|
D | merge_control_flow.cc | 96 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in SafeToMerge() 192 const std::unique_ptr<TF::SideEffectAnalysis>& side_effect_analysis) { in CanAddToIfSegment() 231 const std::unique_ptr<TF::SideEffectAnalysis>& side_effect_analysis) { in FindLastIfInSegment() 243 const std::unique_ptr<TF::SideEffectAnalysis>& side_effect_analysis) { in GetMoveOpsBetweenTwoIfRegions() 291 const std::unique_ptr<TF::SideEffectAnalysis>& side_effect_analysis) { in GetMoveOpList() 328 const std::unique_ptr<TF::SideEffectAnalysis>& side_effect_analysis, in GenerateSegmentsPerIfGroups() 601 auto side_effect_analysis = std::make_unique<TF::SideEffectAnalysis>(module); in OptimizeIfRegions() local
|
D | tpu_cluster_formation.cc | 196 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in hasOpClusterControlDependency() 239 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in CollectClusterSuccessorOps() 569 Block* block, const TF::SideEffectAnalysis::Info& side_effect_analysis) { in FormClustersInBlock() 653 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in FormClustersInFunction() 707 auto& side_effect_analysis = getAnalysis<TF::SideEffectAnalysis>(); in runOnOperation() local
|
D | convert_control_to_data_outputs.cc | 82 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in CollectChainResources() 378 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in ConvertControlToDataOutputs() 455 TF::SideEffectAnalysis side_effect_analysis(module); in runOnOperation() local
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
D | breakup-islands.cc | 102 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in runOnFunction() 235 const TF::SideEffectAnalysis::Info& side_effect_analysis) { in FindSourcesAndSinksInIsland() 270 const TF::SideEffectAnalysis::Info& side_effect_analysis, in BreakUpIsland()
|
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
D | lower_saved_model.cc | 214 const mlir::TF::SideEffectAnalysis::Info &side_effect_analysis, in HoistInvariantOpsInFunction() 343 mlir::TF::SideEffectAnalysis side_effect_analysis(module); in HoistInvariantOps() local
|
D | corert_converter.cc | 39 const mlir::TF::SideEffectAnalysis::Info *side_effect_analysis) in CoreRTConverter()
|
D | tf_to_tfrt.cc | 1659 const mlir::TF::SideEffectAnalysis::Info &side_effect_analysis, in runOnFunction() 1700 const auto &side_effect_analysis = in runOnOperation() local
|