Searched refs:MindDataSet (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/ut/python/train/ |
| D | test_amp.py | 96 class MindDataSet(MindData): class
98 super(MindDataSet, self).__init__(size=2, batch_size=32,
117 dataset = MindDataSet(dataset_types, dataset_shapes)
134 dataset = MindDataSet(dataset_types, dataset_shapes)
155 dataset = MindDataSet(dataset_types, dataset_shapes)
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_auto_parallel_flag.py | 31 class MindDataSet(MindData): class
33 super(MindDataSet, self).__init__(size=2, batch_size=32,
102 dataset = MindDataSet(dataset_types, dataset_shapes)
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_optimizer_with_loss_scale.py | 34 class MindDataSet(MindData): class
36 super(MindDataSet, self).__init__(size=2, batch_size=32,
186 dataset = MindDataSet(dataset_types, dataset_shapes)
200 dataset = MindDataSet(dataset_types, dataset_shapes)
|
| /third_party/mindspore/tests/st/pynative/loss_scale/ |
| D | test_loss_scale.py | 100 class MindDataSet(MindData): class
102 super(MindDataSet, self).__init__(size=2, batch_size=32,
203 dataset = MindDataSet(dataset_types, dataset_shapes)
|
| /third_party/mindspore/tests/ut/python/ir/ |
| D | test_row_tensor.py | 77 class MindDataSet(MindData): class
79 super(MindDataSet, self).__init__(size=2, batch_size=32,
356 dataset = MindDataSet(dataset_types, dataset_shapes)
|