Home
last modified time | relevance | path

Searched defs:num_parallel_workers (Results 1 – 25 of 43) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/tests/ut/python/dataset/
Dtest_pyfunc_multiprocess_autotune.py29 def create_pyfunc_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=None): argument
60 def create_pyop_cop_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=None): argument
105 def create_mixed_map_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=None… argument
133 def create_per_batch_map_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=… argument
162 def create_mp_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=None): argument
Dtest_reset.py76 def create_random_imagenet_dataset(repeat_size, sampler=None, num_parallel_workers=1, to_pil=False,… argument
252 def test_repeatable_reset_imagenet(sampler, num_parallel_workers, to_pil, batch_func): argument
319 def test_repeatable_reset_distributed(shard_id, num_parallel_workers, to_pil): argument
Dtest_pyfunc_multiprocess.py68 def create_dataset_pyop_multiproc(num_parallel_workers=None, max_rowsize=16, batch_size=32, repeat_… argument
Dtest_map_error_samples.py123 def run_replace_test(transforms, dataset_size, num_parallel_workers, python_multiprocessing, expect… argument
146 def run_skip_test(transforms, dataset_size, num_parallel_workers, python_multiprocessing, expected=… argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/engine/
Ddatasets_text.py131num_parallel_workers=None, shuffle=Shuffle.GLOBAL, num_shards=None, shard_id=None, cache=None): argument
234 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=S… argument
477 …it__(self, dataset_files, task='AFQMC', usage='train', num_samples=None, num_parallel_workers=None, argument
578 shard_id=None, num_parallel_workers=None, cache=None): argument
682 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=S… argument
773 def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=True, argument
896 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
1030 …um_samples=None, shuffle=Shuffle.GLOBAL, num_shards=None, shard_id=None, num_parallel_workers=None, argument
1140 num_shards=None, shard_id=None, num_parallel_workers=None, cache=None): argument
1248num_parallel_workers=None, shuffle=None, num_shards=None, shard_id=None, cache=None): argument
[all …]
Ddatasets_vision.py226 def __init__(self, dataset_dir, target_type=None, num_samples=None, num_parallel_workers=1, argument
362 …def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=None, decode=… argument
505 …def __init__(self, dataset_dir, num_parallel_workers=None, shuffle=None, usage='all', sampler=None… argument
629 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
737 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
890num_parallel_workers=None, shuffle=None, decode=None, sampler=None, num_shards=None, argument
1096 …(self, dataset_dir, annotation_file, task="Detection", num_samples=None, num_parallel_workers=None, argument
1300num_parallel_workers=None, shuffle=None, decode=None, sampler=None, num_shards=None, argument
1418 def __init__(self, dataset_dir, name, usage=None, num_samples=None, num_parallel_workers=None, argument
1489num_parallel_workers=None, shuffle=None, sampler=None, num_shards=None, shard_id=None, cache=None): argument
[all …]
Ddatasets_audio.py144 …def __init__(self, dataset_dir, name=None, num_samples=None, num_parallel_workers=None, shuffle=No… argument
256 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
389 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
510 def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=None, argument
614 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
806 num_parallel_workers=None, shuffle=None, sampler=None, num_shards=None, argument
903 def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=None, argument
Ddatasets_standard_format.py99num_parallel_workers=None, shuffle=Shuffle.GLOBAL, num_shards=None, shard_id=None, cache=None): argument
186 …def __init__(self, dataset_files, columns_list=None, num_parallel_workers=None, shuffle=None, num_… argument
349 …_(self, dataset_files, schema=None, columns_list=None, num_samples=None, num_parallel_workers=None, argument
Ddatasets.py329 def __init__(self, children=None, num_parallel_workers=None, cache=None): argument
556 def batch(self, batch_size, drop_remainder=False, num_parallel_workers=None, **kwargs): argument
651 …def padded_batch(self, batch_size, drop_remainder=False, num_parallel_workers=None, pad_info=None): argument
853 num_parallel_workers=None, **kwargs): argument
998 def filter(self, predicate, input_columns=None, num_parallel_workers=None): argument
2111 def __init__(self, children=None, num_parallel_workers=None, cache=None): argument
2123 def __init__(self, children=None, num_parallel_workers=None, cache=None): argument
2291 def __init__(self, children=None, num_parallel_workers=None, cache=None): argument
2303 def __init__(self, children=None, num_parallel_workers=None, cache=None): argument
2315 …def __init__(self, num_parallel_workers=None, num_samples=None, shuffle=True, num_shards=None, sha… argument
[all …]
/third_party/mindspore/mindspore-src/source/tests/st/dataset/
Dtest_gpu_reset.py31 def create_np_dataset(size, num_parallel_workers, python_multiprocessing): argument
83 def test_dataset_reset_sink(fast_recovery, num_parallel_workers, python_multiprocessing): argument
Dtest_gpu_autotune.py46 def create_dataset(data_path, batch_size=32, num_parallel_workers=1): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dgenerator_node.cc30 std::shared_ptr<SamplerObj> sampler, uint32_t num_parallel_workers) in GeneratorNode()
40 … int64_t source_len, std::shared_ptr<SamplerObj> sampler, uint32_t num_parallel_workers) in GeneratorNode()
/third_party/mindspore/mindspore-src/source/tests/st/probability/bnn_layers/
Ddataset.py27 num_parallel_workers=1): argument
/third_party/mindspore/mindspore-src/source/tests/ut/data/dataset/tf_file_dataset/
Dpyvision_dataset_pipeline.json1 …"../data/dataset/test_tf_file_3_images/train-0000-of-0001.data"],"num_parallel_workers":8,"connect… number
/third_party/mindspore/mindspore-src/source/tests/st/export_and_load/
Dtest_ckpt_d2h_async.py77 def create_dataset(data_path, batch_size=32, num_parallel_workers=1): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/
Dconfig.cc56 bool set_num_parallel_workers(int32_t num_parallel_workers) { in set_num_parallel_workers()
/third_party/mindspore/mindspore-src/source/tests/st/networks/models/resnet50/src_thor/
Ddataset.py170 def get_num_parallel_workers(num_parallel_workers): argument
/third_party/mindspore/mindspore-src/source/tests/st/profiler/
Drun_net_with_profiler.py85 def create_dataset(data_path, batch_size=32, repeat_size=1, num_parallel_workers=1): argument
Drun_net.py84 def create_dataset(data_path, batch_size=32, repeat_size=1, num_parallel_workers=1): argument
/third_party/mindspore/mindspore-src/source/tests/st/compile_cache/
Drun_lenet_ps.py86 num_parallel_workers=1): argument
/third_party/mindspore/mindspore-src/source/tests/st/place/general_split/src/
Dlenet.py26 num_parallel_workers=1): argument
/third_party/mindspore/mindspore-src/source/tests/st/msrun/
Dtest_msrun.py87 num_parallel_workers=1): argument
Dtest_msrun_exception.py92 num_parallel_workers=1): argument
/third_party/mindspore/mindspore-src/source/tests/st/ps/full_ps/
Dtest_full_ps_lenet.py88 num_parallel_workers=1): argument
/third_party/mindspore/mindspore-src/source/tests/st/place/optimizer_split/src/
Dnet.py29 num_parallel_workers=1): argument

12