Home
last modified time | relevance | path

Searched refs:dense_shape_host (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dtranspose_op.cc55 const int rank = input.dense_shape_host.size(); in ValidateTransposeInputs()
58 const int num_cols = input.dense_shape_host(rank - 1); in ValidateTransposeInputs()
70 if (rank != output.dense_shape_host.size()) { in ValidateTransposeInputs()
73 output.dense_shape_host.size()); in ValidateTransposeInputs()
75 if (num_rows != output.dense_shape_host(rank - 1)) { in ValidateTransposeInputs()
78 " vs. ", output.dense_shape_host(rank - 1)); in ValidateTransposeInputs()
223 const int rank = input.dense_shape_host.size(); in operator ()()
225 const int num_cols = input.dense_shape_host(rank - 1); in operator ()()
266 const int rank = x.dense_shape_host.size(); in operator ()()
268 const int n = x.dense_shape_host(rank - 1); in operator ()()
[all …]
Dsparse_mat_mul_op.cc606 a.dense_shape_host(a.dense_shape_host.size() - (transpose_a_ ? 1 : 2)); in CSRSparseSparseMatrixMatMul()
611 a.dense_shape_host(a.dense_shape_host.size() - (transpose_a_ ? 2 : 1)); in CSRSparseSparseMatrixMatMul()
619 b.dense_shape_host(b.dense_shape_host.size() - (transpose_b_ ? 2 : 1)); in CSRSparseSparseMatrixMatMul()
637 a.dense_shape_host(a.dense_shape_host.size() - (transpose_a_ ? 1 : 2)); in CSRSparseSparseMatrixMatMul()
643 a.dense_shape_host(a.dense_shape_host.size() - (transpose_a_ ? 2 : 1)); in CSRSparseSparseMatrixMatMul()
651 b.dense_shape_host(b.dense_shape_host.size() - (transpose_b_ ? 2 : 1)); in CSRSparseSparseMatrixMatMul()
680 a.dense_shape_host(a.dense_shape_host.size() - (transpose_a_ ? 1 : 2)); in CSRSparseSparseMatrixMatMul()
684 DCHECK_EQ(m, c->dense_shape_host(c->dense_shape_host.size() - 2)); in CSRSparseSparseMatrixMatMul()
687 a.dense_shape_host(a.dense_shape_host.size() - (transpose_a_ ? 2 : 1)); in CSRSparseSparseMatrixMatMul()
695 b.dense_shape_host(b.dense_shape_host.size() - (transpose_b_ ? 2 : 1)); in CSRSparseSparseMatrixMatMul()
[all …]
Dadd_op.cc298 const int row_dim = a.dense_shape_host.size() == 2 ? 0 : 1; in GetWorkspaceSize()
299 DCHECK_EQ(m, a.dense_shape_host(row_dim)); in GetWorkspaceSize()
300 DCHECK_EQ(m, b.dense_shape_host(row_dim)); in GetWorkspaceSize()
304 const int n = a.dense_shape_host(row_dim + 1); in GetWorkspaceSize()
305 DCHECK_EQ(n, b.dense_shape_host(row_dim + 1)); in GetWorkspaceSize()
326 const int row_dim = a.dense_shape_host.size() == 2 ? 0 : 1; in GetOutputStructure()
327 DCHECK_EQ(m, a.dense_shape_host(row_dim)); in GetOutputStructure()
328 DCHECK_EQ(m, b.dense_shape_host(row_dim)); in GetOutputStructure()
333 const int n = a.dense_shape_host(row_dim + 1); in GetOutputStructure()
334 DCHECK_EQ(n, b.dense_shape_host(row_dim + 1)); in GetOutputStructure()
[all …]
Dmat_mul_op.cc799 DCHECK_EQ(k, a.dense_shape_host(1)); in Compute()
808 DCHECK_EQ(m, a.dense_shape_host(0)); in Compute()
950 const int m = a.dense_shape_host(0); in Compute()
951 const int n = a.dense_shape_host(1); in Compute()
Dsparse_matrix.h621 TTypes<int64>::ConstVec dense_shape_host; member
629 TTypes<int64>::Vec dense_shape_host; member