Searched defs:batch_begin (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_cholesky_op.cc | 114 [&](int64_t batch_begin, int64_t batch_end) { in Compute()
193 [&](int64_t batch_begin, int64_t batch_end) { in Compute()
|
| D | sparse_ordering_amd_op.cc | 99 amd_cost_per_batch, [&](int64_t batch_begin, int64_t batch_end) { in Compute()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 99 auto shard = [&](int64_t batch_begin, int64_t batch_end) { in Compute()
|
| D | csr_sparse_matrix_to_dense_op.cc | 95 auto shard = [&](int64_t batch_begin, int64_t batch_end) { in Compute()
|
| D | sparse_mat_mul_op.cc | 186 matmul_cost_per_batch, [&](int64_t batch_begin, int64_t batch_end) { in Compute()
228 [&](int64_t batch_begin, int64_t batch_end) { in Compute()
|
| D | mat_mul_op.cc | 439 const int64_t batch_begin = batch_and_row_begin / num_rows; in HandleBatchAndRowRange() local
|
| /external/mesa3d/src/gallium/drivers/d3d12/ |
| D | d3d12_resource_state.h | 83 struct d3d12_resource_state batch_begin, batch_end; member
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_band_part_op.cc | 155 const int64_t batch_begin = begin / m; in operator ()() local
|