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