/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> {
|