Searched refs:GetFlatTupleElems (Results 1 – 1 of 1) sorted by relevance
622 static std::vector<HloInstruction*> GetFlatTupleElems( in GetFlatTupleElems() function635 auto flattened_subshape = GetFlatTupleElems(gte, new_instrs); in GetFlatTupleElems()697 return HloInstruction::CreateTuple(GetFlatTupleElems(instr, &new_instrs)); in TryFlattenNestedTuples()