Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py201 def __init__(self, children=None, num_parallel_workers=None, cache=None): argument
429 def batch(self, batch_size, drop_remainder=False, num_parallel_workers=None, per_batch_map=None, argument
631num_parallel_workers=None, python_multiprocessing=False, cache=None, callbacks=None, max_rowsize=1… argument
769 def filter(self, predicate, input_columns=None, num_parallel_workers=None): argument
1881 …def __init__(self, num_parallel_workers=None, num_samples=None, shuffle=True, num_shards=None, sha… argument
1958 …def __init__(self, num_parallel_workers=None, sampler=None, num_samples=None, shuffle=None, num_sh… argument
2164 …def __init__(self, input_dataset, batch_size, drop_remainder=False, num_parallel_workers=None, per… argument
2723num_parallel_workers=None, python_multiprocessing=False, cache=None, callbacks=None, max_rowsize=1… argument
2862 def __init__(self, input_dataset, predicate, input_columns=None, num_parallel_workers=None): argument
3380 …def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=None, sampler… argument
[all …]
Dgraphdata.py96 …def __init__(self, dataset_file, num_parallel_workers=None, working_mode='local', hostname='127.0.… argument
/third_party/mindspore/tests/st/quantization/lenet_quant/
Ddataset.py27 num_parallel_workers=1): argument
/third_party/mindspore/tests/st/probability/bnn_layers/
Ddataset.py27 num_parallel_workers=1): argument
/third_party/mindspore/tests/st/probability/transforms/
Ddataset.py27 num_parallel_workers=1): argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dgenerator_node.cc31 std::shared_ptr<SamplerObj> sampler, uint32_t num_parallel_workers) in GeneratorNode()
42 … int64_t source_len, std::shared_ptr<SamplerObj> sampler, uint32_t num_parallel_workers) in GeneratorNode()
/third_party/mindspore/tests/st/probability/zhusuan/vae/
Dutils.py28 num_parallel_workers=1): argument
/third_party/mindspore/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,"num_sam… number
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dconfig.cc58 bool set_num_parallel_workers(int32_t num_parallel_workers) { in set_num_parallel_workers()
/third_party/mindspore/tests/st/probability/dpn/
Dtest_gpu_svi_vae.py63 num_parallel_workers=1): argument
Dtest_gpu_svi_cvae.py73 num_parallel_workers=1): argument
Dtest_gpu_vae_gan.py130 num_parallel_workers=1): argument
/third_party/mindspore/tests/st/probability/toolbox/
Dtest_uncertainty.py80 num_parallel_workers=1): argument
/third_party/mindspore/tests/st/ps/full_ps/
Dtest_full_ps_lenet.py89 num_parallel_workers=1): argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dgenerator_op.cc28 std::shared_ptr<SamplerRT> sampler, uint32_t num_parallel_workers) in GeneratorOp()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dconfig_manager.h79 int32_t num_parallel_workers() const { return num_parallel_workers_; } in num_parallel_workers() function
Dconfig_manager.cc123 Status ConfigManager::set_num_parallel_workers(int32_t num_parallel_workers) { in set_num_parallel_workers()
/third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/
Ddataset.py296 is_training=True, num_parallel_workers=8): argument
/third_party/mindspore/tests/st/profiler/
Dtest_profiler.py89 def create_dataset(data_path, batch_size=32, repeat_size=1, num_parallel_workers=1): argument
/third_party/mindspore/tests/st/networks/
Dtest_gpu_lenet.py154 num_parallel_workers=1): argument
Dtest_gradient_accumulation.py170 num_parallel_workers=1): argument
/third_party/mindspore/tests/st/ps/part_ps/
Dtest_ps_embedding_heterogeneous_conv2d_adam.py82 num_parallel_workers=1): argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/
Dbindings.cc170 uint32_t num_parallel_workers) { in __anon8ea7d5ca1302()
178 … int64_t dataset_len, py::handle sampler, uint32_t num_parallel_workers) { in __anon8ea7d5ca1302()
/third_party/mindspore/tests/st/fl/cross_silo_femnist/
Dtest_cross_silo_femnist.py235 …eate_dataset_from_folder(data_path, img_size, batch_size=32, repeat_size=1, num_parallel_workers=1, argument
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/
Ddataset.py459 num_parallel_workers=8, python_multiprocessing=False): argument

12