Searched refs:exec_status (Results 1 – 2 of 2) sorted by relevance
38 auto exec_status = gpu_manager->ExecutorForDevice(i); in GetNumAvailableGPUs() local39 if (exec_status.ok()) { in GetNumAvailableGPUs()40 se::StreamExecutor* se = exec_status.ValueOrDie(); in GetNumAvailableGPUs()
1360 auto exec_status = in GetDefaultMinGPUMultiprocessorCount() local1362 if (!exec_status.ok()) { in GetDefaultMinGPUMultiprocessorCount()1366 se::StreamExecutor* se = exec_status.ValueOrDie(); in GetDefaultMinGPUMultiprocessorCount()1604 auto exec_status = in GetValidDeviceIds() local1606 if (!exec_status.ok()) { in GetValidDeviceIds()1609 << exec_status.status().ToString(); in GetValidDeviceIds()1612 se::StreamExecutor* se = exec_status.ValueOrDie(); in GetValidDeviceIds()