| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensor.cpp | 174 int64_t dense_dim, in new_with_dims_sparse()
187 int64_t dense_dim, in new_with_dims_and_tensor_sparse_symint()
318 int64_t dense_dim = values.dim() - 1; in sparse_coo_tensor() local
389 int64_t dense_dim = values.dim() - 1; in _validate_sparse_coo_tensor_args() local
508 auto dense_dim = values.dim() - 1; in _sparse_coo_tensor_unsafe_symint() local
548 int64_t dense_dim) { in sparse_resize_()
557 int64_t dense_dim) { in sparse_resize_and_clear_()
641 int64_t dense_dim = self.dense_dim(); in _coalesce_sparse_cpu() local
|
| H A D | SparseTensorMath.cpp | 733 int64_t dense_dim = dense.dim(); in add_out_dense_sparse_cpu() local
844 const int64_t dense_dim = s_.dense_dim(); in intersection_binary_op_sparse_dense_out() local
867 const auto dense_dim = s.dense_dim(); in intersection_binary_op_sparse_dense_out() local
|
| H A D | SparseCsrTensor.cpp | 378 int64_t dense_dim, in sparse_compressed_tensor_with_dims()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.cpp | 878 static Tensor _batch_tile_tensor(const Tensor& self, IntArrayRef blocksize, const int64_t dense_dim… in _batch_tile_tensor()
998 auto dense_dim = (layout_from == kStrided) ? dense_dim_opt.value_or(0) : self.dense_dim(); in _to_sparse_check_arguments() local
1021 auto dense_dim = *dense_dim_opt; in _to_sparse_check_arguments() local
1045 int64_t dense_dim = dense_dim_opt.value_or(0); in dense_to_sparse_compressed() local
|
| H A D | TensorShape.cpp | 725 int64_t dense_dim) { in check_cat_sparse_dims()
741 int64_t dense_dim = tensors[0].get().dense_dim(); in cat_sparse_impl() local
1280 int64_t dense_dim = dim - sparse_dim + 1; in narrow_copy_sparse() local
1736 int64_t dense_dim = self.dense_dim(); in select_sparse() local
1915 const auto dense_dim = self.dense_dim(); in index_select_sparse_cpu() local
3008 const auto dense_dim = self.dim() - n_batch_dim - 2; in sparse_compressed_transpose() local
3152 int64_t dense_dim = self.dense_dim(); in check_t() local
3389 int64_t dense_dim = self.dense_dim(); in unsqueeze_sparse() local
|
| H A D | TensorFactories.cpp | 1397 int64_t dense_dim = (self.layout() == kStrided ? self.dim() - 2: self.dense_dim()); in zeros_like() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseCsrTensorImpl.h | 77 inline int64_t dense_dim() const noexcept { in dense_dim() function
|
| H A D | SparseCsrTensorImpl.cpp | 108 void SparseCsrTensorImpl::resize_and_clear_(int64_t sparse_dim, int64_t dense_dim, IntArrayRef size… in resize_and_clear_()
|
| H A D | SparseTensorImpl.h | 63 int64_t dense_dim() const { in dense_dim() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op.cc | 399 for (int64_t dense_dim = 0; in ValidateStridedSliceOp() local
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 492 dense_dim: Optional[int] = None # is_sparse, is_sparse_compressed variable in MetaTensorDesc
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Indexing.cu | 1635 const auto dense_dim = self.dense_dim(); in index_select_sparse_cuda() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | ProcessGroupGloo.cpp | 1157 const auto dense_dim = tensor.dense_dim(); in populate_from_sparse_tensor() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 2618 def __init__(self, dense_dim, dense_out, embedding_dim, top_out_in, top_out_out) -> None: argument
|