Searched defs:device_groups (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_sharding_util.h | 256 : device_groups(std::move(device_groups)), in device_groups() function 263 std::vector<std::vector<int64_t>> device_groups; member
|
D | hlo_sharding_util_test.cc | 297 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
|
D | hlo_sharding_util.cc | 1648 std::vector<std::vector<int64_t>> device_groups(Product(group_dim_sizes)); in GroupShardingOnDims() local
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner.h | 562 int64_t bytes, absl::Span<const ReplicaGroup> device_groups) { in GetCommunicationTimeInMilliSec() 567 absl::Span<const ReplicaGroup> device_groups) { in GetCommunicationMultiplier()
|
D | spmd_partitioner_util.cc | 1460 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()
|
D | spmd_partitioner.cc | 2030 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
|