Home
last modified time | relevance | path

Searched refs:thread_dim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc70 const ThreadDim &thread_dim = thread_dim_limit(); in ToMap() local
72 absl::StrCat(thread_dim.x, ",", thread_dim.y, ",", thread_dim.z); in ToMap()
127 const ThreadDim &thread_dim) { in ThreadDimOk() argument
128 auto total_threads = thread_dim.x * thread_dim.y * thread_dim.z; in ThreadDimOk()
137 bool ok = thread_dim.x <= limit.x && thread_dim.y <= limit.y && in ThreadDimOk()
138 thread_dim.z <= limit.z; in ThreadDimOk()
140 VLOG(2) << "thread dim " << thread_dim.ToString() in ThreadDimOk()
Ddevice_description.h332 const ThreadDim &thread_dim);