Searched refs:FusionFitsInBudget (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.h | 81 FusionDecision FusionFitsInBudget(const HloInstruction& instr1,
|
| D | multi_output_fusion.cc | 82 return FusionFitsInBudget(*instr1, *instr2, in LegalToFuse()
154 if (!FusionFitsInBudget(*producer, *consumer, in GetProducerConsumerMultiOutputFusionCandidates()
|
| D | instruction_fusion.cc | 95 !FusionFitsInBudget(*consumer, *producer, in ShouldFuse()
|
| D | horizontal_input_fusion.cc | 141 FusionFitsInBudget(*fusion_anchor, *fused)) { in Run()
|
| D | fusion_merger.cc | 339 FusionDecision fits = FusionFitsInBudget(*fusion, *user); in HandleFusion()
|
| D | gpu_fusible.cc | 409 FusionDecision FusionFitsInBudget(const HloInstruction& instr1, in FusionFitsInBudget() function
|