Searched refs:shuffle_op (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/ |
| D | usps_node.cc | 92 std::shared_ptr<DatasetOp> shuffle_op = nullptr; in Build() local
99 …(AddShuffleOp(op->FileNames().size(), num_shards_, num_rows, 0, connector_que_size_, &shuffle_op)); in Build()
100 shuffle_op->SetTotalRepeats(GetTotalRepeats()); in Build()
101 shuffle_op->SetNumRepeatsPerEpoch(GetNumRepeatsPerEpoch()); in Build()
102 node_ops->push_back(shuffle_op); in Build()
|
| D | text_file_node.cc | 96 std::shared_ptr<DatasetOp> shuffle_op = nullptr; in Build() local
104 …huffleOp(sorted_dataset_files.size(), num_shards_, num_rows, 0, connector_que_size_, &shuffle_op)); in Build()
105 shuffle_op->SetTotalRepeats(GetTotalRepeats()); in Build()
106 shuffle_op->SetNumRepeatsPerEpoch(GetNumRepeatsPerEpoch()); in Build()
107 node_ops->push_back(shuffle_op); in Build()
|
| D | csv_node.cc | 128 std::shared_ptr<DatasetOp> shuffle_op = nullptr; in Build() local
136 …huffleOp(sorted_dataset_files.size(), num_shards_, num_rows, 0, connector_que_size_, &shuffle_op)); in Build()
137 shuffle_op->SetTotalRepeats(GetTotalRepeats()); in Build()
138 shuffle_op->SetNumRepeatsPerEpoch(GetNumRepeatsPerEpoch()); in Build()
139 node_ops->push_back(shuffle_op); in Build()
|
| D | clue_node.cc | 189 std::shared_ptr<DatasetOp> shuffle_op = nullptr; in Build() local
197 …huffleOp(sorted_dataset_files.size(), num_shards_, num_rows, 0, connector_que_size_, &shuffle_op)); in Build()
198 shuffle_op->SetTotalRepeats(GetTotalRepeats()); in Build()
199 shuffle_op->SetNumRepeatsPerEpoch(GetNumRepeatsPerEpoch()); in Build()
200 node_ops->push_back(shuffle_op); in Build()
|
| D | tf_record_node.cc | 144 std::shared_ptr<DatasetOp> shuffle_op = nullptr; in Build() local
151 …AddShuffleOp(sorted_dir_files.size(), num_shards_, num_rows, 0, connector_que_size_, &shuffle_op)); in Build()
152 shuffle_op->SetTotalRepeats(GetTotalRepeats()); in Build()
153 shuffle_op->SetNumRepeatsPerEpoch(GetNumRepeatsPerEpoch()); in Build()
154 node_ops->push_back(shuffle_op); in Build()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | CMakeLists.txt | 18 shuffle_op.cc
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | dataset_node.cc | 63 int32_t connector_que_size, std::shared_ptr<DatasetOp> *shuffle_op) { in AddShuffleOp() argument
64 RETURN_UNEXPECTED_IF_NULL(shuffle_op); in AddShuffleOp()
69 *shuffle_op = std::make_shared<ShuffleOp>(shuffle_size, GetSeed(), connector_que_size, true); in AddShuffleOp()
|
| D | dataset_node.h | 100 int32_t connector_que_size, std::shared_ptr<DatasetOp> *shuffle_op);
|
| /third_party/mindspore/mindspore/lite/minddata/ |
| D | CMakeLists.txt | 161 ${MINDDATA_DIR}/engine/datasetops/shuffle_op.cc
|
| /third_party/mindspore/mindspore/lite/tools/dataset/cropper/ |
| D | dependencies.txt | 1 …shuffle_op.cc", "mindspore/core/base/core_ops.h", "mindspore/ccsrc/minddata/dataset/engine/ir/data…
|