Home
last modified time | relevance | path

Searched defs:num_shards (Results 1 – 25 of 27) sorted by relevance

12

/third_party/mindspore/tests/ut/python/dataset/
Dtest_minddataset_padded.py177 def partitions(num_shards, num_padded, dataset_size): argument
217 def partitions(num_shards, num_padded, dataset_size): argument
289 def partitions(num_shards, num_padded): argument
314 def partitions(num_shards, num_padded): argument
336 def partitions(num_shards, num_padded): argument
360 def partitions(num_shards, num_padded): argument
384 def partitions(num_shards, num_padded): argument
407 def partitions(num_shards, num_padded): argument
433 def partitions(num_shards, num_padded, dataset_size): argument
474 def partitions(num_shards, num_padded, dataset_size): argument
[all …]
Dtest_datasets_sharding.py22 def sharding_config(num_shards, shard_id, num_samples, shuffle, class_index, repeat_cnt=1): argument
57 def sharding_config(num_shards, shard_id, num_samples, repeat_cnt=1): argument
95 def sharding_config(num_shards, shard_id, num_samples, repeat_cnt=1): argument
139 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
164 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
192 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
212 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
231 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
Dtest_minddataset.py254 def partitions(num_shards): argument
276 def partitions(num_shards): argument
301 def partitions(num_shards): argument
326 def partitions(num_shards, expect): argument
Dtest_minddataset_exception.py285 def partitions(num_shards): argument
Dtest_datasets_tfrecord.py214 def get_res(num_shards, shard_id, num_repeats): argument
Dtest_sampler.py250 def test_config(num_shards, shard_id): argument
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/meta/
Dshard_distributed_sample.cc25 ShardDistributedSample::ShardDistributedSample(int num_shards, int shard_id, int no_of_padded_sampl… in ShardDistributedSample()
34 ShardDistributedSample::ShardDistributedSample(int num_shards, int shard_id, bool shuffle, uint32_t… in ShardDistributedSample()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/samplers/
Ddistributed_sampler_ir.cc34 DistributedSamplerObj::DistributedSamplerObj(int64_t num_shards, int64_t shard_id, bool shuffle, in… in DistributedSamplerObj()
119 int64_t num_shards = json_obj["num_shards"]; in from_json() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dtf_record_node.h40int32_t num_shards, int32_t shard_id, bool shard_equal_rows, std::shared_ptr<DatasetCache> cache) in TFRecordNode()
61int32_t num_shards, int32_t shard_id, bool shard_equal_rows, std::shared_ptr<DatasetCache> cache) in TFRecordNode()
Dtext_file_node.cc27int32_t num_shards, int32_t shard_id, std::shared_ptr<DatasetCache> cache) in TextFileNode()
167 int32_t num_shards = json_obj["num_shards"]; in from_json() local
Dcsv_node.cc29 int32_t num_shards, int32_t shard_id, std::shared_ptr<DatasetCache> cache) in CSVNode()
206 int32_t num_shards = json_obj["num_shards"]; in from_json() local
Dclue_node.cc27 … ShuffleMode shuffle, int32_t num_shards, int32_t shard_id, std::shared_ptr<DatasetCache> cache) in CLUENode()
266 int32_t num_shards = json_obj["num_shards"]; in from_json() local
Dusps_node.cc31 int32_t num_shards, int32_t shard_id, std::shared_ptr<DatasetCache> cache) in USPSNode()
Dtf_record_node.cc256 int32_t num_shards = json_obj["num_shards"]; in from_json() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/
Dbindings.cc108 int32_t shuffle, int32_t num_shards, int32_t shard_id) { in __anon9c4d722b0902()
133 int32_t num_shards, int32_t shard_id) { in __anon9c4d722b0d02()
297 int32_t num_shards, int32_t shard_id, bool shard_equal_rows) { in __anon9c4d722b2602()
319 int32_t num_shards, int32_t shard_id) { in __anon9c4d722b2902()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py1881 …def __init__(self, num_parallel_workers=None, num_samples=None, shuffle=True, num_shards=None, sha… argument
1958 …t__(self, num_parallel_workers=None, sampler=None, num_samples=None, shuffle=None, num_shards=None, argument
3381 … extensions=None, class_indexing=None, decode=False, num_shards=None, shard_id=None, cache=None): argument
3503 sampler=None, num_shards=None, shard_id=None, cache=None): argument
3603 …__(self, dataset_file, columns_list=None, num_parallel_workers=None, shuffle=None, num_shards=None, argument
4145 num_parallel_workers=1, shuffle=None, sampler=None, num_shards=None, shard_id=None, argument
4298 … shuffle=Shuffle.GLOBAL, num_shards=None, shard_id=None, shard_equal_rows=False, cache=None): argument
4401 … sampler=None, class_indexing=None, decode=False, num_shards=None, shard_id=None, cache=None): argument
4556 sampler=None, num_shards=None, shard_id=None, cache=None): argument
4680 sampler=None, num_shards=None, shard_id=None, cache=None): argument
[all …]
Dsamplers.py29 def select_sampler(num_samples, input_sampler, shuffle, num_shards, shard_id): argument
354 def __init__(self, num_shards, shard_id, shuffle=True, num_samples=None, offset=-1): argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/
Ddistributed_sampler.cc25 DistributedSamplerRT::DistributedSamplerRT(int64_t num_shards, int64_t shard_id, bool shuffle, int6… in DistributedSamplerRT()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dsamplers.cc32 DistributedSampler::DistributedSampler(int64_t num_shards, int64_t shard_id, bool shuffle, int64_t … in DistributedSampler()
Ddatasets.cc974 … const std::vector<char> &usage, int64_t num_samples, ShuffleMode shuffle, int32_t num_shards, in CLUEDataset()
1010int32_t num_shards, int32_t shard_id, const std::shared_ptr<DatasetCache> &cache) { in CSVDataset()
1254 ShuffleMode shuffle, int32_t num_shards, int32_t shard_id, in TextFileDataset()
1262 ShuffleMode shuffle, int32_t num_shards, int32_t shard_id, in USPSDataset()
1334 … ShuffleMode shuffle, int32_t num_shards, int32_t shard_id, bool shard_equal_rows, in TFRecordDataset()
1343 … ShuffleMode shuffle, int32_t num_shards, int32_t shard_id, bool shard_equal_rows, in TFRecordDataset()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/post/
Dauto_worker_pass.cc35 int32_t num_shards = GlobalContext::config_manager()->get_num_shards_for_auto_num_workers(); in RunOnTree() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dconfig_manager.h207 …void set_num_shards_for_auto_num_workers(int32_t num_shards) { auto_num_workers_num_shards_ = num_… in set_num_shards_for_auto_num_workers()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Ddataset_node.cc126 Status ValidateDatasetShardParams(const std::string &dataset_name, int32_t num_shards, int32_t shar… in ValidateDatasetShardParams()
196 std::shared_ptr<SamplerObj> SelectSampler(int64_t num_samples, bool shuffle, int32_t num_shards, in… in SelectSampler()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/samplers/
Dbindings.cc48 uint32_t seed, int64_t offset, bool even_dist) { in __anonc9e25c810302()
/third_party/mindspore/tests/ut/data/dataset/tf_file_dataset/
Dpyvision_dataset_pipeline.json1 …es/train-0000-of-0001.data"],"num_parallel_workers":8,"num_samples":0,"num_shards":1,"op_type":"TF… number

12