Home
last modified time | relevance | path

Searched refs:sparse_matrix_t (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/mkl/
H A DSparseBlas.h20 sparse_matrix_t *A, const sparse_index_base_t indexing, const MKL_INT rows, \
44 sparse_matrix_t *A, const sparse_index_base_t indexing, \
70 const sparse_matrix_t A, const struct matrix_descr descr, \
91 const sparse_operation_t operation, const sparse_matrix_t A, \
92 const scalar_t alpha, const sparse_matrix_t B, sparse_matrix_t *C
112 const sparse_matrix_t source, sparse_index_base_t *indexing, MKL_INT *rows, \
137 const sparse_matrix_t A, const struct matrix_descr descr, \
159 const sparse_operation_t operation, const sparse_matrix_t A, \
160 const sparse_matrix_t B, const sparse_layout_t layout, scalar_t *C, \
183 const sparse_matrix_t A, const struct matrix_descr descr, \
[all …]
H A DSparseDescriptors.h99 sparse_matrix_t raw_descriptor; in MklSparseCsrDescriptor()
136 sparse_matrix_t raw_descriptor = nullptr; in MklSparseCsrDescriptor()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Doperator.h55 } sparse_matrix_t; typedef
133 sparse_matrix_t sparse_matrix;
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/
H A DSparseCsrLinearAlgebra.cpp54 sparse_matrix_t A{nullptr};
H A DSparseBlasImpl.cpp115 void mkl_result_copy_(const Tensor& input, sparse_matrix_t mkl_desc) { in mkl_result_copy_()