Searched refs:dataset_generator (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/dynamic_shape/ |
| D | test_getnext_dynamic_pipeline.py | 46 def dataset_generator(): function
78 dataset = ds.GeneratorDataset(dataset_generator, ["data1", "data2", "data3", "data4", "data5"])
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_datasets_generator.py | 792 dataset_generator = DatasetGenerator()
793 data1 = ds.GeneratorDataset(dataset_generator, ["data"])
843 dataset_generator = DatasetGenerator()
844 data1 = ds.GeneratorDataset(dataset_generator, ["data"], num_shards=3, shard_id=0)
859 dataset_generator = DatasetGenerator()
860 data1 = ds.GeneratorDataset(dataset_generator, ["data"], num_shards=3, shard_id=0)
|