Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_device_info.h35 int threads_per_core_limit; member
Dbuffer_comparator.cc615 gpu_device_info.threads_per_core_limit = in DeviceCompare()
616 executor->GetDeviceDescription().threads_per_core_limit(); in DeviceCompare()
Dlaunch_dimensions.cc97 (gpu_device_info.threads_per_core_limit / capped_threads_per_block); in CalculateLaunchDimensions()
Dgpu_compiler.cc971 gpu_device_info.threads_per_core_limit = in GetGpuDeviceInfo()
972 stream_exec->GetDeviceDescription().threads_per_core_limit(); in GetGpuDeviceInfo()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dhlo_to_llvm_ir.cc51 gpu_device_info.threads_per_core_limit = 2048; in CompileAndPrintLlvmIr()
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc78 result["Threads Per Core Limit"] = absl::StrCat(threads_per_core_limit()); in ToMap()
Ddevice_description.h90 const int64 &threads_per_core_limit() const { in threads_per_core_limit() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h132 int64_t threads_per_core_limit; member