Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc78 result["Threads Per Block Limit"] = absl::StrCat(threads_per_block_limit()); in ToMap()
129 auto threads_per_block_limit = device_description.threads_per_block_limit(); in ThreadDimOk() local
130 if (total_threads > threads_per_block_limit) { in ThreadDimOk()
132 << " vs limit " << threads_per_block_limit; in ThreadDimOk()
153 *threads_per_block = device_description.threads_per_block_limit(); in CalculateDimensionality()
Ddevice_description.h88 const int64 &threads_per_block_limit() const { in threads_per_block_limit() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dpartition_assignment.cc43 int64 threads_per_block = device_desc.threads_per_block_limit(); in ThreadsPerBlockLimit()
Dir_emitter_unnested.cc1311 ir_emitter_context_->device_description().threads_per_block_limit() || in HandleSort()