| D | batch_op.cc | 31 BatchOp::Builder::Builder(int32_t batch_size) : builder_drop_(false), builder_pad_(false), builder_… in Builder()
38 Status BatchOp::Builder::Build(std::shared_ptr<BatchOp> *ptr) { in Build()
41 …*ptr = std::make_shared<BatchOp>(builder_batch_size_, builder_drop_, builder_pad_, builder_op_conn… in Build()
45 …*ptr = std::make_shared<BatchOp>(builder_batch_size_, builder_drop_, builder_pad_, builder_op_conn… in Build()
52 BatchOp::BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size, int32_t num_worker… in BatchOp() function in mindspore::dataset::BatchOp
77 BatchOp::BatchOp(int32_t batch_size, bool drop, bool pad, int32_t op_queue_size, int32_t num_worker… in BatchOp() function in mindspore::dataset::BatchOp
96 Status BatchOp::operator()() { in operator ()()
155 void BatchOp::Print(std::ostream &out, bool show_all) const { in Print()
169 Status BatchOp::BatchRows(const std::unique_ptr<TensorQTable> *src, TensorRow *dest, dsize_t batch_… in BatchRows()
229 Status BatchOp::WorkerEntry(int32_t workerId) { in WorkerEntry()
[all …]
|