Searched refs:PreAssignCNodeMemory (Results 1 – 2 of 2) sorted by relevance
169 DeviceAddressPtr PreAssignCNodeMemory(const AnfNodePtr &anf_node, size_t index) const;
131 address = PreAssignCNodeMemory(input_node, input_node_with_index.second); in RunOpGetCommunicationInputInfo()808 DeviceAddressPtr KernelRuntime::PreAssignCNodeMemory(const AnfNodePtr &anf_node, size_t index) cons… in PreAssignCNodeMemory() function in mindspore::device::KernelRuntime821 return PreAssignCNodeMemory(input_node_with_index.first, input_node_with_index.second); in PreAssignCNodeMemory()854 address = PreAssignCNodeMemory(input_node, input_node_with_index.second); in AssignCommunicationNodeInputMem()