Home
last modified time | relevance | path

Searched refs:FusionWouldBeTooLarge (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.h65 bool FusionWouldBeTooLarge(const HloInstruction& instr1,
Dmulti_output_fusion.cc79 return !FusionWouldBeTooLarge(*instr1, *instr2); in LegalToFuse()
148 if (FusionWouldBeTooLarge(*producer, *consumer)) { in GetProducerConsumerMultiOutputFusionCandidates()
Dinstruction_fusion.cc94 if (FusionWouldBeTooLarge(*consumer, *producer, in ShouldFuse()
Dhorizontal_input_fusion.cc131 !FusionWouldBeTooLarge(*fusion_anchor, *fused)) { in Run()
Dfusion_merger.cc284 return FusionWouldBeTooLarge(*fusion, *user); in HandleFusion()
Dgpu_fusible.cc370 bool FusionWouldBeTooLarge(const HloInstruction& instr1, in FusionWouldBeTooLarge() function