Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils_test.cc28 GetParticipatingReplicas( in TEST()
45 GetParticipatingReplicas( in TEST()
Dcollective_ops_utils.cc53 StatusOr<std::vector<int>> GetParticipatingReplicas( in GetParticipatingReplicas() function
98 GetParticipatingReplicas(replica_id, total_replica_count, in GetParticipatingDevices()
Dcollective_ops_utils.h42 StatusOr<std::vector<int>> GetParticipatingReplicas(
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc639 xla::GetParticipatingReplicas( in __xla_cpu_runtime_AllToAll()