Searched defs:count_data (Results 1 – 4 of 4) sorted by relevance
535 counting_sink<OutputIterator> count_data; // for counting member545 counting_sink<OutputIterator> count_data; member
58 auto count_data = thrust::device_ptr<index_t>(count.mutable_data_ptr<index_t>()); in embedding_dense_backward_cuda_scan() local
317 auto count_data = count.mutable_data_ptr<index_t>(); in embedding_dense_backward_cuda() local
209 auto count_data = count.mutable_data_ptr<index_t>(); in embedding_bag_backward_cuda_sum_avg() local