Searched defs:sub_iter (Results 1 – 13 of 13) sorted by relevance
65 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_index_kernel() local120 for (auto& sub_iter : iter.with_32bit_indexing()) { in index_fill_kernel_impl() local157 for (auto& sub_iter : iter.with_32bit_indexing()) { in index_copy_kernel_impl() local281 for (auto& sub_iter : iter.with_32bit_indexing()) { in cuda_take_put_kernel() local431 for (auto& sub_iter : iter.with_32bit_indexing()) { in flip_kernel_impl() local
89 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_kernel_nocast() local112 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_kernel() local317 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_kernel_multiple_outputs() local
127 for (auto& sub_iter : iter.with_32bit_indexing()) { in operator ()() local349 for (auto& sub_iter : iter.with_32bit_indexing()) { in operator ()() local
89 for (auto& sub_iter : iter.with_32bit_indexing()) { in jitted_gpu_kernel() local
63 for (const auto& sub_iter: iter.with_32bit_indexing()) { in glu_backward_cuda_out() local
57 for (auto& sub_iter : iter.with_32bit_indexing()) { in _compute_linear_combination_internal_kernel() local
112 for (auto& sub_iter : iter.with_32bit_indexing()) { in unpack_pivots_cuda_kernel() local
68 for (auto& sub_iter : iter.with_32bit_indexing()) { in _unfold_backward_internal_kernel() local
83 for (auto& sub_iter : iter.with_32bit_indexing()) { in binary_op_intersection_kernel() local
1181 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_reduce_kernel() local1289 for (auto& sub_iter : iter.with_32bit_indexing()) { in jitted_gpu_reduce_kernel() local
121 auto sub_iter = TensorIterator(iter); in parallel_dim_reduction() local164 TensorIterator sub_iter(*this); in foreach_reduced_elt() local
1368 auto sub_iter = TensorIterator(iter); in index_select_out_cpu_() local
297 for (auto& sub_iter : iter.with_32bit_indexing()) { in jitted_gpu_kernel_dynamic() local