Home
last modified time | relevance | path

Searched refs:fusibles (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmulti_output_fusion.cc119 instr_node.fusibles.emplace_back(candidate, profit); in Run()
120 candidate_node.fusibles.emplace_back(instruction, profit); in Run()
212 auto it = fusion_node.fusibles.begin(); in GetNewFusibles()
213 while (it != fusion_node.fusibles.end()) { in GetNewFusibles()
216 it = fusion_node.fusibles.erase(it); in GetNewFusibles()
227 for (const auto& it : fused_node.fusibles) { in GetNewFusibles()
239 fused_node.fusibles.clear(); in GetNewFusibles()
277 candidate_node.fusibles.emplace_back(it.first, profit); in UpdateAfterFuse()
290 candidate_node.fusibles.emplace_back(it.first, profit); in UpdateAfterFuse()
Dmulti_output_fusion.h134 std::list<std::pair<HloInstruction*, int64_t>> fusibles; member
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc504 auto fusibles = fusion_candidates.GetNextSpanOfFusions(); in Run() local
505 if (fusibles.empty()) { in Run()
507 } else if (fusibles.size() == 1) { in Run()
516 for (HloInstruction* instr : fusibles) { in Run()