Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.h149 GpuDeviceInfo GetGpuDeviceInfo(se::StreamExecutor* stream_exec);
Dgpu_compiler.cc867 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in RunBackend()
963 GpuDeviceInfo GetGpuDeviceInfo(se::StreamExecutor* stream_exec) { in GetGpuDeviceInfo() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc54 GpuDeviceInfo gpu_device_info = GetGpuDeviceInfo(stream_exec); in CompileMlirModule()