Home
last modified time | relevance | path

Searched refs: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() argument
113 for (int64 batch_index = batch_begin; batch_index < batch_end; in Compute()
191 [&](int64 batch_begin, int64 batch_end) { in Compute() argument
192 for (int64 batch_index = batch_begin; batch_index < batch_end; in Compute()
Dsparse_ordering_amd_op.cc99 amd_cost_per_batch, [&](int64 batch_begin, int64 batch_end) { in Compute() argument
100 for (int64 batch_index = batch_begin; batch_index < batch_end; in Compute()
Dcsr_sparse_matrix_to_sparse_tensor_op.cc99 auto shard = [&](int64 batch_begin, int64 batch_end) { in Compute() argument
100 for (int64 batch_idx = batch_begin; batch_idx < batch_end; ++batch_idx) { in Compute()
Dcsr_sparse_matrix_to_dense_op.cc95 auto shard = [&](int64 batch_begin, int64 batch_end) { in Compute() argument
96 for (int64 batch_idx = batch_begin; batch_idx < batch_end; ++batch_idx) { in Compute()
Dsparse_mat_mul_op.cc186 matmul_cost_per_batch, [&](int64 batch_begin, int64 batch_end) { in Compute() argument
187 for (int64 batch_idx = batch_begin; batch_idx < batch_end; in Compute()
228 [&](int64 batch_begin, int64 batch_end) { in Compute() argument
229 for (int64 batch_idx = batch_begin; batch_idx < batch_end; in Compute()
Dmat_mul_op.cc430 const int64 batch_begin = batch_and_row_begin / num_rows; in HandleBatchAndRowRange() local
433 for (int64 batch_idx = batch_begin; batch_idx <= batch_end_inclusive; in HandleBatchAndRowRange()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_band_part_op.cc155 const int64 batch_begin = begin / m; in operator ()() local
157 for (int64 batch = batch_begin; batch < batch_end; ++batch) { in operator ()()