Searched defs:num_of_segments (Results 1 – 1 of 1) sorted by relevance
49 int64_t num_of_segments = *num_of_segments_ptr; in krn_partials_per_segment() local67 int64_t num_of_segments = *num_of_segments_ptr; in krn_partial_segment_offset() local90 int64_t num_of_segments = *num_of_segments_ptr; in compute_grad_weight_bags() local135 int64_t num_of_segments = *num_of_segments_ptr; in compute_grad_weight() local168 int64_t num_of_segments = *num_of_segments_ptr; in sum_and_scatter() local194 int64_t num_of_segments = *num_of_segments_ptr; in compute_num_of_partial_segments() local200 …ite_num_of_segments_for_legacy_thrust_path(int64_t *num_of_segments_ptr, int64_t num_of_segments) { in write_num_of_segments_for_legacy_thrust_path()239 …int64_t num_of_segments = embedding_backward_cuda_kernel_unique_by_key<index_t>(sorted_indices, se… in embedding_backward_cuda_kernel() local