Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc146 const HloInstruction* GetRealHeroForMultiOutputFusion( in GetRealHeroForMultiOutputFusion() function
182 auto* instr_1 = GetRealHeroForMultiOutputFusion(instr1); in ShapesCompatibleForMultiOutputFusion()
183 auto* instr_2 = GetRealHeroForMultiOutputFusion(instr2); in ShapesCompatibleForMultiOutputFusion()
Dgpu_fusible.h76 const HloInstruction* GetRealHeroForMultiOutputFusion(
Dhorizontal_input_fusion.cc37 const HloInstruction* real_hero = GetRealHeroForMultiOutputFusion(instr); in GetInputShapeForMultiOutputFusion()