Home
last modified time | relevance | path

Searched refs:DatasetCache (Results 1 – 25 of 65) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Ddatasets.h51 class DatasetCache; variable
315 const std::shared_ptr<DatasetCache> &cache = nullptr,
347 const std::shared_ptr<DatasetCache> &cache = nullptr,
380 const std::shared_ptr<DatasetCache> &cache = nullptr,
709 … const std::vector<std::vector<char>> &project_columns, const std::shared_ptr<DatasetCache> &cache,
844 const std::shared_ptr<DatasetCache> &cache);
855 const std::shared_ptr<DatasetCache> &cache);
866 … const std::reference_wrapper<Sampler> sampler, const std::shared_ptr<DatasetCache> &cache);
886 const std::shared_ptr<DatasetCache> &cache = nullptr) {
902 const std::shared_ptr<DatasetCache> &cache = nullptr) {
[all …]
/third_party/mindspore/tests/ut/python/dataset/
Dtest_cache_map.py61 some_cache = ds.DatasetCache(session_id=session_id, size=0)
95 some_cache = ds.DatasetCache(session_id=session_id, size=0)
119 some_cache = ds.DatasetCache(session_id=session_id, size=0)
159 some_cache = ds.DatasetCache(session_id=session_id, size=0)
193 some_cache = ds.DatasetCache(session_id=session_id, size=0)
229 some_cache = ds.DatasetCache(session_id=session_id, size=0)
274 some_cache = ds.DatasetCache(session_id=session_id, size=0)
315 some_cache = ds.DatasetCache(session_id=session_id, size=0)
355 some_cache = ds.DatasetCache(session_id=session_id, size=0)
395 some_cache = ds.DatasetCache(session_id=session_id, size=0)
[all …]
Dtest_cache_nomap.py70 some_cache = ds.DatasetCache(session_id=session_id, size=0)
104 some_cache = ds.DatasetCache(session_id=session_id, size=0)
142 some_cache = ds.DatasetCache(session_id=session_id, size=0)
191 some_cache = ds.DatasetCache(session_id=session_id, size=0)
239 some_cache = ds.DatasetCache(session_id=session_id, size=0)
278 some_cache = ds.DatasetCache(session_id=session_id, size=0)
321 some_cache = ds.DatasetCache(session_id=session_id, size=0)
352 some_cache = ds.DatasetCache(session_id=session_id, size=0)
379 some_cache = ds.DatasetCache(session_id=session_id, size=0)
412 some_cache = ds.DatasetCache(session_id=session_id, size=0, prefetch_size=32)
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/
Ddatasets.h48 class DatasetCache; variable
198 const std::shared_ptr<DatasetCache> &cache = nullptr,
229 const std::shared_ptr<DatasetCache> &cache = nullptr,
261 const std::shared_ptr<DatasetCache> &cache = nullptr,
423 … const std::vector<std::vector<char>> &project_columns, const std::shared_ptr<DatasetCache> &cache,
458 const std::shared_ptr<DatasetCache> &cache);
461 const std::shared_ptr<DatasetCache> &cache);
464 … const std::reference_wrapper<Sampler> sampler, const std::shared_ptr<DatasetCache> &cache);
483 const std::shared_ptr<DatasetCache> &cache = nullptr) {
499 const std::shared_ptr<DatasetCache> &cache = nullptr) {
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc466 … const std::shared_ptr<DatasetCache> &cache, std::vector<std::shared_ptr<DSCallback>> callbacks) { in MapDataset()
840 std::shared_ptr<DatasetCache> CreateDatasetCacheCharIF(session_id_type id, uint64_t mem_sz, bool sp… in CreateDatasetCacheCharIF()
852 … const std::shared_ptr<Sampler> &sampler, const std::shared_ptr<DatasetCache> &cache) { in AlbumDataset()
861 const std::shared_ptr<DatasetCache> &cache) { in AlbumDataset()
869 … const std::reference_wrapper<Sampler> sampler, const std::shared_ptr<DatasetCache> &cache) { in AlbumDataset()
880 const std::shared_ptr<DatasetCache> &cache) { in CelebADataset()
888 const std::shared_ptr<DatasetCache> &cache) { in CelebADataset()
897 const std::shared_ptr<DatasetCache> &cache) { in CelebADataset()
905 … const std::shared_ptr<Sampler> &sampler, const std::shared_ptr<DatasetCache> &cache) { in Cifar10Dataset()
911 const Sampler *sampler, const std::shared_ptr<DatasetCache> &cache) { in Cifar10Dataset()
[all …]
/third_party/mindspore/tests/ut/cpp/dataset/
Dc_api_cache_test.cc35 …std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false, "127.0.0.1", … in TEST_F()
55 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
83 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
124 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
167 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
208 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
250 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
291 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
332 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
373 std::shared_ptr<DatasetCache> some_cache = CreateDatasetCache(env_session, 0, false); in TEST_F()
[all …]
Ddeserialize_test.cc94 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
152 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
220 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
259 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
273 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
306 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
336 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
357 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
373 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
404 std::shared_ptr<DatasetCache> cache = nullptr; in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Ddataset_node.h48 class DatasetCache; variable
143 explicit DatasetNode(const std::shared_ptr<DatasetCache> &dataset_cache);
262 std::shared_ptr<DatasetCache> GetDatasetCache() { return cache_; } in GetDatasetCache()
272 std::shared_ptr<DatasetNode> SetDatasetCache(const std::shared_ptr<DatasetCache> &cache);
328 std::shared_ptr<DatasetCache> cache_;
355 …explicit MappableSourceNode(const std::shared_ptr<DatasetCache> &dataset_cache) : DatasetNode(data… in MappableSourceNode()
386 …explicit NonMappableSourceNode(const std::shared_ptr<DatasetCache> &dataset_cache) : DatasetNode(d… in NonMappableSourceNode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/cache/
Ddataset_cache.h27 class DatasetCache {
40 static Status from_json(nlohmann::json json_obj, std::shared_ptr<DatasetCache> *cache);
Ddataset_cache.cc30 Status DatasetCache::from_json(nlohmann::json json_obj, std::shared_ptr<DatasetCache> *cache) { in from_json()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dcifar10_node.cc35 std::shared_ptr<DatasetCache> cache) in Cifar10Node()
136 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
137 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dcifar100_node.cc35 std::shared_ptr<SamplerObj> sampler, std::shared_ptr<DatasetCache> cache) in Cifar100Node()
135 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
136 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dmnist_node.cc34 std::shared_ptr<DatasetCache> cache) in MnistNode()
129 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
130 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dalbum_node.cc37 … const std::shared_ptr<SamplerObj> &sampler, const std::shared_ptr<DatasetCache> &cache) in AlbumNode()
164 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
165 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dimage_folder_node.cc38 std::shared_ptr<DatasetCache> cache = nullptr) in ImageFolderNode()
155 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
156 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dflickr_node.cc35 std::shared_ptr<SamplerObj> sampler, std::shared_ptr<DatasetCache> cache) in FlickrNode()
165 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
166 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Drandom_node.h41 std::shared_ptr<DatasetCache> cache) in RandomNode()
50 std::shared_ptr<DatasetCache> cache) in RandomNode()
Dtext_file_node.cc27 … int32_t num_shards, int32_t shard_id, std::shared_ptr<DatasetCache> cache) in TextFileNode()
169 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
170 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dminddata_node.h36 … ShuffleMode shuffle_mode = ShuffleMode::kGlobal, std::shared_ptr<DatasetCache> cache = nullptr);
41 … ShuffleMode shuffle_mode = ShuffleMode::kGlobal, std::shared_ptr<DatasetCache> cache = nullptr);
Dceleba_node.cc38 … const std::set<std::string> &extensions, const std::shared_ptr<DatasetCache> &cache) in CelebANode()
204 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
205 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dmanifest_node.cc37 std::shared_ptr<DatasetCache> cache) in ManifestNode()
185 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
186 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dtf_record_node.h40 … int32_t num_shards, int32_t shard_id, bool shard_equal_rows, std::shared_ptr<DatasetCache> cache) in TFRecordNode()
61 … int32_t num_shards, int32_t shard_id, bool shard_equal_rows, std::shared_ptr<DatasetCache> cache) in TFRecordNode()
Dcsv_node.cc29 int32_t num_shards, int32_t shard_id, std::shared_ptr<DatasetCache> cache) in CSVNode()
208 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
209 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dcoco_node.cc35 …onst bool &decode, const std::shared_ptr<SamplerObj> &sampler, std::shared_ptr<DatasetCache> cache, in CocoNode()
218 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
219 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()
Dvoc_node.cc37 std::shared_ptr<DatasetCache> cache, bool extra_metadata) in VOCNode()
210 std::shared_ptr<DatasetCache> cache = nullptr; in from_json()
211 RETURN_IF_NOT_OK(DatasetCache::from_json(json_obj, &cache)); in from_json()

123