Searched refs:n_vec_to_read (Results 1 – 1 of 1) sorted by relevance
167 const int n_vec_to_read = N/vec_size; in compute_stats() local170 for (int i = thrx; i < n_vec_to_read; i += numx) { in compute_stats()244 const int n_vec_to_read = N/vec_size; in vectorized_layer_norm_kernel_impl() local249 for (int i = thrx; i < n_vec_to_read; i += numx) { in vectorized_layer_norm_kernel_impl()