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