Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc265 const int64 row_begin, const int64 num_shard_rows, in GetSparseMatrixRef()
307 [&](int64 batch_idx, int64 row_begin, int64 row_end) { in SparseDenseMatMulWithoutTransposedLHS()
378 [&](int64 batch_idx, int64 row_begin, int64 row_end) { in SparseDenseMatMulWithTransposedLHS()
443 const int64 row_begin = current_batch_row_begin % num_rows; in HandleBatchAndRowRange() local
Dtranspose_op.cc241 const int64 row_begin = input.row_ptr(row_idx); in operator ()() local
/external/tensorflow/tensorflow/core/kernels/
Dmatrix_band_part_op.cc157 const int64 row_begin = begin > batch * m ? begin % m : 0; in operator ()() local