| /third_party/mindspore/mindspore-src/source/tests/st/probability/bnn_layers/ |
| D | dataset.py | 26 def create_dataset(data_path, batch_size=32, repeat_size=1, function
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/resnet50/src/ |
| D | dataset.py | 25 def create_dataset(dataset_path, do_train, repeat_num=1, batch_size=32): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/export_and_load/ |
| D | op_parallel.py | 57 def create_dataset(batch_size): function
|
| D | test_ckpt_d2h_async.py | 77 def create_dataset(data_path, batch_size=32, num_parallel_workers=1): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/train/ |
| D | test_early_stopping_and_reduce_LR_on_plateau_callbacks.py | 40 def create_dataset(num_data, batch_size=512, repeat_size=1): function
|
| D | test_fit.py | 36 def create_dataset(num_data, batch_size=16, repeat_size=1): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/map_parameter/network/src/ |
| D | dataset.py | 33 def create_dataset(epoch_size=1, batch_size=32, step_size=1, resize_height=224, function
|
| /third_party/mindspore/mindspore-src/source/tests/st/place/simple_dynamic_shape/src/ |
| D | dynamic_embedding.py | 38 def create_dataset(): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/dataset/ |
| D | test_gpu_python_obj.py | 28 def create_dataset(size, needs_batch): function
|
| D | test_gpu_autotune.py | 46 def create_dataset(data_path, batch_size=32, num_parallel_workers=1): function
|
| D | test_gpu_lenet.py | 96 def create_dataset(dataset_path, do_train, batch_size=32, train_image_size=28, target="Ascend", dis… function
|
| D | test_dataset_with_multiprocessing.py | 122 def create_dataset(): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/hypercomplex/ |
| D | test_resnet.py | 18 def create_dataset(dataset_dir, batch_size, usage=None): function
|
| D | test_mnist.py | 19 def create_dataset(dataset_dir, batch_size, usage=None): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/deeplabv3/src/ |
| D | md_dataset.py | 81 def create_dataset(args, data_url, epoch_num=1, batch_size=1, usage="train", shuffle=True): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/profiler/ |
| D | run_net_with_profiler.py | 85 def create_dataset(data_path, batch_size=32, repeat_size=1, num_parallel_workers=1): function
|
| D | run_net.py | 84 def create_dataset(data_path, batch_size=32, repeat_size=1, num_parallel_workers=1): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/mem_reuse/ |
| D | resnet_cifar_normal.py | 59 def create_dataset(repeat_num=1, training=True): function
|
| D | resnet_cifar_memreuse.py | 59 def create_dataset(repeat_num=1, training=True): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/compile_cache/ |
| D | run_lenet_ps.py | 85 def create_dataset(data_path, batch_size=32, repeat_size=1, function
|
| /third_party/mindspore/mindspore-src/source/tests/st/place/general_split/src/ |
| D | lenet.py | 25 def create_dataset(data_path, batch_size=32, repeat_size=1, rank_id=0, rank_size=1, function
|
| /third_party/mindspore/mindspore-src/source/tests/st/msrun/ |
| D | test_msrun.py | 86 def create_dataset(data_path, batch_size=32, repeat_size=1, function
|
| D | test_msrun_exception.py | 91 def create_dataset(data_path, batch_size=32, repeat_size=1, function
|
| /third_party/mindspore/mindspore-src/source/tests/st/ps/full_ps/ |
| D | test_full_ps_lenet.py | 87 def create_dataset(data_path, batch_size=32, repeat_size=1, function
|
| /third_party/mindspore/mindspore-src/source/tests/st/place/optimizer_split/src/ |
| D | net.py | 28 def create_dataset(data_path, batch_size=32, repeat_size=1, rank_id=0, rank_size=1, function
|