Searched refs:prefetch_size_ (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | cache_base_op.cc | 52 prefetch_size_(1), in CacheBase()
56 if (prefetch_size_ < prefetch_sz_per_thread) { in CacheBase()
57 prefetch_size_ = prefetch_sz_per_thread; in CacheBase()
58 MS_LOG(DEBUG) << "Per worker prefetch size : " << prefetch_size_; in CacheBase()
94 prefetch_keys.reserve(prefetch_size_); in FetchSamplesToWorkers()
103 if (row_cnt_ % prefetch_size_ == 0) { in FetchSamplesToWorkers()
275 prefetch_keys.reserve(prefetch_size_); in Prefetcher()
277 cache_miss.reserve(prefetch_size_); in Prefetcher()
|
| D | device_queue_op.h | 79 const int32_t GetPrefetchSize() { return prefetch_size_; } in GetPrefetchSize()
180 const int32_t prefetch_size_; variable
|
| D | cache_base_op.h | 107 int32_t prefetch_size_; variable
|
| D | device_queue_op.cc | 36 prefetch_size_(prefetch_size), in DeviceQueueOp()
675 out << "\nChannel name: " << channel_name_ << "\nPrefetch size: " << prefetch_size_ << "\n\n"; in Print()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/ |
| D | cache_client.h | 116 prefetch_size_ = prefetch_sz; in SetPrefetchSize()
127 int32_t GetPrefetchSize() const { return prefetch_size_; } in GetPrefetchSize()
140 int32_t prefetch_size_; variable
233 int32_t GetPrefetchSize() const { return prefetch_size_; } in GetPrefetchSize()
288 int32_t prefetch_size_; variable
|
| D | cache_client.cc | 27 … cache_mem_sz_(0), spill_(false), hostname_(""), port_(0), num_connections_(0), prefetch_size_(0) { in Builder()
32 prefetch_size_ = cfg->prefetch_size(); // prefetch size in Builder()
39 prefetch_size_); in Build()
47 CHECK_FAIL_RETURN_SYNTAX_ERROR(prefetch_size_ > 0, "prefetch size must be positive."); in SanityCheck()
65 prefetch_size_(prefetch_size), in CacheClient()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/ |
| D | config_manager.cc | 50 prefetch_size_(kDftPrefetchSize), in ConfigManager()
93 set_prefetch_size(j.value("prefetchSize", prefetch_size_)); in FromJson()
165 void ConfigManager::set_prefetch_size(int32_t prefetch_size) { prefetch_size_ = prefetch_size; } in set_prefetch_size()
|
| D | config_manager.h | 109 int32_t prefetch_size() const { return prefetch_size_; } in prefetch_size()
259 int32_t prefetch_size_; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | transfer_node.cc | 39 prefetch_size_(GlobalContext::config_manager()->prefetch_size()), in TransferNode()
53 out << (Name() + "(prefetch_size:" + std::to_string(prefetch_size_) + ",send_epoch_end:" + in Print()
98 …auto op = std::make_shared<DeviceQueueOp>(queue_name_, type, device_id_, prefetch_size_, send_epoc… in Build()
|
| D | transfer_node.h | 77 int32_t PrefetchSize() const { return prefetch_size_; } in PrefetchSize()
99 int32_t prefetch_size_; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | generator_op.h | 84 int32_t prefetch_size_; variable
|
| D | generator_op.cc | 33 prefetch_size_(prefetch_size), in GeneratorOp()
|