Searched refs:gtes (Results 1 – 3 of 3) sorted by relevance
118 std::vector<HloInstruction*> gtes; in ExpandInstruction() local119 gtes.reserve(all_to_all_group_size); in ExpandInstruction()121 gtes.push_back(all_to_all->parent()->AddInstruction( in ExpandInstruction()126 HloInstruction::CreateConcatenate(all_to_all->shape(), gtes, split_dim)); in ExpandInstruction()
894 std::vector<HloInstruction*> gtes; in TryFlattenNestedTuples() local897 gtes.push_back(add_new_instr(HloInstruction::CreateGetTupleElement( in TryFlattenNestedTuples()901 UnflattenTupleInstr(absl::MakeSpan(gtes), while_shape, &new_instrs); in TryFlattenNestedTuples()
9215 "gtes",21222 "gtes",