Home
last modified time | relevance | path

Searched defs:compressed_indices (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensor.cpp128 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 DSparseCsrTensorMath.cpp298 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 Dtensor_new.cpp968 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 DSparseCsrTensorImpl.h56 const Tensor& compressed_indices() const { in compressed_indices() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorConversions.cpp182 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 DTensorFactories.cpp1291 auto compressed_indices = at::empty(compressed_indices_size, options); in zeros_sparse_compressed_symint() local