| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_minddataset_padded.py | 218 repeat_size = 5
233 data_set = data_set.repeat(repeat_size)
263 assert local_index == dataset_size * repeat_size
271 assert num_padded_iter == num_padded * repeat_size
272 assert num_iter == dataset_size * num_shards * repeat_size
472 repeat_size = 3
488 data_set = data_set.repeat(repeat_size)
514 assert local_index == dataset_size * repeat_size
520 assert num_padded_iter == num_padded * repeat_size
521 assert num_iter == dataset_size * num_shards * repeat_size
[all …]
|
| /third_party/mindspore/tests/st/quantization/lenet_quant/ |
| D | dataset.py | 26 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
58 mnist_ds = mnist_ds.repeat(repeat_size)
|
| /third_party/mindspore/tests/st/probability/bnn_layers/ |
| D | dataset.py | 26 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
58 mnist_ds = mnist_ds.repeat(repeat_size)
|
| /third_party/mindspore/tests/st/probability/transforms/ |
| D | dataset.py | 26 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
58 mnist_ds = mnist_ds.repeat(repeat_size)
|
| /third_party/mindspore/tests/st/probability/zhusuan/vae/ |
| D | utils.py | 27 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
70 mnist_ds = mnist_ds.repeat(repeat_size)
|
| D | vae_mnist.py | 121 repeat_size = 1
143 ds_train = create_dataset(os.path.join(mnist_path, "train"), batch_size, repeat_size)
|
| /third_party/mindspore/tests/st/probability/dpn/ |
| D | test_gpu_svi_vae.py | 62 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
86 mnist_ds = mnist_ds.repeat(repeat_size)
|
| D | test_gpu_svi_cvae.py | 72 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
96 mnist_ds = mnist_ds.repeat(repeat_size)
|
| D | test_gpu_vae_gan.py | 129 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
153 mnist_ds = mnist_ds.repeat(repeat_size)
|
| /third_party/mindspore/tests/st/probability/toolbox/ |
| D | test_uncertainty.py | 79 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
111 mnist_ds = mnist_ds.repeat(repeat_size)
|
| /third_party/mindspore/tests/st/ps/full_ps/ |
| D | test_full_ps_lenet.py | 88 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
120 mnist_ds = mnist_ds.repeat(repeat_size)
|
| /third_party/mindspore/tests/st/fl/cross_silo_femnist/ |
| D | test_cross_silo_femnist.py | 235 def create_dataset_from_folder(data_path, img_size, batch_size=32, repeat_size=1, num_parallel_work… argument
268 mnist_ds = mnist_ds.repeat(repeat_size)
317 …t = create_dataset_from_folder(train_path, args.img_size, args.client_batch_size, args.repeat_size)
|
| /third_party/mindspore/tests/st/ps/part_ps/ |
| D | test_ps_embedding_heterogeneous_conv2d_adam.py | 81 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
119 mnist_ds = mnist_ds.repeat(repeat_size)
|
| /third_party/mindspore/tests/st/profiler/ |
| D | test_profiler.py | 89 def create_dataset(data_path, batch_size=32, repeat_size=1, num_parallel_workers=1): argument
115 mnist_ds = mnist_ds.repeat(repeat_size)
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gradient_accumulation.py | 169 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
201 mnist_ds = mnist_ds.repeat(repeat_size)
|
| D | test_gpu_lenet.py | 153 def create_dataset(data_path, batch_size=32, repeat_size=1, argument
185 mnist_ds = mnist_ds.repeat(repeat_size)
|