Home
last modified time | relevance | path

Searched refs:CSRSparseMatrix (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/sparse/
Dconj_op.cc50 Status operator()(const CSRSparseMatrix& a, CSRSparseMatrix* b) { in operator ()()
55 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in operator ()()
76 Status operator()(const CSRSparseMatrix& a, CSRSparseMatrix* b) { \
77 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( \
92 CONJ_VARIANT_UNARY_OP, DEVICE_CPU, CSRSparseMatrix,
98 CONJ_VARIANT_UNARY_OP, DEVICE_GPU, CSRSparseMatrix,
Dsparse_matrix.cc27 constexpr const char CSRSparseMatrix::kTypeName[];
30 REGISTER_UNARY_VARIANT_DECODE_FUNCTION(CSRSparseMatrix,
31 CSRSparseMatrix::kTypeName);
35 CSRSparseMatrix, DIRECTION, CSRSparseMatrix::DeviceCopy)
Dsoftmax_op.cc49 const CSRSparseMatrix* logits_matrix; in Compute()
64 CSRSparseMatrix output_matrix; in Compute()
70 CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
103 OpKernelContext* ctx, const CSRSparseMatrix& logits, \
121 const CSRSparseMatrix* softmax_matrix; in Compute()
129 const CSRSparseMatrix* grad_softmax_matrix; in Compute()
174 CSRSparseMatrix gradient_matrix; in Compute()
177 ctx, CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
212 OpKernelContext* ctx, const CSRSparseMatrix& softmax, \
213 const CSRSparseMatrix& grad_softmax, \
Dkernels.h211 const CSRSparseMatrix& input_matrix,
212 CSRSparseMatrix* output_matrix);
219 Status operator()(OpKernelContext* ctx, const CSRSparseMatrix& logits,
225 Status operator()(OpKernelContext* ctx, const CSRSparseMatrix& softmax,
226 const CSRSparseMatrix& grad_softmax,
235 Status Compute(OpKernelContext* ctx, const CSRSparseMatrix& a,
236 typename TTypes<T>::ConstScalar b, CSRSparseMatrix* c);
244 Status Compute(OpKernelContext* ctx, const CSRSparseMatrix& a,
245 typename TTypes<T>::ConstFlat b, CSRSparseMatrix* c);
Dsparse_matrix.h35 class CSRSparseMatrix {
105 CSRSparseMatrix() : metadata_{false, DT_INVALID} {} in CSRSparseMatrix() function
107 CSRSparseMatrix(const CSRSparseMatrix& rhs) in CSRSparseMatrix() function
117 CSRSparseMatrix(CSRSparseMatrix&& rhs) in CSRSparseMatrix() function
130 CSRSparseMatrix& operator=(CSRSparseMatrix&& rhs) {
151 CSRSparseMatrix* matrix) { in CreateCSRSparseMatrix()
152 *matrix = CSRSparseMatrix(dtype, dense_shape, batch_pointers, row_pointers, in CreateCSRSparseMatrix()
383 const CSRSparseMatrix& from, CSRSparseMatrix* to, in DeviceCopy()
400 CSRSparseMatrix(DataType dtype, const Tensor& dense_shape, in CSRSparseMatrix() function
549 const CSRSparseMatrix& a, in CSRSparseMatrixBinaryHelper()
[all …]
Dmul_op.cc46 const CSRSparseMatrix* a_matrix; in Compute()
81 CSRSparseMatrix c_matrix; in Compute()
128 Status Compute(OpKernelContext* ctx, const CSRSparseMatrix& a, in Compute()
129 typename TTypes<T>::ConstScalar b, CSRSparseMatrix* c) { in Compute()
134 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
156 OpKernelContext* ctx, const CSRSparseMatrix& a, \
157 typename TTypes<T>::ConstFlat b, CSRSparseMatrix* c); \
Dzeros_op.cc49 CSRSparseMatrix matrix; in Compute()
69 const CSRSparseMatrix& x, in CSRSparseMatrixZerosLikeHelper()
70 CSRSparseMatrix* y) { in CSRSparseMatrixZerosLikeHelper()
87 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, CSRSparseMatrix,
Dsparse_mat_mul_op.cc104 const CSRSparseMatrix* input_matrix_a; in Compute()
105 const CSRSparseMatrix* input_matrix_b; in Compute()
247 CSRSparseMatrix output_csr_matrix; in Compute()
248 OP_REQUIRES_OK(ctx, CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
266 const CSRSparseMatrix& csr_matrix, const int rank, const int batch_index, in GetSparseMatrixRef()
315 const CSRSparseMatrix* a_matrix; in Compute()
316 const CSRSparseMatrix* b_matrix; in Compute()
370 CSRSparseMatrix c; in Compute()
391 const CSRSparseMatrix* a_input_matrix; in Compute()
394 CSRSparseMatrix a_matrix_transposed; in Compute()
[all …]
Dadd_op.cc52 Status operator()(const CSRSparseMatrix& a, const CSRSparseMatrix& b, in operator ()()
53 CSRSparseMatrix* c) { in operator ()()
158 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in operator ()()
202 const CSRSparseMatrix* a_matrix; in Compute()
203 const CSRSparseMatrix* b_matrix; in Compute()
233 CSRSparseMatrix c_matrix; in Compute()
264 ADD_VARIANT_BINARY_OP, DEVICE_GPU, CSRSparseMatrix,
Dkernels_gpu.cu.cc285 const CSRSparseMatrix& a, in CSRSparseMatrixBatchMulVecImpl()
287 CSRSparseMatrix* c) { in CSRSparseMatrixBatchMulVecImpl()
293 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in CSRSparseMatrixBatchMulVecImpl()
332 OpKernelContext* ctx, const CSRSparseMatrix& a, \
333 typename TTypes<T>::ConstFlat b, CSRSparseMatrix* c) { \
420 const CSRSparseMatrix& logits, in CSRSparseMatrixSoftmaxGPUImpl()
468 OpKernelContext* ctx, const CSRSparseMatrix& logits, \
608 OpKernelContext* ctx, const CSRSparseMatrix& softmax, in CSRSparseMatrixSoftmaxGradGPUImpl()
609 const CSRSparseMatrix& grad_softmax, in CSRSparseMatrixSoftmaxGradGPUImpl()
675 OpKernelContext* ctx, const CSRSparseMatrix& softmax, \
[all …]
Dtranspose_op.cc102 const CSRSparseMatrix* input_matrix; in Compute()
112 CSRSparseMatrix output_matrix; in Compute()
151 OpKernelContext* ctx, bool conjugate, const CSRSparseMatrix& input_matrix, in operator ()()
152 CSRSparseMatrix* output_matrix) { in operator ()()
182 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in operator ()()
Dsparse_cholesky_op.cc79 const CSRSparseMatrix* input_matrix; in Compute()
212 CSRSparseMatrix output_csr_matrix; in Compute()
215 CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
230 const CSRSparseMatrix& sparse_matrix, in ValidateInputs()
Dzeros_op.h44 const Tensor& dense_shape_t, CSRSparseMatrix* matrix) { in operator()
77 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in operator()
Dsparse_tensor_to_csr_sparse_matrix_op.cc97 CSRSparseMatrix output_csr_matrix; in Compute()
99 ctx, CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
262 CSRSparseMatrix matrix; in ComputeAsync()
265 CSRSparseMatrix::CreateCSRSparseMatrix( in ComputeAsync()
Dcsr_sparse_matrix_to_sparse_tensor_op.cc46 Status ValidateCSRSparseMatrix(const CSRSparseMatrix& csr_sparse_matrix, in ValidateCSRSparseMatrix()
73 const CSRSparseMatrix* csr_sparse_matrix; in Compute()
142 const CSRSparseMatrix* csr_sparse_matrix; in Compute()
Ddense_to_csr_sparse_matrix_op.cc119 CSRSparseMatrix output_csr_matrix; in Compute()
120 OP_REQUIRES_OK(ctx, CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
329 CSRSparseMatrix matrix; in ComputeAsync()
332 CSRSparseMatrix::CreateCSRSparseMatrix( in ComputeAsync()
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_test.py50 a_sm = sparse_csr_matrix_ops.CSRSparseMatrix(a_st)
69 a_sm = sparse_csr_matrix_ops.CSRSparseMatrix(a_mats)
91 a_sm = sparse_csr_matrix_ops.CSRSparseMatrix(a_mats, indices=indices)
113 a_sm = sparse_csr_matrix_ops.CSRSparseMatrix(a_mats)
115 self.assertIsInstance(a_sm_conj, sparse_csr_matrix_ops.CSRSparseMatrix)
134 a_sm = sparse_csr_matrix_ops.CSRSparseMatrix(a_mats)
139 self.assertIsInstance(a_sm_t, sparse_csr_matrix_ops.CSRSparseMatrix)
163 a_sm = sparse_csr_matrix_ops.CSRSparseMatrix(a_mats)
164 b_sm = sparse_csr_matrix_ops.CSRSparseMatrix(b_mats)
179 self.assertIsInstance(c_sm, sparse_csr_matrix_ops.CSRSparseMatrix)
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseMatrixTranspose.pbtxt6 description: "A CSRSparseMatrix."
10 description: "A CSRSparseMatrix."
16 summary: "Transposes the inner (matrix) dimensions of a CSRSparseMatrix."
Dapi_def_SparseMatrixAdd.pbtxt6 description: "A CSRSparseMatrix."
10 description: "A CSRSparseMatrix."
22 description: "A CSRSparseMatrix."
Dapi_def_SparseMatrixSoftmax.pbtxt6 description: "A CSRSparseMatrix."
10 description: "A CSRSparseMatrix."
12 summary: "Calculates the softmax of a CSRSparseMatrix."
Dapi_def_CSRSparseMatrixToDense.pbtxt6 description: "A batched CSRSparseMatrix."
12 summary: "Convert a (possibly batched) CSRSparseMatrix to dense."
Dapi_def_DenseToCSRSparseMatrix.pbtxt14 description: "A (possibly batched) CSRSparseMatrix."
16 summary: "Converts a dense tensor to a (possibly batched) CSRSparseMatrix."
Dapi_def_SparseTensorToCSRSparseMatrix.pbtxt18 description: "A (possibly batched) CSRSparseMatrix."
20 summary: "Converts a SparseTensor to a (possibly batched) CSRSparseMatrix."
Dapi_def_SparseMatrixSparseMatMul.pbtxt6 description: "A CSRSparseMatrix."
10 description: "A CSRSparseMatrix."
14 description: "A CSRSparseMatrix."
103 a: A `CSRSparseMatrix`.
104 b: A `CSRSparseMatrix` with the same type and rank as `a`.
Dapi_def_SparseMatrixSparseCholesky.pbtxt6 description: "A `CSRSparseMatrix`."
74 # Convert the CSRSparseMatrix Cholesky factor to a dense Tensor
92 input: A `CSRSparseMatrix`.

12