Searched refs:element_instr (Results 1 – 1 of 1) sorted by relevance
114 auto get_loop_shape = [&](const HloInstruction* element_instr) { in ShapesCompatibleForMultiOutputFusion() argument117 if (IsReductionToVector(*element_instr)) { in ShapesCompatibleForMultiOutputFusion()118 return element_instr->operand(0)->shape(); in ShapesCompatibleForMultiOutputFusion()120 return element_instr->shape(); in ShapesCompatibleForMultiOutputFusion()