Searched refs:batch_map_func (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.h | 99 Builder &SetBatchMapFunc(py::function batch_map_func) { in SetBatchMapFunc() argument
100 builder_batch_map_func_ = batch_map_func; in SetBatchMapFunc()
155 py::function batch_size_func, py::function batch_map_func, PadInfo pad_map);
|
| D | batch_op.cc | 54 py::function batch_size_func, py::function batch_map_func, PadInfo pad_map) in BatchOp() argument
62 batch_map_func_(batch_map_func), in BatchOp()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | batch_node.cc | 36 py::function batch_map_func, in BatchNode() argument
45 batch_map_func_(batch_map_func), in BatchNode()
|
| D | batch_node.h | 38 …nst std::vector<std::string> &col_order, py::function batch_size_func, py::function batch_map_func,
|