Searched refs:dst_device_address (Results 1 – 1 of 1) sorted by relevance
2038 auto dst_device_address = AssignExtraMemForGraphOutput(tensor, node, output_index); in UpdateOutputTensors() local2039 MS_EXCEPTION_IF_NULL(dst_device_address); in UpdateOutputTensors()2040 … if (!dst_device_address->SyncDeviceToDevice(trans::GetRuntimePaddingShape(node, output_index), in UpdateOutputTensors()2046 tensor_device_addr_map_[tensor] = dst_device_address; in UpdateOutputTensors()