120   TORCH_CHECK(select_indices.numel() == add_indices.numel());  in index_select_add()
129   auto numel = add_indices.numel();  in index_select_add()  local
137   for (const auto i : c10::irange(numel)) {  in index_select_add()
208     int64_t output_size = offsets.numel() - 1;  in index_select_add()
213       output_size = offsets.numel() - 1;  in index_select_add()
215       output_size = offsets.numel();  in index_select_add()
216       offsets_include_last.resize(offsets.numel() + 1);  in index_select_add()
217       if (offsets.numel() > 0) {  in index_select_add()
221             sizeof(index_t) * offsets.numel());  in index_select_add()
223       offsets_include_last[offsets.numel()] = select_indices.numel();  in index_select_add()
309     TORCH_CHECK(select_indices.numel() == add_indices.numel());  in index_select_add()
321     auto numel = add_indices.numel();  in index_select_add()  local
331     for (const auto i : c10::irange(numel)) {  in index_select_add()
386     int64_t output_size = offsets.numel() - 1;  in index_select_add()
391       output_size = offsets.numel() - 1;  in index_select_add()
393       output_size = offsets.numel();  in index_select_add()
394       offsets_include_last.resize(offsets.numel() + 1);  in index_select_add()
395       if (offsets.numel() > 0) {  in index_select_add()
399             sizeof(index_t) * offsets.numel());  in index_select_add()
401       offsets_include_last[offsets.numel()] = select_indices.numel();  in index_select_add()
454     AT_ASSERT(select_indices.numel() == add_indices.numel());  in index_select_add()
466     auto numel = add_indices.numel();  in index_select_add()  local
467     for (const auto i : c10::irange(numel)) {  in index_select_add()
508   AT_ASSERT(select_indices.numel() == add_indices.numel());  in index_select_scale_add()
517   auto numel = add_indices.numel();  in index_select_scale_add()  local
528   for (const auto i : c10::irange(numel)) {  in index_select_scale_add()
574     int64_t output_size = offsets.numel() - 1;  in index_select_scale_add()
579       output_size = offsets.numel() - 1;  in index_select_scale_add()
581       output_size = offsets.numel();  in index_select_scale_add()
582       offsets_include_last.resize(offsets.numel() + 1);  in index_select_scale_add()
586           sizeof(index_t) * offsets.numel());  in index_select_scale_add()
587       offsets_include_last[offsets.numel()] = select_indices.numel();  in index_select_scale_add()
600           scale_fp32.numel());  in index_select_scale_add()
605           scale_fp32.numel());  in index_select_scale_add()
645     for (const auto i : c10::irange(scale.numel())) {  in index_select_scale_add()
692     AT_ASSERT(select_indices.numel() == add_indices.numel());  in index_select_scale_add()
705     auto numel = add_indices.numel();  in index_select_scale_add()  local
712     for (const auto i : c10::irange(numel)) {  in index_select_scale_add()
763     int64_t output_size = offsets.numel() - 1;  in index_select_scale_add()
768       output_size = offsets.numel() - 1;  in index_select_scale_add()
770       output_size = offsets.numel();  in index_select_scale_add()
771       offsets_include_last.resize(offsets.numel() + 1);  in index_select_scale_add()
775           sizeof(index_t) * offsets.numel());  in index_select_scale_add()
776       offsets_include_last[offsets.numel()] = select_indices.numel();  in index_select_scale_add()
829     AT_ASSERT(select_indices.numel() == add_indices.numel());  in index_select_scale_add()
842     auto numel = add_indices.numel();  in index_select_scale_add()  local
845     for (const auto i : c10::irange(numel)) {  in index_select_scale_add()
907     TORCH_CHECK(per_sample_weights.value().numel() == indices.numel());  in check_arguments()
1073   int64_t numIndices = indices.numel();  in embedding_bag_cpu_max_out()
1534   int64_t numel = indices.numel();  in _embedding_bag_dense_backward_cpu_sum_mean()  local
1539     [&indices, &offset2bag, &bag_size_, &num_weights, &numel, &per_sample_weights,  in _embedding_bag_dense_backward_cpu_sum_mean()
1546     auto counts = compute_counts(num_weights, indices_data, numel);  in _embedding_bag_dense_backward_cpu_sum_mean()
1548         compute_counts_uniq(num_weights, indices_data, numel, counts);  in _embedding_bag_dense_backward_cpu_sum_mean()
1586     if (numel > 1000) {  in _embedding_bag_dense_backward_cpu_sum_mean()
1676   if (indices.numel() != 0 && offset2bag.numel() == 0) {  in _embedding_bag_per_sample_weights_backward_cpu_template()