Home
last modified time | relevance | path

Searched defs:total_threads (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddevice_description.cc119 const int64_t total_threads = thread_dim.x * thread_dim.y * thread_dim.z; in ThreadDimOk() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc313 int64_t total_threads = launch_dimensions_.launch_bound(); in EmitLoop() local