Searched defs:compressed_indices (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseCsrTensor.cpp | 128 static void _validate_sparse_compressed_tensor_args_worker(const Tensor& compressed_indices, const … in _validate_sparse_compressed_tensor_args_worker()
322 void _validate_sparse_compressed_tensor_args(const Tensor& compressed_indices, const Tensor& plain_… in _validate_sparse_compressed_tensor_args()
439 auto compressed_indices = at::empty(compressed_indices_size, options_.dtype(index_dtype)); in sparse_compressed_tensor_with_dims() local
453 const Tensor& compressed_indices, in _sparse_compressed_tensor_unsafe_symint()
480 Tensor _sparse_compressed_tensor_unsafe_template(const Tensor& compressed_indices, in _sparse_compressed_tensor_unsafe_template()
521 const Tensor& compressed_indices, in _estimate_sparse_compressed_tensor_size()
580 const Tensor& compressed_indices, in sparse_compressed_tensor()
610 const Tensor& compressed_indices, in sparse_compressed_tensor()
713 auto compressed_indices = at::empty(compressed_indices_size, options); in empty_sparse_compressed() local
883 auto compressed_indices = AT_DISPATCH_ROW_SPARSE_COMPRESSED_LAYOUTS(self.layout(), in clone_sparse_compressed() local
|
| H A D | SparseCsrTensorMath.cpp | 298 auto compressed_indices = AT_DISPATCH_ROW_SPARSE_COMPRESSED_LAYOUTS(input.layout(), in get_result_tensor_for_unary_op() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_new.cpp | 968 Tensor compressed_indices = internal_new_from_data( in sparse_compressed_tensor_ctor_worker() local
1024 Tensor compressed_indices = internal_new_from_data( in sparse_compressed_tensor_ctor_worker() local
1323 Tensor compressed_indices = internal_new_from_data( in _validate_sparse_compressed_tensor_args() local
1394 Tensor compressed_indices = internal_new_from_data( in _validate_sparse_compressed_tensor_args_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseCsrTensorImpl.h | 56 const Tensor& compressed_indices() const { in compressed_indices() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.cpp | 182 Tensor& compressed_indices, in reshape_2d_sparse_compressed_members_to_nd_batched()
1074 Tensor compressed_indices; in dense_to_sparse_compressed() local
1269 Tensor compressed_indices, plain_indices; in sparse_compressed_to_flipped() local
|
| H A D | TensorFactories.cpp | 1291 auto compressed_indices = at::empty(compressed_indices_size, options); in zeros_sparse_compressed_symint() local
|