Home
last modified time | relevance | path

Searched refs:ChooseFusionKind (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.h105 HloInstruction::FusionKind ChooseFusionKind(const HloInstruction& producer,
Dinstruction_fusion.cc130 return ChooseFusionKind(*producer, *consumer); in ChooseKind()
Dfusion_merger.cc301 user->shape(), ChooseFusionKind(*fusion, *user), user)); in HandleFusion()
Dmulti_output_fusion.cc313 ChooseFusionKind(*producer, *consumer_for_fusion), in DoMultiOutputFusion()
Dgpu_fusible.cc519 HloInstruction::FusionKind ChooseFusionKind(const HloInstruction& /*producer*/, in ChooseFusionKind() function