Searched refs:UnflattenTupleInstr (Results 1 – 1 of 1) sorted by relevance
582 static std::unique_ptr<HloInstruction> UnflattenTupleInstr( in UnflattenTupleInstr() function610 UnflattenTupleInstr(instrs.subspan(0, num_leaves), in UnflattenTupleInstr()689 UnflattenTupleInstr(absl::MakeSpan(gtes), while_shape, &new_instrs); in TryFlattenNestedTuples()