Searched defs:num_threads_x (Results 1 – 3 of 3) sorted by relevance
69 const uint32_t num_threads_x = 1 << log_num_threads_x; in tensor_kernel_scan_innermost_dim_with_indices() local228 const uint32_t num_threads_x = (1 << log_num_threads_x); in scan_innermost_dim_with_indices() local308 const uint32_t num_threads_x = 1 << log_num_threads_x; in tensor_kernel_scan_innermost_dim_impl() local382 const uint32_t num_threads_x = 1 << log_num_threads_x; in tensor_kernel_scan_innermost_dim() local427 const uint32_t num_threads_x = (1 << log_num_threads_x); in scan_innermost_dim() local
146 uint32_t num_threads_x; member
3622 int64_t num_threads_x = tiling_scheme.GetNumThreadsFor(kDimX); in GenerateReductionCodegenState() local4533 const ReductionDimensions& reduction_dimensions, int num_threads_x, in CanVectorizeReduction()4583 int64_t num_threads_x = [&] { in ComputeReductionCodegenInfo() local