Home
last modified time | relevance | path

Searched refs:with_32bit_indexing (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DIndexKernel.cu65 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()
H A DLoops.cuh89 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()
H A DActivation.cpp63 for (const auto& sub_iter: iter.with_32bit_indexing()) { in glu_backward_cuda_out()
H A DScatterGatherKernel.cu127 for (auto& sub_iter : iter.with_32bit_indexing()) { in operator ()()
349 for (auto& sub_iter : iter.with_32bit_indexing()) { in operator ()()
H A DFunctionOfAMatrixUtilsKernel.cu57 for (auto& sub_iter : iter.with_32bit_indexing()) { in _compute_linear_combination_internal_kernel()
H A DCrossKernel.cu84 for (auto&& sub_iter: iter.with_32bit_indexing()) { in cross_impl()
H A DJitLoops.cuh89 for (auto& sub_iter : iter.with_32bit_indexing()) { in jitted_gpu_kernel()
H A DLinearAlgebra.cu112 for (auto& sub_iter : iter.with_32bit_indexing()) { in unpack_pivots_cuda_kernel()
H A DUnfoldBackwardKernel.cu68 for (auto& sub_iter : iter.with_32bit_indexing()) { in _unfold_backward_internal_kernel()
H A DSparseBinaryOpIntersectionKernel.cu83 for (auto& sub_iter : iter.with_32bit_indexing()) { in binary_op_intersection_kernel()
H A DDistributionTemplates.h138 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()
H A DReduce.cuh1181 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator.cu297 for (auto& sub_iter : iter.with_32bit_indexing()) { in jitted_gpu_kernel_dynamic()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.h507 SplitUntil32Bit with_32bit_indexing() const;
H A DTensorIterator.cpp1667 SplitUntil32Bit TensorIteratorBase::with_32bit_indexing() const { in with_32bit_indexing() function in at::TensorIteratorBase
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUnaryOpsKernel.cpp99 for (auto& sub_it : it->with_32bit_indexing()) { in LogitMKLKernel()