p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

g# This is a combination of 2 commits. 

fix andorid build broke by pr6017
This commit is contained in:
Zirui Wu 2020-10-15 09:30:06 -04:00
parent c68f36c81e
commit 51eec90069
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/minddata/dataset/engine/datasetops/batch_op.cc
View File

@ -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);
}