Searched defs:nDims (Results 1 – 8 of 8) sorted by relevance
76 auto nDims = safe_downcast<uint32_t>(self.dim()); in permute() local
81 auto nDims = safe_downcast<uint32_t>(self.dim()); in transpose() local
239 auto nDims = safe_downcast<uint32_t>(self.dim()); in slice() local
97 int64_t nDims = self.dim(); in fill_diagonal_() local
1055 int64_t nDims = self.dim(); in diagonal() local1123 int64_t nDims = self.dim() + 1; in diag_embed() local
289 int nDims, c10::MemoryFormat memory_format) { in parallel_cat()481 int nDims = materialized[valid].get().dim(); in TORCH_IMPL_FUNC() local
32 int64_t nDims = static_cast<int64_t>(sizes.size()); in unsqueeze_copy_to() local
1038 auto nDims = sym_sizes.size(); in unsqueeze_to() local