Searched refs:size_in_kib (Results 1 – 1 of 1) sorted by relevance
68 int64 size_in_kib = sizes_in_kib[i]; in MakeCrossReplicaReductions() local73 F32, {static_cast<int32>(size_in_kib * 1024 / sizeof(float))}); in MakeCrossReplicaReductions()