Home
last modified time | relevance | path

Searched refs:gpu_root_names (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.cc171 struct TargetDeviceFunction gpu_root_names = GetDeviceFunctionRoot(func_id); in ObtainDeviceFunctionName() local
174 return StrCat(gpu_root_names.nvptx_root, "f"); in ObtainDeviceFunctionName()
176 return gpu_root_names.nvptx_root; in ObtainDeviceFunctionName()
182 return StrCat(gpu_root_names.amdgpu_root, "_f32"); in ObtainDeviceFunctionName()
184 return StrCat(gpu_root_names.amdgpu_root, "_f64"); in ObtainDeviceFunctionName()