Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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
H A Dhlo_sharding_util.h256 : device_groups(std::move(device_groups)), in device_groups() function
263 std::vector<std::vector<int64_t>> device_groups; member
H A Dhlo_sharding_util.cc1648 std::vector<std::vector<int64_t>> device_groups(Product(group_dim_sizes)); in GroupShardingOnDims() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A 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()
H A Dspmd_partitioner.h562 int64_t bytes, absl::Span<const ReplicaGroup> device_groups) { in GetCommunicationTimeInMilliSec()
567 absl::Span<const ReplicaGroup> device_groups) { in GetCommunicationMultiplier()
H A 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
/aosp_15_r20/packages/modules/Bluetooth/android/app/src/com/android/bluetooth/csip/
DCsipSetCoordinatorService.java577 Map<Integer, Integer> device_groups = in getGroupId() local
593 Map<Integer, Integer> device_groups = in getGroupUuidMapByDevice() local
/aosp_15_r20/frameworks/native/vulkan/vkjson/
H A Dvkjson.h165 std::vector<VkJsonDeviceGroup> device_groups; member
/aosp_15_r20/out/soong/.intermediates/packages/modules/Bluetooth/android/app/Bluetooth/android_common_apex33/javac/
DBluetooth.jarMETA-INF/ META-INF/MANIFEST.MF android/ android/bluetooth/ android/ ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/Bluetooth/android/app/Bluetooth/android_common_apex33/combined/
DBluetooth.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx.annotation_annotation- ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/Bluetooth/android/app/Bluetooth/android_common_apex33/jarjar/
DBluetooth.jarorg/jspecify/annotations/Nullable.class Nullable.java package org.jspecify.annotations public ...
/aosp_15_r20/out/soong/.intermediates/packages/modules/Bluetooth/android/app/Bluetooth/android_common_apex33/withres/
DBluetooth.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/androidx.annotation_annotation- ...