Home
last modified time | relevance | path

Searched defs:batch_size_limit (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dinput_split_metadata.cc29 int batch_size_limit) { in compute_task_size_from_open_batch()
36 int batch_size_limit) { in compute_head_task_size()
45 int batch_size_limit) { in compute_tail_task_size()
58 int compute_num_batches(int task_size_from_open_batch, int batch_size_limit) { in compute_num_batches()
65 int batch_size_limit) in InputSplitMetadata()
H A Dinput_split_metadata_test.cc40 const int batch_size_limit = std::get<2>(batch_task_param); in TEST() local
H A Dbatch_input_task.h217 int batch_size_limit, in BatchInputTask()
H A Dshared_batch_scheduler_test.cc459 const size_t batch_size_limit = 100; in TEST_P() local
690 const size_t batch_size_limit = 10; in TEST_P() local