Home
last modified time | relevance | path

Searched refs:prefetch_size (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore/dataset/engine/
Dcache_client.py54 prefetch_size=None): argument
68 if prefetch_size is not None:
69 check_pos_int32(prefetch_size, "prefetch_size")
76 self.prefetch_size = prefetch_size
78 …he_client = CacheClient(session_id, size, spilling, hostname, port, num_connections, prefetch_size)
95 new_cache.prefetch_size = copy.deepcopy(self.prefetch_size, memodict)
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dconfig.cc45 bool set_prefetch_size(int32_t prefetch_size) { in set_prefetch_size() argument
46 if (prefetch_size <= 0) { in set_prefetch_size()
47 MS_LOG(ERROR) << "Prefetch size given is not within the required range: " << prefetch_size; in set_prefetch_size()
50 _config->set_op_connector_size(prefetch_size); in set_prefetch_size()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dconfig_manager.h109 int32_t prefetch_size() const { return prefetch_size_; } in prefetch_size() function
146 void set_prefetch_size(int32_t prefetch_size);
Dconfig_manager.cc165 void ConfigManager::set_prefetch_size(int32_t prefetch_size) { prefetch_size_ = prefetch_size; } in set_prefetch_size() argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Dconfig.h41 bool set_prefetch_size(int32_t prefetch_size);
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dgenerator_op.cc27 … std::vector<DataType> column_types, int32_t prefetch_size, int32_t connector_size, in GeneratorOp() argument
33 prefetch_size_(prefetch_size), in GeneratorOp()
Dgenerator_op.h44 std::vector<DataType> column_types, int32_t prefetch_size, int32_t connector_size,
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/
Dcache_client.cc32 prefetch_size_ = cfg->prefetch_size(); // prefetch size in Builder()
58 int32_t port, int32_t num_connections, int32_t prefetch_size) in CacheClient() argument
65 prefetch_size_(prefetch_size), in CacheClient()
Dcache_client.h148 int32_t num_connections, int32_t prefetch_size);
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Ddevice_queue_op.h66 …QueueOp(std::string channel_name, DeviceType device_type, int32_t device_id, int32_t prefetch_size,
Ddevice_queue_op.cc30 …QueueOp(std::string channel_name, DeviceType device_type, int32_t device_id, int32_t prefetch_size, in DeviceQueueOp() argument
36 prefetch_size_(prefetch_size), in DeviceQueueOp()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dtransfer_node.cc39 prefetch_size_(GlobalContext::config_manager()->prefetch_size()), in TransferNode()
/third_party/mindspore/tests/ut/python/dataset/
Dtest_cache_nomap.py412 some_cache = ds.DatasetCache(session_id=session_id, size=0, prefetch_size=32)
1028 some_cache = ds.DatasetCache(session_id=session_id, size=0, prefetch_size=1)
1065 some_cache = ds.DatasetCache(session_id=session_id, size=0, prefetch_size=100)
Dtest_cache_map.py1249 some_cache = ds.DatasetCache(session_id=session_id, size=0, prefetch_size=1)
1286 some_cache = ds.DatasetCache(session_id=session_id, size=0, prefetch_size=100)
/third_party/mesa3d/src/panfrost/bifrost/
Dbifrost_compile.c3643 unsigned prefetch_size = BIFROST_SHADER_PREFETCH - final_clause; in bi_pack_clauses() local
3646 memset(util_dynarray_grow(binary, uint8_t, prefetch_size), in bi_pack_clauses()
3647 0, prefetch_size); in bi_pack_clauses()
/third_party/mindspore/
DRELEASE.md352 ###### `mindspore.dataset.Dataset.device_que` interface removes unused parameter `prefetch_size`([!…
354 Previously, we have a parameter `prefetch_size` in `device_que` to define the prefetch number of re…
364 device_que(prefetch_size=None, send_epoch_end=True, create_data_info_queue=False)