Home
last modified time | relevance | path

Searched refs:deviceBuffer (Results 1 – 3 of 3) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dhdi_device_v2_0.cpp520 for (const Buffer& deviceBuffer : deviceBuffers) { in AllocateDeviceBufferForOfflineModel() local
521 status = ReleaseBuffer(deviceBuffer.data); in AllocateDeviceBufferForOfflineModel()
552 void* deviceBuffer {nullptr}; in CopyOfflineModelToDevice() local
559 deviceBuffer = deviceBuffers[i].data; in CopyOfflineModelToDevice()
563 … errno_t errorCode = memcpy_s(deviceBuffer, deviceBufferSize, offlineModel, offlineModelSize); in CopyOfflineModelToDevice()
603 for (const Buffer& deviceBuffer : deviceBuffers) { in PrepareOfflineModel() local
604 status = ReleaseBuffer(deviceBuffer.data); in PrepareOfflineModel()
654 for (const Buffer& deviceBuffer : deviceBuffers) { in PrepareOfflineModel() local
655 ret = ReleaseBuffer(deviceBuffer.data); in PrepareOfflineModel()
H A Dhdi_device_v2_1.cpp515 for (const Buffer& deviceBuffer : deviceBuffers) { in AllocateDeviceBufferForOfflineModel() local
516 status = ReleaseBuffer(deviceBuffer.data); in AllocateDeviceBufferForOfflineModel()
547 void* deviceBuffer {nullptr}; in CopyOfflineModelToDevice() local
554 deviceBuffer = deviceBuffers[i].data; in CopyOfflineModelToDevice()
558 … errno_t errorCode = memcpy_s(deviceBuffer, deviceBufferSize, offlineModel, offlineModelSize); in CopyOfflineModelToDevice()
598 for (const Buffer& deviceBuffer : deviceBuffers) { in PrepareOfflineModel() local
599 status = ReleaseBuffer(deviceBuffer.data); in PrepareOfflineModel()
649 for (const Buffer& deviceBuffer : deviceBuffers) { in PrepareOfflineModel() local
650 ret = ReleaseBuffer(deviceBuffer.data); in PrepareOfflineModel()
H A Dnnexecutor.cpp1062 void* deviceBuffer = outputTensors[i]->GetBuffer(); in Run() local
1063 if (deviceBuffer == nullptr) { in Run()
1068 …= memcpy_s(m_outputTensors[i].userBuffer, m_outputTensors[i].userBufferLength, deviceBuffer, size); in Run()