Function VisibleDeviceCount

tensorflow/core/common_runtime/direct_session.cc:239–251

}

int VisibleDeviceCount(const string& visible_device_list) {
#if GOOGLE_CUDA
  if (visible_device_list.empty()) {
    return GPUMachineManager()->VisibleDeviceCount();
  }

  const std::vector<string> order_str =
      str_util::Split(visible_device_list, ',');
  return order_str.size();
#else
  return 0;
#endif
}

int VisibleDeviceCount() {
#if GOOGLE_CUDA