Home
last modified time | relevance | path

Searched defs:GpuDeviceInfo (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_device_info.h31 struct GpuDeviceInfo { struct
32 int threads_per_block_limit;
33 int threads_per_warp;
34 int shared_memory_per_block;
35 int threads_per_core_limit;
36 int core_count;
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h153 struct GpuDeviceInfo { struct
162 void set_tensorflow_gpu_device_info(GpuDeviceInfo* g) { in set_tensorflow_gpu_device_info() argument