Searched refs:element_instr (Results 1 – 1 of 1) sorted by relevance
169 auto get_loop_shape = [&](const HloInstruction* element_instr) { in ShapesCompatibleForMultiOutputFusion() argument172 if (IsReductionFromOrToContiguousDimensions(*element_instr)) { in ShapesCompatibleForMultiOutputFusion()173 return element_instr->operand(0)->shape(); in ShapesCompatibleForMultiOutputFusion()175 return element_instr->shape(); in ShapesCompatibleForMultiOutputFusion()