|
|
|
@@ -64,8 +64,7 @@ BatchOp::BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size, |
|
|
|
batch_num_(0), |
|
|
|
batch_cnt_(0) { |
|
|
|
// Adjust connector queue size. After batch each row is batch_size times larger |
|
|
|
int32_t queue_size; |
|
|
|
queue_size = std::max(1, op_queue_size / start_batch_size_); |
|
|
|
int32_t queue_size = std::max(1, op_queue_size / start_batch_size_); |
|
|
|
if (num_workers == 1) { |
|
|
|
// ensure there is at least 2 queue slots for whole operation.. If only 1 worker, incrase it to 2 |
|
|
|
queue_size = std::max(2, queue_size); |
|
|
|
@@ -85,8 +84,7 @@ BatchOp::BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size, |
|
|
|
pad_info_(pad_map), |
|
|
|
batch_num_(0), |
|
|
|
batch_cnt_(0) { |
|
|
|
int32_t queue_size; |
|
|
|
queue_size = std::max(1, op_queue_size / start_batch_size_); |
|
|
|
int32_t queue_size = std::max(1, op_queue_size / start_batch_size_); |
|
|
|
if (num_workers == 1) { |
|
|
|
// ensure there is at least 2 queue slots for whole operation.. If only 1 worker, incrase it to 2 |
|
|
|
queue_size = std::max(2, queue_size); |
|
|
|
|