Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_device_info.h33 int threads_per_warp; member
Dbuffer_comparator.cc611 gpu_device_info.threads_per_warp = in DeviceCompare()
612 executor->GetDeviceDescription().threads_per_warp(); in DeviceCompare()
Dlaunch_dimensions.cc47 threads_per_block = gpu_device_info.threads_per_warp; in ThreadsPerBlockLimit()
Dgpu_compiler.cc967 gpu_device_info.threads_per_warp = in GetGpuDeviceInfo()
968 stream_exec->GetDeviceDescription().threads_per_warp(); in GetGpuDeviceInfo()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dhlo_to_llvm_ir.cc48 gpu_device_info.threads_per_warp = 32; in CompileAndPrintLlvmIr()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h134 int64_t threads_per_warp; member
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.h95 const int64 &threads_per_warp() const { return threads_per_warp_; } in threads_per_warp() function