Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_cholesky_op.cc114 [&](int64_t batch_begin, int64_t batch_end) { in Compute()
193 [&](int64_t batch_begin, int64_t batch_end) { in Compute()
H A Dsparse_mat_mul_op.cc186 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()
H A Dsparse_ordering_amd_op.cc99 amd_cost_per_batch, [&](int64_t batch_begin, int64_t batch_end) { in Compute()
H A Dcsr_sparse_matrix_to_dense_op.cc95 auto shard = [&](int64_t batch_begin, int64_t batch_end) { in Compute()
H A Dcsr_sparse_matrix_to_sparse_tensor_op.cc99 auto shard = [&](int64_t batch_begin, int64_t batch_end) { in Compute()
H A Dmat_mul_op.cc439 const int64_t batch_begin = batch_and_row_begin / num_rows; in HandleBatchAndRowRange() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/d3d12/
H A Dd3d12_resource_state.h83 struct d3d12_resource_state batch_begin, batch_end; member
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_band_part_op.cc155 const int64_t batch_begin = begin / m; in operator ()() local