Searched defs:plain_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 …ompressed_tensor_args_worker(const Tensor& compressed_indices, const Tensor& plain_indices, const … in _validate_sparse_compressed_tensor_args_worker()
322 …parse_compressed_tensor_args(const Tensor& compressed_indices, const Tensor& plain_indices, const … in _validate_sparse_compressed_tensor_args()
440 auto plain_indices = at::empty(plain_indices_size, options_.dtype(index_dtype)); in sparse_compressed_tensor_with_dims() local
454 const Tensor& plain_indices, in _sparse_compressed_tensor_unsafe_symint()
481 const Tensor& plain_indices, in _sparse_compressed_tensor_unsafe_template()
522 const Tensor& plain_indices, in _estimate_sparse_compressed_tensor_size()
581 const Tensor& plain_indices, in sparse_compressed_tensor()
611 const Tensor& plain_indices, in sparse_compressed_tensor()
714 auto plain_indices = at::empty(plain_indices_and_values_size, options); in empty_sparse_compressed() local
887 auto plain_indices = AT_DISPATCH_ROW_SPARSE_COMPRESSED_LAYOUTS(self.layout(), in clone_sparse_compressed() local
|
| H A D | SparseCsrTensorMath.cpp | 302 auto plain_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 | 976 Tensor plain_indices = internal_new_from_data( in sparse_compressed_tensor_ctor_worker() local
1032 Tensor plain_indices = internal_new_from_data( in sparse_compressed_tensor_ctor_worker() local
1331 Tensor plain_indices = internal_new_from_data( in _validate_sparse_compressed_tensor_args() local
1402 Tensor plain_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 | 59 const Tensor& plain_indices() const { in plain_indices() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.cpp | 183 Tensor& plain_indices, in reshape_2d_sparse_compressed_members_to_nd_batched()
1094 Tensor& plain_indices = compressed_rows_layout ? col_indices : row_indices; in dense_to_sparse_compressed() local
1269 Tensor compressed_indices, plain_indices; in sparse_compressed_to_flipped() local
|
| H A D | TensorFactories.cpp | 1293 auto plain_indices = at::empty(plain_indices_and_values_size, options); in zeros_sparse_compressed_symint() local
|