Searched refs:task_and_device (Results 1 – 1 of 1) sorted by relevance
323 auto& task_and_device = topology(x, y, z, core); in ParseTopologyAttr() local324 if (task_and_device.task != -1) in ParseTopologyAttr()327 task_and_device = {task, device}; in ParseTopologyAttr()397 TaskAndDevice task_and_device = topology(x, y, z, core); in GetGeneralTPUExecutionDeviceAssignment() local398 const int task = task_and_device.task; in GetGeneralTPUExecutionDeviceAssignment()399 const int device = task_and_device.device; in GetGeneralTPUExecutionDeviceAssignment()