Searched refs:fusion_node_evaluations_ (Results 1 – 5 of 5) sorted by relevance
107 if (fusion_node_evaluations_.find(consumer) == in ShouldFuse()108 fusion_node_evaluations_.end()) { in ShouldFuse()112 fusion_node_evaluations_.emplace(consumer, in ShouldFuse()115 if (fusion_node_evaluations_.at(consumer).CodeDuplicationTooHigh(producer)) { in ShouldFuse()135 auto evaluation = fusion_node_evaluations_.find(fusion_instruction); in FuseInstruction()136 if (evaluation == fusion_node_evaluations_.end()) { in FuseInstruction()137 evaluation = fusion_node_evaluations_ in FuseInstruction()
43 fusion_node_evaluations_.clear(); in Run()59 fusion_node_evaluations_; variable
44 auto evaluation = fusion_node_evaluations_.find(fusion_instruction); in FuseInstruction()45 if (evaluation == fusion_node_evaluations_.end()) { in FuseInstruction()47 fusion_node_evaluations_ in FuseInstruction()69 if (fusion_node_evaluations_.find(consumer) == in EvaluateEmittedInstructions()70 fusion_node_evaluations_.end()) { in EvaluateEmittedInstructions()71 fusion_node_evaluations_.emplace(consumer, in EvaluateEmittedInstructions()74 return fusion_node_evaluations_.at(consumer).EvaluateEmittedInstructions( in EvaluateEmittedInstructions()80 fusion_node_evaluations_; member in xla::InstructionFusionForTesting
127 if (fusion_node_evaluations_.find(consumer) == in ShouldFuse()128 fusion_node_evaluations_.end()) { in ShouldFuse()132 fusion_node_evaluations_.emplace(consumer, in ShouldFuse()135 if (fusion_node_evaluations_.at(consumer).CodeDuplicationTooHigh( in ShouldFuse()207 auto evaluation = fusion_node_evaluations_.find(fusion_instruction); in FuseInstruction()208 if (evaluation == fusion_node_evaluations_.end()) { in FuseInstruction()209 evaluation = fusion_node_evaluations_ in FuseInstruction()
34 fusion_node_evaluations_.clear(); in Run()50 fusion_node_evaluations_; variable