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.h122 HloInstruction::FusionKind ChooseFusionKind(const HloInstruction& producer,
Dinstruction_fusion.cc124 return ChooseFusionKind(*producer, *consumer); in ChooseKind()
Dfusion_merger.cc185 user->shape(), ChooseFusionKind(*instruction, *user), user)); in FuseIntoAllUsers()
Dmulti_output_fusion.cc339 ChooseFusionKind(*producer, *consumer_for_fusion), in DoMultiOutputFusion()
Dgpu_fusible.cc569 HloInstruction::FusionKind ChooseFusionKind(const HloInstruction& /*producer*/, in ChooseFusionKind() function