Searched refs:MallocOutputMem (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | memory_manager.h | 49 uint8_t *MallocOutputMem(const AnfNodePtr &node, size_t index, MemType type, size_t size,
|
| D | memory_manager.cc | 77 uint8_t *MemoryManager::MallocOutputMem(const AnfNodePtr &node, size_t index, MemType type, size_t … in MallocOutputMem() function in mindspore::device::MemoryManager
|
| D | kernel_runtime.cc | 794 output_ptr = mem_manager_->MallocOutputMem(node, 0, type, total_size, address, true); in AssignCommunicationNodeOutputMem()
882 …uint8_t *input_ptr = mem_manager_->MallocOutputMem(prenode_index.first, prenode_index.second, type… in AssignCommunicationNodeInputMem()
926 …uint8_t *ptr = mem_manager_->MallocOutputMem(node, i, type, output_sizes[i], device_address, false… in AssignNodeOutputMem()
942 …uint8_t *ptr = mem_manager_->MallocOutputMem(node, index, kStaticMem, tensor_address->size(), devi… in AssignExtraStaticMem()
|