Searched defs:dense_ndim (Results 1 – 2 of 2) sorted by relevance
161 const auto dense_ndim = values.dim() - batch_ndim - block_ndim - 1; in _validate_sparse_compressed_tensor_args_worker() local537 const int64_t dense_ndim = values.dim() - batch_ndim - block_ndim - 1; in _estimate_sparse_compressed_tensor_size() local
1456 const auto dense_ndim = self.dense_dim(); in permute_sparse_coo() local