Home
last modified time | relevance | path

Searched defs:num_shard_rows (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc265 const int64_t row_begin, const int64_t num_shard_rows, in GetSparseMatrixRef()
311 const int64_t num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithoutTransposedLHS() local
382 const int64_t num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithTransposedLHS() local
447 const int64_t num_shard_rows = in HandleBatchAndRowRange() local