Searched refs:max_size_in_gb (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/util/ |
| D | circular_pool.cc | 202 CircularPool::CircularPool(int max_size_in_gb, int arena_size, bool is_cuda_malloc) in CircularPool() argument
203 : unlimited_(max_size_in_gb <= 0), in CircularPool()
204 max_size_in_mb_(unlimited_ ? std::numeric_limits<int32_t>::max() : max_size_in_gb * 1024), in CircularPool()
209 CircularPool::CircularPool(int max_size_in_gb, int arena_size) in CircularPool() argument
210 : unlimited_(max_size_in_gb <= 0), in CircularPool()
211 max_size_in_mb_(unlimited_ ? std::numeric_limits<int32_t>::max() : max_size_in_gb * 1024), in CircularPool()
217 Status CircularPool::CreateCircularPool(std::shared_ptr<MemoryPool> *out_pool, int max_size_in_gb, … in CreateCircularPool() argument
223 auto pool = new (std::nothrow) CircularPool(max_size_in_gb, arena_size, is_cuda_malloc); in CreateCircularPool()
238 Status CircularPool::CreateCircularPool(std::shared_ptr<MemoryPool> *out_pool, int max_size_in_gb, … in CreateCircularPool() argument
244 auto pool = new (std::nothrow) CircularPool(max_size_in_gb, arena_size); in CreateCircularPool()
|
| D | circular_pool.h | 89 static Status CreateCircularPool(std::shared_ptr<MemoryPool> *out_pool, int max_size_in_gb = -1,
92 static Status CreateCircularPool(std::shared_ptr<MemoryPool> *out_pool, int max_size_in_gb = -1,
108 CircularPool(int max_size_in_gb, int arena_size, bool is_cuda_malloc);
112 CircularPool(int max_size_in_gb, int arena_size);
|