Home
last modified time | relevance | path

Searched defs:device_groups (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h256 : device_groups(std::move(device_groups)), in device_groups() function
263 std::vector<std::vector<int64_t>> device_groups; member
Dhlo_sharding_util_test.cc297 std::vector<std::vector<int64_t>> device_groups = {{0, 2}, {1, 3}}; in TEST() local
316 std::vector<std::vector<int64_t>> device_groups = {{0, 2, 4, 6}, in TEST() local
339 std::vector<std::vector<int64_t>> device_groups = {{0, 1, 4, 5}, in TEST() local
365 std::vector<std::vector<int64_t>> device_groups = {{0, 1}, {2, 3}}; in TEST() local
385 std::vector<std::vector<int64_t>> device_groups = {{0, 2}, {1, 3}}; in TEST() local
433 std::vector<std::vector<int64_t>> device_groups = {{0, 2}, {1, 3}}; in TEST() local
Dhlo_sharding_util.cc1648 std::vector<std::vector<int64_t>> device_groups(Product(group_dim_sizes)); in GroupShardingOnDims() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h562 int64_t bytes, absl::Span<const ReplicaGroup> device_groups) { in GetCommunicationTimeInMilliSec()
567 absl::Span<const ReplicaGroup> device_groups) { in GetCommunicationMultiplier()
Dspmd_partitioner_util.cc1460 const std::vector<std::vector<int64_t>>& device_groups, SpmdBuilder* b) { in GetInGroupPartitionId()
1478 const std::vector<std::vector<int64_t>>& device_groups) { in GetPerGroupCollectiveOpsCreator()
1559 const std::vector<std::vector<int64_t>>& device_groups, SpmdBuilder* b) { in CreatePerGroupPartitioningState()
1581 const std::vector<std::vector<int64_t>>& device_groups, in PerGroupSliceFromReplicated()
1624 const std::vector<std::vector<int64_t>>& device_groups) { in FindMatchingPartitionedDimsForGrouping()
Dspmd_partitioner.cc2030 std::vector<ReplicaGroup> device_groups; in CreateReplicaGroups() local
4242 std::vector<ReplicaGroup> device_groups; in GetDefaultCollectiveOpsCreator() local
4303 std::vector<ReplicaGroup> device_groups; in GetDefaultCollectiveOpsCreator() local