| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | fully-connected-sparse.c | 82 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 D | operator-run.c | 1123 .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 D | sparse_csr_matrix_ops.cc | 96 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 D | sparse_cholesky_op.cc | 119 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 D | BUILD | 22 name = "sparse_matrix",
23 srcs = ["sparse_matrix.cc"],
24 hdrs = ["sparse_matrix.h"],
64 ":sparse_matrix",
|
| H A D | sparse_ordering_amd_op.cc | 106 Eigen::Map<const SparseMatrix> sparse_matrix( in Compute() local
115 amd_ordering(sparse_matrix.template selfadjointView<Eigen::Lower>(), in Compute()
|
| H A D | mat_mul_op.cc | 322 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 D | sparse_mat_mul_op.cc | 272 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 D | csr_sparse_matrix_ops_test.py | 1116 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 D | api_def_SparseMatrixNNZ.pbtxt | 5 name: "sparse_matrix"
10 description: "The number of nonzeroes of `sparse_matrix`."
12 summary: "Returns the number of nonzeroes of `sparse_matrix`."
|
| H A D | api_def_SparseMatrixSparseCholesky.pbtxt | 69 ordering_amd = sparse_csr_matrix_ops.sparse_matrix_ordering_amd(sparse_matrix)
72 sparse_matrix, ordering_amd, type=tf.float32))
|
| H A D | api_def_SparseMatrixZeros.pbtxt | 9 name: "sparse_matrix"
|
| H A D | api_def_SparseTensorToCSRSparseMatrix.pbtxt | 17 name: "sparse_matrix"
|
| H A D | api_def_CSRSparseMatrixToSparseTensor.pbtxt | 5 name: "sparse_matrix"
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-018.md | 41 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 D | SparseMatrixNNZ.pbtxt | 4 name: "sparse_matrix"
|
| H A D | SparseMatrixZeros.pbtxt | 8 name: "sparse_matrix"
|
| H A D | CSRSparseMatrixToSparseTensor.pbtxt | 4 name: "sparse_matrix"
|
| H A D | SparseTensorToCSRSparseMatrix.pbtxt | 16 name: "sparse_matrix"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| H A D | SparseMatrixNNZ.pbtxt | 4 name: "sparse_matrix"
|
| H A D | SparseMatrixZeros.pbtxt | 8 name: "sparse_matrix"
|
| H A D | SparseTensorToCSRSparseMatrix.pbtxt | 16 name: "sparse_matrix"
|
| H A D | CSRSparseMatrixToSparseTensor.pbtxt | 4 name: "sparse_matrix"
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | operator.h | 133 sparse_matrix_t sparse_matrix; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/mkl/ |
| H A D | SparseDescriptors.h | 74 : public MklSparseDescriptor<sparse_matrix, &mkl_sparse_destroy> {
|