Searched refs:device_ptr_list (Results 1 – 2 of 2) sorted by relevance
179 auto device_ptr_list = MallocContinuousMemFromMemPool(total_size, size_list); in MallocContinuousMemFromMemPool() local180 if (device_ptr_list.empty()) { in MallocContinuousMemFromMemPool()183 if (addr_list.size() != device_ptr_list.size()) { in MallocContinuousMemFromMemPool()185 << device_ptr_list.size(); in MallocContinuousMemFromMemPool()188 MS_EXCEPTION_IF_NULL(device_ptr_list[i]); in MallocContinuousMemFromMemPool()190 addr_list[i]->ptr_ = device_ptr_list[i]; in MallocContinuousMemFromMemPool()201 std::vector<void *> device_ptr_list; in MallocContinuousMemFromMemPool() local203 device_ptr_list.emplace_back(nullptr); in MallocContinuousMemFromMemPool()205 return device_ptr_list; in MallocContinuousMemFromMemPool()
41 auto device_ptr_list = MallocContinuousMemFromMemPool(total_size, size_list); in MallocContinuousMemFromMemPool() local42 if (device_ptr_list.size() == 0) { in MallocContinuousMemFromMemPool()45 if (addr_list.size() != device_ptr_list.size()) { in MallocContinuousMemFromMemPool()54 auto new_addr = device_ptr_list[i]; in MallocContinuousMemFromMemPool()