Function GpuPlatformName

tensorflow/core/common_runtime/gpu/gpu_init.cc:45–53

}

string GpuPlatformName() {
#if TENSORFLOW_USE_ROCM
  return "ROCM";
#else
  // This function will return "CUDA" even when building TF without GPU support
  // This is done to preserve existing functionality
  return "CUDA";
#endif
}

}  // namespace tensorflow