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.cc182 auto* instr_1 = GetRealHeroForMultiOutputFusion(instr1); in ShapesCompatibleForMultiOutputFusion() local
184 if (IsReductionFromOrToContiguousDimensions(*instr_1) && in ShapesCompatibleForMultiOutputFusion()
186 !AreFusedReductionOutputsConsistent({instr_1, instr_2}, instr_1)) { in ShapesCompatibleForMultiOutputFusion()
190 return ShapeUtil::EqualIgnoringElementType(get_loop_shape(instr_1), in ShapesCompatibleForMultiOutputFusion()