Searched refs:row_begin (Results 1 – 4 of 4) 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
270 csr_matrix.row_pointers_vec(batch_index)(row_begin); in GetSparseMatrixRef()
273 csr_matrix.row_pointers_vec(batch_index)(row_begin + row_idx) - in GetSparseMatrixRef()
307 [&](int64 batch_idx, int64 row_begin, int64 row_end) { in SparseDenseMatMulWithoutTransposedLHS() argument
308 const int64 num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithoutTransposedLHS()
314 lhs, batch_idx, row_begin, num_shard_rows, &row_ptrs); in SparseDenseMatMulWithoutTransposedLHS()
326 row_begin * num_rhs_cols, in SparseDenseMatMulWithoutTransposedLHS()
378 [&](int64 batch_idx, int64 row_begin, int64 row_end) { in SparseDenseMatMulWithTransposedLHS() argument
379 const int64 num_shard_rows = row_end - row_begin; in SparseDenseMatMulWithTransposedLHS()
385 lhs, batch_idx, row_begin, num_shard_rows, &row_ptrs); in SparseDenseMatMulWithTransposedLHS()
[all …]
|
| D | transpose_op.cc | 241 const int64 row_begin = input.row_ptr(row_idx); in operator ()() local
243 for (int64 i = row_begin; i < row_end; ++i) { in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_band_part_op.cc | 158 const int64 row_begin = begin > batch * m ? begin % m : 0; in operator ()() local
160 for (int64 row = row_begin; row < row_end; ++row) { in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 176 const int64 row_begin = begin[0]; in Compute() local
184 const int64 row = row_begin + i; in Compute()
|