Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_cholesky_op.cc112 [&](int64 batch_begin, int64 batch_end) { in Compute()
191 [&](int64 batch_begin, int64 batch_end) { in Compute()
Dsparse_ordering_amd_op.cc99 amd_cost_per_batch, [&](int64 batch_begin, int64 batch_end) { in Compute()
Dcsr_sparse_matrix_to_dense_op.cc95 auto shard = [&](int64 batch_begin, int64 batch_end) { in Compute()
Dcsr_sparse_matrix_to_sparse_tensor_op.cc99 auto shard = [&](int64 batch_begin, int64 batch_end) { in Compute()
Dsparse_mat_mul_op.cc186 matmul_cost_per_batch, [&](int64 batch_begin, int64 batch_end) { in Compute()
228 [&](int64 batch_begin, int64 batch_end) { in Compute()
Dmat_mul_op.cc430 const int64 batch_begin = batch_and_row_begin / num_rows; in HandleBatchAndRowRange() local
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_band_part_op.cc155 const int64 batch_begin = begin / m; in operator ()() local