Home
last modified time | relevance | path

Searched refs:get_batch_size (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/tests/ut/python/dataset/
Dtest_get_size.py27 assert data.get_batch_size() == 1
31 assert data.get_batch_size() == 1
35 assert data.get_batch_size() == 2
39 assert data.get_batch_size() == 2
43 assert data.get_batch_size() == 2
47 assert data.get_batch_size() == 2
Dtest_datasets_flickr.py101 assert data3.get_batch_size() == 1
104 assert data3.get_batch_size() == 2
115 assert data4.get_batch_size() == 1
118 assert data4.get_batch_size() == 2
Dtest_datasets_cifarop.py123 assert data4.get_batch_size() == 1
126 assert data4.get_batch_size() == 7
135 assert data5.get_batch_size() == 1
138 assert data5.get_batch_size() == 7
293 assert data3.get_batch_size() == 1
296 assert data3.get_batch_size() == 3
306 assert data4.get_batch_size() == 3
Dtest_datasets_sbd.py107 assert data3.get_batch_size() == 1
110 assert data3.get_batch_size() == 3
122 assert data4.get_batch_size() == 1
125 assert data4.get_batch_size() == 3
Dtest_datasets_mnist.py108 assert data4.get_batch_size() == 1
111 assert data4.get_batch_size() == 7
120 assert data5.get_batch_size() == 1
123 assert data5.get_batch_size() == 7
Dtest_datasets_div2k.py92 assert data3.get_batch_size() == 1
98 assert data3.get_batch_size() == 3
107 assert data4.get_batch_size() == 1
112 assert data4.get_batch_size() == 3
Dtest_datasets_usps.py141 assert train_data.get_batch_size() == 1
143 assert train_data.get_batch_size() == 2
154 assert train_data.get_batch_size() == 1
157 assert train_data.get_batch_size() == 2
Dtest_datasets_cityscapes.py115 assert data3.get_batch_size() == 1
118 assert data3.get_batch_size() == 3
130 assert data4.get_batch_size() == 1
133 assert data4.get_batch_size() == 3
Dtest_datasets_flowers102.py167 assert all_data.get_batch_size() == 1
169 assert all_data.get_batch_size() == 3
Dtest_datasets_sbu.py147 assert dataset.get_batch_size() == 1
Dtest_cache_map.py239 ds1.get_batch_size()
Dtest_cache_nomap.py2004 ds1.get_batch_size()
/third_party/mindspore/tests/
Ddataset_mock.py40 def get_batch_size(self): member in MindData
/third_party/mindspore/tests/st/ops/ascend/
Dtest_tdt_data_ms.py74 batch_size = data_set.get_batch_size()
/third_party/mindspore/tests/ut/python/parallel/
Dtest_control_flow.py51 def get_batch_size(self): member in DatasetLenet
Dtest_pipeline_shared_parameter.py52 def get_batch_size(self): member in DatasetLenet
Dtest_pipeline_split.py52 def get_batch_size(self): member in DatasetLenet
/third_party/mindspore/tests/st/data_transfer/
Dtest_tdt_data_transfer.py78 batch_size = dataset.get_batch_size()
/third_party/mindspore/mindspore/train/train_thor/
Dmodel_thor.py59 batch_size = exec_dataset.get_batch_size()
/third_party/mindspore/mindspore/train/
D_utils.py61 batch_size = exec_dataset.get_batch_size()
/third_party/mindspore/tests/st/pynative/loss_scale/
Dtest_loss_scale.py54 def get_batch_size(self): member in MindData
/third_party/mindspore/tests/st/networks/models/resnet50/src_thor/
Dmodel_thor.py65 batch_size = exec_dataset.get_batch_size()
/third_party/mindspore/mindspore/train/callback/
D_summary_collector.py804 batch_size = dataset.get_batch_size()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py1789 num_batch *= self.get_batch_size()
1792 def get_batch_size(self): member in Dataset