Searched refs:ReplicaIdForDevice (Results 1 – 4 of 4) sorted by relevance
57 StatusOr<int> ReplicaIdForDevice(GlobalDeviceId device_id) const;
68 StatusOr<int> DeviceAssignment::ReplicaIdForDevice( in ReplicaIdForDevice() function in xla::DeviceAssignment
626 ->ReplicaIdForDevice(xla::GlobalDeviceId(device_ordinal)) in __xla_cpu_runtime_AllToAll()714 ->ReplicaIdForDevice(xla::GlobalDeviceId(device_ordinal)) in __xla_cpu_runtime_ReplicaId()730 ->ReplicaIdForDevice(xla::GlobalDeviceId(device_ordinal)) in __xla_cpu_runtime_CollectivePermute()
246 params.device_assn->ReplicaIdForDevice(global_device_id)); in ExecuteOnStream()