Home
last modified time | relevance | path

Searched refs:per_group_strings (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1680 std::vector<std::string> per_group_strings(device_groups.size()); in CreatePerGroupPartitioningState() local
1681 for (int64 i = 0; i < per_group_strings.size(); ++i) { in CreatePerGroupPartitioningState()
1682 per_group_strings[i] = absl::StrJoin(device_groups[i], ","); in CreatePerGroupPartitioningState()
1685 state.reshard_cache->groupd_caches[absl::StrJoin(per_group_strings, ";")]; in CreatePerGroupPartitioningState()