Home
last modified time | relevance | path

Searched refs:sparse_matrix (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dfully-connected-sparse.c82 fully_connected->sparse_matrix.indices_dtype = kernel_indices_dtype; in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
85 fully_connected->sparse_matrix.col_indices_w32 = in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
87 fully_connected->sparse_matrix.row_values_w32 = in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
91 fully_connected->sparse_matrix.col_indices_w16 = in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
93 fully_connected->sparse_matrix.row_values_w16 = in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
97 fully_connected->sparse_matrix.col_indices_w8 = in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
99 fully_connected->sparse_matrix.row_values_w8 = in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
109 fully_connected->sparse_matrix.values = kernel_values; in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
110 fully_connected->sparse_matrix.row_block_size = kernel_row_block_size; in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
111 fully_connected->sparse_matrix.col_block_size = kernel_col_block_size; in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
H A Doperator-run.c1123 .kernel_col_indices = op->sparse_matrix.col_indices, in pytorch_qnnp_run_operator()
1124 .kernel_row_values = op->sparse_matrix.row_values, in pytorch_qnnp_run_operator()
1125 .kernel_values = op->sparse_matrix.values, in pytorch_qnnp_run_operator()
1157 if (op->sparse_matrix.row_block_size == 1 && in pytorch_qnnp_run_operator()
1158 op->sparse_matrix.col_block_size == 4) { in pytorch_qnnp_run_operator()
1166 } else if (op->sparse_matrix.row_block_size == 8 && in pytorch_qnnp_run_operator()
1167 op->sparse_matrix.col_block_size == 1) { in pytorch_qnnp_run_operator()
1200 .kernel_indices_dtype = op->sparse_matrix.indices_dtype, in pytorch_qnnp_run_operator()
1201 .kernel_values = op->sparse_matrix.values, in pytorch_qnnp_run_operator()
1214 op->sparse_matrix.col_indices_w32; in pytorch_qnnp_run_operator()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dsparse_csr_matrix_ops.cc96 ShapeHandle sparse_matrix = sparse_matrix_shape_and_type.shape; in __anona88575120202() local
97 TF_RETURN_IF_ERROR(c->WithRankAtMost(sparse_matrix, 3, &sparse_matrix)); in __anona88575120202()
98 if (!c->RankKnown(sparse_matrix)) { in __anona88575120202()
101 int rank = c->Rank(sparse_matrix); in __anona88575120202()
158 ShapeHandle sparse_matrix = sparse_matrix_shape_and_type.shape; in __anona88575120402() local
159 TF_RETURN_IF_ERROR(c->WithRankAtMost(sparse_matrix, 3, &sparse_matrix)); in __anona88575120402()
160 if (!c->RankKnown(sparse_matrix)) { in __anona88575120402()
163 c->set_output(0, sparse_matrix); in __anona88575120402()
205 ShapeHandle sparse_matrix = sparse_matrix_shape_and_type.shape; in __anona88575120602() local
206 TF_RETURN_IF_ERROR(c->WithRankAtLeast(sparse_matrix, 2, &sparse_matrix)); in __anona88575120602()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_cholesky_op.cc119 Eigen::Map<const SparseMatrix> sparse_matrix( in Compute() local
141 sparse_matrix.template selfadjointView<Eigen::Upper>() in Compute()
231 Status ValidateInputs(const CSRSparseMatrix& sparse_matrix, in ValidateInputs() argument
234 if (sparse_matrix.dtype() != DataTypeToEnum<T>::value) in ValidateInputs()
238 " but saw dtype: ", DataTypeString(sparse_matrix.dtype())); in ValidateInputs()
240 const Tensor& dense_shape = sparse_matrix.dense_shape(); in ValidateInputs()
263 *batch_size = sparse_matrix.batch_size(); in ValidateInputs()
H A DBUILD22 name = "sparse_matrix",
23 srcs = ["sparse_matrix.cc"],
24 hdrs = ["sparse_matrix.h"],
64 ":sparse_matrix",
H A Dsparse_ordering_amd_op.cc106 Eigen::Map<const SparseMatrix> sparse_matrix( in Compute() local
115 amd_ordering(sparse_matrix.template selfadjointView<Eigen::Lower>(), in Compute()
H A Dmat_mul_op.cc322 auto sparse_matrix = GetSparseMatrixRef( in SparseDenseMatMulWithoutTransposedLHS() local
337 output_map.noalias() = sparse_matrix * rhs_map; in SparseDenseMatMulWithoutTransposedLHS()
393 auto sparse_matrix = GetSparseMatrixRef( in SparseDenseMatMulWithTransposedLHS() local
415 b_dense_map.transpose() * sparse_matrix.conjugate(); in SparseDenseMatMulWithTransposedLHS()
418 b_dense_map.transpose() * sparse_matrix; in SparseDenseMatMulWithTransposedLHS()
H A Dsparse_mat_mul_op.cc272 Eigen::Map<const SparseMatrix> sparse_matrix( in GetSparseMatrixRef() local
281 if (transpose) return sparse_matrix.transpose(); in GetSparseMatrixRef()
282 if (adjoint) return sparse_matrix.adjoint(); in GetSparseMatrixRef()
283 return sparse_matrix; in GetSparseMatrixRef()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
H A Dcsr_sparse_matrix_ops_test.py1116 sparse_matrix = dense_to_csr_sparse_matrix(
1121 sparse_matrix)
1124 sparse_matrix, ordering_amd, type=dtype))
1169 sparse_matrix = dense_to_csr_sparse_matrix(
1172 sparse_matrix)
1176 sparse_matrix, ordering_amd, type=dtype))
1211 sparse_matrix = dense_to_csr_sparse_matrix(dense_matrix)
1213 sparse_matrix)
1217 sparse_matrix, ordering_amd, type=dtypes.float32)
1258 sparse_matrix = dense_to_csr_sparse_matrix(
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_SparseMatrixNNZ.pbtxt5 name: "sparse_matrix"
10 description: "The number of nonzeroes of `sparse_matrix`."
12 summary: "Returns the number of nonzeroes of `sparse_matrix`."
H A Dapi_def_SparseMatrixSparseCholesky.pbtxt69 ordering_amd = sparse_csr_matrix_ops.sparse_matrix_ordering_amd(sparse_matrix)
72 sparse_matrix, ordering_amd, type=tf.float32))
H A Dapi_def_SparseMatrixZeros.pbtxt9 name: "sparse_matrix"
H A Dapi_def_SparseTensorToCSRSparseMatrix.pbtxt17 name: "sparse_matrix"
H A Dapi_def_CSRSparseMatrixToSparseTensor.pbtxt5 name: "sparse_matrix"
/aosp_15_r20/external/tensorflow/tensorflow/security/advisory/
H A Dtfsa-2021-018.md41 const CSRSparseMatrix& sparse_matrix,
44 OP_REQUIRES(ctx, sparse_matrix.dtype() == DataTypeToEnum<T>::value, ...)
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DSparseMatrixNNZ.pbtxt4 name: "sparse_matrix"
H A DSparseMatrixZeros.pbtxt8 name: "sparse_matrix"
H A DCSRSparseMatrixToSparseTensor.pbtxt4 name: "sparse_matrix"
H A DSparseTensorToCSRSparseMatrix.pbtxt16 name: "sparse_matrix"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DSparseMatrixNNZ.pbtxt4 name: "sparse_matrix"
H A DSparseMatrixZeros.pbtxt8 name: "sparse_matrix"
H A DSparseTensorToCSRSparseMatrix.pbtxt16 name: "sparse_matrix"
H A DCSRSparseMatrixToSparseTensor.pbtxt4 name: "sparse_matrix"
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Doperator.h133 sparse_matrix_t sparse_matrix; member
/aosp_15_r20/external/pytorch/aten/src/ATen/mkl/
H A DSparseDescriptors.h74 : public MklSparseDescriptor<sparse_matrix, &mkl_sparse_destroy> {

12