Searched refs:sparse_matrix_a (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 104 Status ValidateInputs(const CSRSparseMatrix& sparse_matrix_a, in ValidateInputs() argument
107 if (sparse_matrix_a.dtype() != dense_tensor_b.dtype()) { in ValidateInputs()
110 DataTypeString(sparse_matrix_a.dtype()), in ValidateInputs()
113 *rank = sparse_matrix_a.dims(); in ValidateInputs()
121 if (sparse_matrix_a.batch_size() != *batch_size) { in ValidateInputs()
123 sparse_matrix_a.batch_size(), " vs. ", in ValidateInputs()
126 const auto& a_dense_shape = sparse_matrix_a.dense_shape().vec<int64>(); in ValidateInputs()
170 const CSRSparseMatrix* sparse_matrix_a; in Compute() local
171 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &sparse_matrix_a)); in Compute()
176 OP_REQUIRES_OK(ctx, this->ValidateInputs(*sparse_matrix_a, matrix_b, &rank, in Compute()
[all …]
|