Searched refs:num_shard_rows (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 265 const int64 row_begin, const int64 num_shard_rows, in GetSparseMatrixRef() argument
268 row_ptrs->resize(num_shard_rows + 1); in GetSparseMatrixRef()
271 for (int64 row_idx = 0; row_idx <= num_shard_rows; ++row_idx) { in GetSparseMatrixRef()
279 num_shard_rows /* num_rows */, num_cols /* num_cols */, in GetSparseMatrixRef()
280 row_ptrs->at(num_shard_rows) /* total_nnz */, row_ptrs->data(), in GetSparseMatrixRef()
308 const int64 num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithoutTransposedLHS() local
314 lhs, batch_idx, row_begin, num_shard_rows, &row_ptrs); in SparseDenseMatMulWithoutTransposedLHS()
327 num_shard_rows, num_rhs_cols); in SparseDenseMatMulWithoutTransposedLHS()
379 const int64 num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithTransposedLHS() local
385 lhs, batch_idx, row_begin, num_shard_rows, &row_ptrs); in SparseDenseMatMulWithTransposedLHS()
[all …]
|