Searched refs:op_queue_size (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | filter_op.cc | 32 …:FilterOp(const std::vector<std::string> &in_col_names, int32_t num_workers, int32_t op_queue_size, in FilterOp() argument
34 …: ParallelOp(num_workers, op_queue_size), predicate_func_(std::move(predicate_func)), in_columns_(… in FilterOp()
35 worker_queues_.Init(num_workers, op_queue_size); in FilterOp()
|
| D | batch_op.cc | 52 BatchOp::BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size, int32_t num_worker… in BatchOp() argument
55 : ParallelOp(num_workers, op_queue_size), in BatchOp()
67 int32_t queue_size = std::max(1, op_queue_size / start_batch_size_); in BatchOp()
77 BatchOp::BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size, int32_t num_worker… in BatchOp() argument
79 : ParallelOp(num_workers, op_queue_size), in BatchOp()
87 int32_t queue_size = std::max(1, op_queue_size / start_batch_size_); in BatchOp()
|
| D | batch_op.h | 153 BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size, int32_t num_workers,
157 BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size, int32_t num_workers,
|
| D | filter_op.h | 43 FilterOp(const std::vector<std::string> &in_col_names, int32_t num_workers, int32_t op_queue_size,
|