Home
last modified time | relevance | path

Searched refs:DynamicMemBuf (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_dynamic_allocator.h47 struct DynamicMemBuf { struct
48 DynamicMemBuf(DeviceMemPtr addr, DynamicMemBufStatus status, size_t size) in DynamicMemBuf() argument
54 using DynamicMemBufPtr = std::shared_ptr<DynamicMemBuf>; argument
Dmem_dynamic_allocator.cc67 continuous_mem_buf = std::make_shared<DynamicMemBuf>(buf_addr, kMemBufUsed, size_list[i]); in AllocContinuousTensorMem()
130 auto mem_buf = std::make_shared<DynamicMemBuf>(device_addr, kMemBufUsed, real_alloc_size); in AddMemBlockAndMemBuf()
179 auto new_mem_buf = std::make_shared<DynamicMemBuf>(newbuf_addr, kMemBufIdle, newbuf_size); in DivideMemBuf()