| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 96 ShapeHandle sparse_matrix = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30202() local
97 TF_RETURN_IF_ERROR(c->WithRankAtMost(sparse_matrix, 3, &sparse_matrix)); in __anon59a9d6c30202()
98 if (!c->RankKnown(sparse_matrix)) { in __anon59a9d6c30202()
101 int rank = c->Rank(sparse_matrix); in __anon59a9d6c30202()
158 ShapeHandle sparse_matrix = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30402() local
159 TF_RETURN_IF_ERROR(c->WithRankAtMost(sparse_matrix, 3, &sparse_matrix)); in __anon59a9d6c30402()
160 if (!c->RankKnown(sparse_matrix)) { in __anon59a9d6c30402()
163 c->set_output(0, sparse_matrix); in __anon59a9d6c30402()
205 ShapeHandle sparse_matrix = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30602() local
206 TF_RETURN_IF_ERROR(c->WithRankAtLeast(sparse_matrix, 2, &sparse_matrix)); in __anon59a9d6c30602()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| 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`."
|
| D | api_def_SparseMatrixZeros.pbtxt | 9 name: "sparse_matrix"
|
| D | api_def_CSRSparseMatrixToSparseTensor.pbtxt | 5 name: "sparse_matrix"
|
| D | api_def_SparseTensorToCSRSparseMatrix.pbtxt | 17 name: "sparse_matrix"
|
| 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))
|
| D | api_def_SparseMatrixOrderingAMD.pbtxt | 55 ordering_amd = sparse_csr_matrix_ops.sparse_matrix_ordering_amd(sparse_matrix)
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_cholesky_op.cc | 117 Eigen::Map<const SparseMatrix> sparse_matrix( in Compute() local
139 sparse_matrix.template selfadjointView<Eigen::Upper>() in Compute()
230 const CSRSparseMatrix& sparse_matrix, in ValidateInputs() argument
233 OP_REQUIRES(ctx, sparse_matrix.dtype() == DataTypeToEnum<T>::value, in ValidateInputs()
237 " but saw dtype: ", DataTypeString(sparse_matrix.dtype()))); in ValidateInputs()
239 const Tensor& dense_shape = sparse_matrix.dense_shape(); in ValidateInputs()
264 *batch_size = sparse_matrix.batch_size(); in ValidateInputs()
|
| D | BUILD | 17 name = "sparse_matrix",
18 srcs = ["sparse_matrix.cc"],
19 hdrs = ["sparse_matrix.h"],
59 ":sparse_matrix",
|
| 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()
|
| D | mat_mul_op.cc | 313 auto sparse_matrix = GetSparseMatrixRef( in SparseDenseMatMulWithoutTransposedLHS() local
328 output_map.noalias() = sparse_matrix * rhs_map; in SparseDenseMatMulWithoutTransposedLHS()
384 auto sparse_matrix = GetSparseMatrixRef( in SparseDenseMatMulWithTransposedLHS() local
406 b_dense_map.transpose() * sparse_matrix.conjugate(); in SparseDenseMatMulWithTransposedLHS()
409 b_dense_map.transpose() * sparse_matrix; in SparseDenseMatMulWithTransposedLHS()
|
| 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()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 1134 sparse_matrix = dense_to_csr_sparse_matrix(
1139 sparse_matrix)
1142 sparse_matrix, ordering_amd, type=dtype))
1188 sparse_matrix = dense_to_csr_sparse_matrix(
1191 sparse_matrix)
1195 sparse_matrix, ordering_amd, type=dtype))
1230 sparse_matrix = dense_to_csr_sparse_matrix(dense_matrix)
1232 sparse_matrix)
1236 sparse_matrix, ordering_amd, type=dtypes.float32)
1277 sparse_matrix = dense_to_csr_sparse_matrix(
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseMatrixNNZ.pbtxt | 4 name: "sparse_matrix"
|
| D | SparseMatrixZeros.pbtxt | 8 name: "sparse_matrix"
|
| D | CSRSparseMatrixToSparseTensor.pbtxt | 4 name: "sparse_matrix"
|
| D | SparseTensorToCSRSparseMatrix.pbtxt | 16 name: "sparse_matrix"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseMatrixNNZ.pbtxt | 4 name: "sparse_matrix"
|
| D | SparseMatrixZeros.pbtxt | 8 name: "sparse_matrix"
|
| D | CSRSparseMatrixToSparseTensor.pbtxt | 4 name: "sparse_matrix"
|
| D | SparseTensorToCSRSparseMatrix.pbtxt | 16 name: "sparse_matrix"
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_test.cc | 1045 std::vector<int8_t> sparse_matrix; member
1106 data.sparse_matrix.push_back(data.matrix[k]); in SetupMatrixVectorData()
1155 data.sparse_matrix.data(), data.ledger.data(), rows, cols, in TestSparseDotprodMatrixBatchVectorMultiply()
2248 data.sparse_matrix.data(), data.ledger.data(), data.rows, data.cols, in BM_DotprodSparseMultiply()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 34766 func CSRSparseMatrixToSparseTensor(scope *Scope, sparse_matrix tf.Output, type_ tf.DataType) (indic…
34774 sparse_matrix,
37914 …ope *Scope, indices tf.Output, values tf.Output, dense_shape tf.Output) (sparse_matrix tf.Output) {
46799 func SparseMatrixZeros(scope *Scope, dense_shape tf.Output, type_ tf.DataType) (sparse_matrix tf.Ou…
50953 func SparseMatrixNNZ(scope *Scope, sparse_matrix tf.Output) (nnz tf.Output) {
50960 sparse_matrix,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 673 …argspec: "args=[\'sparse_matrix\', \'type\', \'name\'], varargs=None, keywords=None, defaults=[\'N…
4325 … argspec: "args=[\'sparse_matrix\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 673 …argspec: "args=[\'sparse_matrix\', \'type\', \'name\'], varargs=None, keywords=None, defaults=[\'N…
4325 … argspec: "args=[\'sparse_matrix\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|