Searched refs:used_device_ids (Results 1 – 1 of 1) sorted by relevance
374 std::vector<bool> used_device_ids(bound_x * bound_y * bound_z * bound_core, in GetGeneralTPUExecutionDeviceAssignment() local402 if (used_device_ids[device_id]) in GetGeneralTPUExecutionDeviceAssignment()406 used_device_ids[device_id] = true; in GetGeneralTPUExecutionDeviceAssignment()