Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DScanUtils.cuh21 integer log_num_threads_x = 0; in get_log_num_threads_x_inner_scan() local
63 … const uint32_t num_threads, const uint32_t log_num_threads_x, in tensor_kernel_scan_innermost_dim_with_indices()
227 const uint32_t log_num_threads_x = get_log_num_threads_x_inner_scan<uint32_t>(num_rows, row_size); in scan_innermost_dim_with_indices() local
306 const uint32_t log_num_threads_x, in tensor_kernel_scan_innermost_dim_impl()
377 const uint32_t log_num_threads_x, in tensor_kernel_scan_innermost_dim()
426 const uint32_t log_num_threads_x = get_log_num_threads_x_inner_scan<uint32_t>(num_rows, row_size); in scan_innermost_dim() local