|
|
|
@ -44,7 +44,7 @@ Status BatchOp::Builder::Build(std::shared_ptr<BatchOp> *ptr) {
|
|
|
|
|
builder_batch_size_func_, builder_batch_map_func_, builder_pad_map_);
|
|
|
|
|
#else
|
|
|
|
|
*ptr = std::make_shared<BatchOp>(builder_batch_size_, builder_drop_, builder_pad_, builder_op_connector_size_,
|
|
|
|
|
builder_num_workers_, builder_cols_to_map_, builder_pad_map_);
|
|
|
|
|
builder_num_workers_, builder_in_names_, builder_pad_map_);
|
|
|
|
|
#endif
|
|
|
|
|
return Status::OK();
|
|
|
|
|
}
|
|
|
|
@ -86,7 +86,7 @@ BatchOp::BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size,
|
|
|
|
|
start_batch_size_(batch_size),
|
|
|
|
|
drop_(drop),
|
|
|
|
|
pad_(pad),
|
|
|
|
|
pyfunc_column_names_(cols_to_map),
|
|
|
|
|
in_col_names_(cols_to_map),
|
|
|
|
|
pad_info_(pad_map) {
|
|
|
|
|
worker_queues_.Init(num_workers, op_queue_size);
|
|
|
|
|
}
|
|
|
|
|