Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils_test.cc60 GetParticipatingDevices(GlobalDeviceId(42), device_assignment, in TEST()
83 GetParticipatingDevices(GlobalDeviceId(42), device_assignment, in TEST()
100 GetParticipatingDevices(GlobalDeviceId(44), device_assignment, in TEST()
Dcollective_ops_utils.cc78 StatusOr<std::vector<GlobalDeviceId>> GetParticipatingDevices( in GetParticipatingDevices() function
Dcollective_ops_utils.h48 StatusOr<std::vector<GlobalDeviceId>> GetParticipatingDevices(
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.cc75 GetParticipatingDevices(global_device_id, *params.device_assn, in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc606 xla::GetParticipatingDevices(xla::GlobalDeviceId(device_ordinal), in GetRendezvousKey()