Home
last modified time | relevance | path

Searched refs:MnistDataset (Results 1 – 25 of 35) sorted by relevance

12

/third_party/mindspore/tests/ut/python/dataset/
Dtest_datasets_mnist.py63 data1 = ds.MnistDataset(DATA_DIR, num_samples=100, shuffle=False)
84 data1 = ds.MnistDataset(DATA_DIR)
91 data2 = ds.MnistDataset(DATA_DIR, num_samples=500)
98 data3 = ds.MnistDataset(DATA_DIR, num_samples=200)
106 data4 = ds.MnistDataset(DATA_DIR, num_samples=100)
118 data5 = ds.MnistDataset(DATA_DIR, num_samples=100)
138 data = ds.MnistDataset(DATA_DIR, sampler=sampler)
155 data1 = ds.MnistDataset(DATA_DIR, sampler=sampler)
156 data2 = ds.MnistDataset(DATA_DIR, shuffle=False, num_samples=num_samples)
174 ds.MnistDataset(DATA_DIR, shuffle=False, sampler=ds.PKSampler(3))
[all …]
Dtest_datasets_get_dataset_size.py80 ds_total = ds.MnistDataset(MNIST_DATA_DIR)
84 test_size = ds.MnistDataset(MNIST_DATA_DIR, usage="test").get_dataset_size()
86 train_size = ds.MnistDataset(MNIST_DATA_DIR, usage="train").get_dataset_size()
88 all_size = ds.MnistDataset(MNIST_DATA_DIR, usage="all").get_dataset_size()
91 ds_shard_1_0 = ds.MnistDataset(MNIST_DATA_DIR, num_shards=1, shard_id=0)
94 ds_shard_2_0 = ds.MnistDataset(MNIST_DATA_DIR, num_shards=2, shard_id=0)
97 ds_shard_3_0 = ds.MnistDataset(MNIST_DATA_DIR, num_shards=3, shard_id=0)
253 dataset1 = ds.MnistDataset(MNIST_DATA_DIR, num_samples=2000, num_shards=4, shard_id=0)
262 dataset2 = ds.MnistDataset(MNIST_DATA_DIR, num_samples=3000, num_shards=4, shard_id=0)
Dtest_get_size.py133 data = ds.MnistDataset("../data/dataset/testMnistData")
137 data = ds.MnistDataset("../data/dataset/testMnistData", num_samples=10)
140 data = ds.MnistDataset("../data/dataset/testMnistData", num_samples=90000)
Dtest_random_solarize_op.py80 mnist_1 = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
81 mnist_2 = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
Dtest_equalize.py216 data_set = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
218 ds_orig = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
Dtest_random_sharpness.py273 data = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
276 data = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
Dtest_autocontrast.py211 data_set = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
213 ds_orig = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
Dtest_get_col_names.py113 data = ds.MnistDataset(MNIST_DIR)
Dtest_random_color.py236 mnist_ds = ds.MnistDataset(dataset_dir=MNIST_DATA_DIR, num_samples=2, shuffle=False)
Dtest_datasets_sharding.py232 …data1 = ds.MnistDataset(mnist_dir, num_shards=num_shards, shard_id=shard_id, num_samples=num_sampl…
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/
Ddatasets.h520 class MnistDataset : public Dataset {
522 MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage,
524 …MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage, const Sampler *…
526 MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage,
528 ~MnistDataset() = default;
540 inline std::shared_ptr<MnistDataset> Mnist(const std::string &dataset_dir, const std::string &usage…
543 …return std::make_shared<MnistDataset>(StringToChar(dataset_dir), StringToChar(usage), sampler, cac…
553 inline std::shared_ptr<MnistDataset> Mnist(const std::string &dataset_dir, const std::string &usage,
556 …return std::make_shared<MnistDataset>(StringToChar(dataset_dir), StringToChar(usage), sampler, cac…
566 inline std::shared_ptr<MnistDataset> Mnist(const std::string &dataset_dir, const std::string &usage,
[all …]
/third_party/mindspore/tests/st/probability/bnn_layers/
Ddataset.py32 mnist_ds = ds.MnistDataset(data_path)
/third_party/mindspore/tests/st/probability/transforms/
Ddataset.py32 mnist_ds = ds.MnistDataset(data_path)
/third_party/mindspore/tests/st/summary/
Ddataset.py31 … mnist_ds = ds.MnistDataset(os.path.join(mnist_path, mode), num_samples=num_samples, shuffle=False)
/third_party/mindspore/tests/st/quantization/lenet_quant/
Ddataset.py32 mnist_ds = ds.MnistDataset(data_path)
/third_party/mindspore/tests/st/probability/zhusuan/vae/
Dutils.py37 mnist_ds = ds.MnistDataset(data_path)
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Ddatasets.h2205 class MnistDataset : public Dataset {
2213 MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage,
2221 …MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage, const Sampler *…
2229 MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage,
2233 ~MnistDataset() = default;
2244 inline std::shared_ptr<MnistDataset> Mnist(const std::string &dataset_dir, const std::string &usage…
2247 …return std::make_shared<MnistDataset>(StringToChar(dataset_dir), StringToChar(usage), sampler, cac…
2257 inline std::shared_ptr<MnistDataset> Mnist(const std::string &dataset_dir, const std::string &usage,
2260 …return std::make_shared<MnistDataset>(StringToChar(dataset_dir), StringToChar(usage), sampler, cac…
2270 inline std::shared_ptr<MnistDataset> Mnist(const std::string &dataset_dir, const std::string &usage,
[all …]
Dsamplers.h46 friend class MnistDataset; variable
/third_party/mindspore/tests/st/probability/dpn/
Dtest_gpu_svi_vae.py68 mnist_ds = ds.MnistDataset(data_path)
Dtest_gpu_svi_cvae.py78 mnist_ds = ds.MnistDataset(data_path)
Dtest_gpu_vae_gan.py135 mnist_ds = ds.MnistDataset(data_path)
/third_party/mindspore/tests/st/probability/toolbox/
Dtest_uncertainty.py85 mnist_ds = ds.MnistDataset(data_path)
/third_party/mindspore/tests/st/ps/full_ps/
Dtest_full_ps_lenet.py94 mnist_ds = ds.MnistDataset(data_path)
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc1233 MnistDataset::MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage, in MnistDataset() function in mindspore::dataset::MnistDataset
1239 MnistDataset::MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage, co… in MnistDataset() function in mindspore::dataset::MnistDataset
1245 MnistDataset::MnistDataset(const std::vector<char> &dataset_dir, const std::vector<char> &usage, in MnistDataset() function in mindspore::dataset::MnistDataset
/third_party/mindspore/tests/st/ps/part_ps/
Dtest_ps_embedding_heterogeneous_conv2d_adam.py87 mnist_ds = ds.MnistDataset(data_path)

12