tensorflow/core/runtime_fallback/util/gpu/gpu_utils.cc
// Helper to lookup GPU platform (CUDA vs ROCm) from a given TensorHandle.
Helper to lookup GPU platform (CUDA vs ROCm) from a given TensorHandle.
Helper to lookup GPU platform ( CUDA vs ROCm ) from a given TensorHandle .
static tfrt::gpu::wrapper::Platform GetTfrtGpuPlatformHelper(
    tensorflow::TensorHandle* th) {
  auto device = th->op_device();
  auto gpu_device = static_cast<tensorflow::BaseGPUDevice*>(device);
  return GetTfrtGpuPlatform(gpu_device);
}
static tfrt :: gpu :: wrapper :: Platform GetTfrtGpuPlatformHelper ( tensorflow :: TensorHandle * th ) { auto device = th -> op_device ( ) ; auto gpu_device = static_cast < tensorflow :: BaseGPUDevice * > ( device ) ; return GetTfrtGpuPlatform ( gpu_device ) ; }
Helper to lookup GPU platform (CUDA vs ROCm) from a given TensorHandle.
Helper to lookup GPU platform ( CUDA vs ROCm ) from a given TensorHandle .