Searched refs:xla_assignment (Results 1 – 1 of 1) sorted by relevance
150 DeviceAssignment xla_assignment(devices.size(), devices[0].size()); in DevicesToDeviceAssignment() local170 xla_assignment(replica, partition) = devices[replica][partition]->id(); in DevicesToDeviceAssignment()173 return xla_assignment; in DevicesToDeviceAssignment()