Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc79 result["Threads Per Block Limit"] = absl::StrCat(threads_per_block_limit()); in ToMap()
132 const int64 threads_per_block_limit = in ThreadDimOk() local
133 device_description.threads_per_block_limit(); in ThreadDimOk()
134 if (total_threads > threads_per_block_limit) { in ThreadDimOk()
136 << " vs limit " << threads_per_block_limit; in ThreadDimOk()
157 *threads_per_block = device_description.threads_per_block_limit(); in CalculateDimensionality()
Ddevice_description.h84 const int64 &threads_per_block_limit() const { in threads_per_block_limit() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_device_info.h32 int threads_per_block_limit; member
Dbuffer_comparator.cc609 gpu_device_info.threads_per_block_limit = in DeviceCompare()
610 executor->GetDeviceDescription().threads_per_block_limit(); in DeviceCompare()
Dlaunch_dimensions.cc38 int64 threads_per_block = gpu_device_info.threads_per_block_limit; in ThreadsPerBlockLimit()
Dgpu_compiler.cc965 gpu_device_info.threads_per_block_limit = in GetGpuDeviceInfo()
966 stream_exec->GetDeviceDescription().threads_per_block_limit(); in GetGpuDeviceInfo()
Dir_emitter_unnested.cc2964 ir_emitter_context_->gpu_device_info().threads_per_block_limit || in EmitSortFromMlir()
2973 ir_emitter_context_->gpu_device_info().threads_per_block_limit, in EmitSortFromMlir()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dhlo_to_llvm_ir.cc47 gpu_device_info.threads_per_block_limit = 1024; in CompileAndPrintLlvmIr()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h133 int64_t threads_per_block_limit; member
/external/tensorflow/tensorflow/stream_executor/gpu/
Dredzone_allocator.cc224 executor->GetDeviceDescription().threads_per_block_limit(), num_elements); in RunRedzoneChecker()