Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc143 auto* instr_1 = GetRealHeroForMultiOutputFusion(instr1); in ShapesCompatibleForMultiOutputFusion() local
145 if (IsReductionFromOrToContiguousDimensions(*instr_1) && in ShapesCompatibleForMultiOutputFusion()
147 !AreFusedReductionOutputsConsistent({instr_1, instr_2}, instr_1)) { in ShapesCompatibleForMultiOutputFusion()
151 return ShapeUtil::EqualIgnoringElementType(get_loop_shape(instr_1), in ShapesCompatibleForMultiOutputFusion()