Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
DIndexingUtils.h153 DimVector indexed_sizes; member
/external/pytorch/aten/src/ATen/native/cuda/
DIndexKernel.cu297 const auto indexed_sizes = std::vector<int64_t>(indexed.sizes().rbegin(), indexed.sizes().rend()); in cuda_take_put_kernel() local