Searched refs:with_32bit_indexing (Results 1 – 16 of 16) sorted by relevance
65 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_index_kernel()120 for (auto& sub_iter : iter.with_32bit_indexing()) { in index_fill_kernel_impl()157 for (auto& sub_iter : iter.with_32bit_indexing()) { in index_copy_kernel_impl()281 for (auto& sub_iter : iter.with_32bit_indexing()) { in cuda_take_put_kernel()431 for (auto& sub_iter : iter.with_32bit_indexing()) { in flip_kernel_impl()
89 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_kernel_nocast()112 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_kernel()317 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_kernel_multiple_outputs()
63 for (const auto& sub_iter: iter.with_32bit_indexing()) { in glu_backward_cuda_out()
127 for (auto& sub_iter : iter.with_32bit_indexing()) { in operator ()()349 for (auto& sub_iter : iter.with_32bit_indexing()) { in operator ()()
57 for (auto& sub_iter : iter.with_32bit_indexing()) { in _compute_linear_combination_internal_kernel()
84 for (auto&& sub_iter: iter.with_32bit_indexing()) { in cross_impl()
89 for (auto& sub_iter : iter.with_32bit_indexing()) { in jitted_gpu_kernel()
112 for (auto& sub_iter : iter.with_32bit_indexing()) { in unpack_pivots_cuda_kernel()
68 for (auto& sub_iter : iter.with_32bit_indexing()) { in _unfold_backward_internal_kernel()
83 for (auto& sub_iter : iter.with_32bit_indexing()) { in binary_op_intersection_kernel()
138 for (auto& sub_iter : iter.with_32bit_indexing()) { in distribution_nullary_kernel()242 for (auto& sub_iter : iter.with_32bit_indexing()) { in distribution_binary_kernel()
1181 for (auto& sub_iter : iter.with_32bit_indexing()) { in gpu_reduce_kernel()1289 for (auto& sub_iter : iter.with_32bit_indexing()) { in jitted_gpu_reduce_kernel()
297 for (auto& sub_iter : iter.with_32bit_indexing()) { in jitted_gpu_kernel_dynamic()
507 SplitUntil32Bit with_32bit_indexing() const;
1667 SplitUntil32Bit TensorIteratorBase::with_32bit_indexing() const { in with_32bit_indexing() function in at::TensorIteratorBase
99 for (auto& sub_it : it->with_32bit_indexing()) { in LogitMKLKernel()