Home
last modified time | relevance | path

Searched refs:shuffle_mode (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dminddata_node.cc31 ShuffleMode shuffle_mode, std::shared_ptr<DatasetCache> cache) in MindDataNode() argument
42 shuffle_mode_(shuffle_mode) {}
46 ShuffleMode shuffle_mode, std::shared_ptr<DatasetCache> cache) in MindDataNode() argument
57 shuffle_mode_(shuffle_mode) {}
137 int64_t num_padded, ShuffleMode shuffle_mode) { in BuildMindDatasetSamplerChain() argument
149 if (shuffle_mode != ShuffleMode::kFalse) { in BuildMindDatasetSamplerChain()
150 op->UpdateShuffleMode(shuffle_mode); in BuildMindDatasetSamplerChain()
Dminddata_node.h36 … ShuffleMode shuffle_mode = ShuffleMode::kGlobal, std::shared_ptr<DatasetCache> cache = nullptr);
41 … ShuffleMode shuffle_mode = ShuffleMode::kGlobal, std::shared_ptr<DatasetCache> cache = nullptr);
75 int64_t num_padded, ShuffleMode shuffle_mode);
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/include/
Dshard_operator.h56 …virtual void UpdateShuffleMode(dataset::ShuffleMode shuffle_mode) { shuffle_mode_ = shuffle_mode; } in UpdateShuffleMode() argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Ddatasets.h1946 ShuffleMode shuffle_mode = ShuffleMode::kGlobal,
1966 ShuffleMode shuffle_mode = ShuffleMode::kGlobal,
1986 int64_t num_padded, ShuffleMode shuffle_mode = ShuffleMode::kGlobal,
2006 ShuffleMode shuffle_mode = ShuffleMode::kGlobal,
2026 ShuffleMode shuffle_mode = ShuffleMode::kGlobal,
2046 ShuffleMode shuffle_mode = ShuffleMode::kGlobal,
2073 int64_t num_padded = 0, ShuffleMode shuffle_mode = ShuffleMode::kGlobal,
2076 padded_sample, num_padded, shuffle_mode, cache);
2098 ShuffleMode shuffle_mode = ShuffleMode::kGlobal,
2101 padded_sample, num_padded, shuffle_mode, cache);
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/
Dbindings.cc217 … const py::dict &padded_sample, int64_t num_padded, ShuffleMode shuffle_mode) { in __anon9c4d722b1a02() argument
223 … num_padded, shuffle_mode, nullptr); in __anon9c4d722b1a02()
229 … const py::dict &padded_sample, int64_t num_padded, ShuffleMode shuffle_mode) { in __anon9c4d722b1a02() argument
235 padded_sample_json, num_padded, shuffle_mode, nullptr); in __anon9c4d722b1a02()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc1144 int64_t num_padded, ShuffleMode shuffle_mode, in MindDataDataset() argument
1152 sample, num_padded, shuffle_mode, cache); in MindDataDataset()
1158 … const nlohmann::json *padded_sample, int64_t num_padded, ShuffleMode shuffle_mode, in MindDataDataset() argument
1166 sample, num_padded, shuffle_mode, cache); in MindDataDataset()
1173 int64_t num_padded, ShuffleMode shuffle_mode, in MindDataDataset() argument
1182 sample, num_padded, shuffle_mode, cache); in MindDataDataset()
1189 int64_t num_padded, ShuffleMode shuffle_mode, in MindDataDataset() argument
1198 sampler_obj, sample, num_padded, shuffle_mode, cache); in MindDataDataset()
1204 … const nlohmann::json *padded_sample, int64_t num_padded, ShuffleMode shuffle_mode, in MindDataDataset() argument
1213 sampler_obj, sample, num_padded, shuffle_mode, cache); in MindDataDataset()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dmindrecord_op.cc46 … const ShuffleMode shuffle_mode, std::unique_ptr<ShardReader> shard_reader, in MindRecordOp() argument
58 shuffle_mode_(shuffle_mode), in MindRecordOp()
/third_party/mindspore/tests/ut/cpp/dataset/
Dmind_record_op_test.cc57 ShuffleMode shuffle_mode = ShuffleMode::kGlobal; in CreateMindRecord() local
60 nullptr, sample_bytes, shuffle_mode, std::move(shard_reader), std::move(sampler)); in CreateMindRecord()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py96 shuffle_mode = cde.ShuffleMode.GLOBAL # Global shuffle
99 shuffle_mode = cde.ShuffleMode.GLOBAL # Global shuffle
101 shuffle_mode = cde.ShuffleMode.FALSE # No shuffle
103 shuffle_mode = ShuffleToShuffleMode[shuffle]
104 return shuffle_mode