Searched refs:get_sparse_impl (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensor.cpp | 82 return get_sparse_impl(self)->sparse_dim(); in sparse_dim_sparse()
86 return get_sparse_impl(self)->dense_dim(); in dense_dim_sparse()
90 return get_sparse_impl(self)->coalesced(); in is_coalesced_sparse()
99 return get_sparse_impl(self)->nnz(); in _nnz_sparse()
107 return get_sparse_impl(self)->indices(); in _indices_sparse()
111 return get_sparse_impl(self)->values(); in _values_sparse()
115 get_sparse_impl(self)->set_coalesced(coalesced); in _coalesced_sparse_()
123 return get_sparse_impl(self)->indices().alias(); in indices_sparse()
134 return get_sparse_impl(self)->values().alias(); in values_sparse()
181 get_sparse_impl(self)->resize_and_clear_(sparse_dim, dense_dim, size); in new_with_dims_sparse()
[all …]
|
| H A D | SparseTensorMath.cpp | 127 get_sparse_impl(r)->set_nnz_and_narrow(t._nnz()); in mul_out_sparse_zerodim()
181 get_sparse_impl(r)->set_nnz_and_narrow(t._nnz()); in pow_out_sparse_scalar()
250 get_sparse_impl(r)->set_nnz_and_narrow(t_tmp._nnz()); in div_out_sparse_zerodim()
330 get_sparse_impl(result)->set_nnz_and_narrow(dividend_tmp._nnz()); in floor_divide_out_sparse_zerodim()
517 get_sparse_impl(r)->set_indices_and_values_unsafe(r_indices, r_values); in add_out_sparse_contiguous()
518 get_sparse_impl(r)->set_nnz_and_narrow(r_i); in add_out_sparse_contiguous()
851 auto* res_impl = get_sparse_impl(res); in intersection_binary_op_sparse_dense_out()
886 auto* res_impl = get_sparse_impl(res); in intersection_binary_op_sparse_dense_out()
987 auto* res_impl = get_sparse_impl(res); in intersection_binary_op_sparse_dense_out()
1174 get_sparse_impl(r)->set_indices_and_values_unsafe(r_indices, r_values); in mul_out_sparse_cpu()
[all …]
|
| H A D | SparseUnaryOps.cpp | 117 auto *input_impl = sparse::get_sparse_impl(input); in coalesced_unary_ufunc_out()
118 auto *result_impl = sparse::get_sparse_impl(result); in coalesced_unary_ufunc_out()
|
| H A D | SparseBinaryOpIntersectionCommon.h | 30 using at::sparse::get_sparse_impl;
414 auto* res_sparse_impl = get_sparse_impl(res);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseCUDATensorMath.cu | 207 get_sparse_impl(r_)->resize_and_clear_(1, 1, {m, n}); in hspmm_out_sparse_cuda()
234 std::vector<int64_t> new_size = get_sparse_impl(newSparse)->sizes().vec(); in hspmm_out_sparse_cuda()
236 …get_sparse_impl(newSparse)->raw_resize_(get_sparse_impl(newSparse)->sparse_dim(), get_sparse_impl(… in hspmm_out_sparse_cuda()
239 get_sparse_impl(r_)->set_indices_and_values_unsafe(indices, values); in hspmm_out_sparse_cuda()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | SparseTensorUtils.h | 27 inline SparseTensorImpl* get_sparse_impl(const SparseTensor& self) { in get_sparse_impl() function
39 get_sparse_impl(self)->set_indices_and_values_unsafe(indices, values); in alias_into_sparse()
|
| H A D | TensorShape.cpp | 2869 at::sparse::get_sparse_impl(self)->raw_resize_(self.sparse_dim(), self.dense_dim(), sizes); in sparse_transpose_()
2886 …at::sparse::get_sparse_impl(self)->raw_resize_(self._indices().size(0), self._values().dim() - 1, … in sparse_transpose_()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | input_buffer.cpp | 55 auto* impl = at::sparse::get_sparse_impl(var); in record_stream_any_impl()
|