tensorflow/core/runtime_fallback/util/gpu/gpu_utils.cc
// Lookup GPU platform (CUDA vs ROCm) from a given TensorHandle.
Lookup GPU platform (CUDA vs ROCm) from a given TensorHandle.
Lookup GPU platform ( CUDA vs ROCm ) from a given TensorHandle .
tfrt::gpu::wrapper::Platform GetTfrtGpuPlatform(tensorflow::TensorHandle* th) {
  static auto gpu_platform = GetTfrtGpuPlatformHelper(th);
  return gpu_platform;
}
tfrt :: gpu :: wrapper :: Platform GetTfrtGpuPlatform ( tensorflow :: TensorHandle * th ) { static auto gpu_platform = GetTfrtGpuPlatformHelper ( th ) ; return gpu_platform ; }
Lookup GPU platform (CUDA vs ROCm) from a given TensorHandle.
Lookup GPU platform ( CUDA vs ROCm ) from a given TensorHandle .
// Cache lookup result assuming TF does not mix CUDA and ROCm tensor handles.