Home
last modified time | relevance | path

Searched defs:selfInfo (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DIndexing.cu922 cuda::detail::TensorInfo<scalar_t, unsigned int> selfInfo = in index_add_cuda_impl() local
974 cuda::detail::TensorInfo<scalar_t, uint64_t> selfInfo = in index_add_cuda_impl() local
1095 cuda::detail::TensorInfo<scalar_t, unsigned int> selfInfo = in index_reduce_func_cuda_impl() local
1147 cuda::detail::TensorInfo<scalar_t, uint64_t> selfInfo = in index_reduce_func_cuda_impl() local
1394 …auto selfInfo = tensorInfoLegacyIfScalar(cuda::detail::getTensorInfo<const scalar_t, unsigned int… in index_select_out_cuda_impl() local
1441 …auto selfInfo = tensorInfoLegacyIfScalar(cuda::detail::getTensorInfo<const scalar_t, uint64_t>(sel… in index_select_out_cuda_impl() local