Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddevice_description.cc119 const int64_t total_threads = thread_dim.x * thread_dim.y * thread_dim.z; in ThreadDimOk() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dparallel_loop_emitter.cc313 int64_t total_threads = launch_dimensions_.launch_bound(); in EmitLoop() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCUDATensorMath.cu641 int64_t total_threads = input_nnz; in _sparse_sum_backward_cuda() local
/aosp_15_r20/external/coreboot/src/cpu/x86/
H A Dmp_init.c1017 const int total_threads = global_num_aps + 1; /* +1 for BSP */ in mp_run_on_all_aps() local
/aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/core/
H A Dkernel.rs1222 let total_threads = block.iter().take(work_dim).product::<usize>(); in suggest_local_size() localVariable