Home
last modified time | relevance | path

Searched refs:deviceBuffers (Results 1 – 4 of 4) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dhdi_device_v2_0.cpp511 deviceBuffers.clear(); in AllocateDeviceBufferForOfflineModel()
528 deviceBuffers.clear(); in AllocateDeviceBufferForOfflineModel()
536 deviceBuffers.emplace_back(modelBuffer); in AllocateDeviceBufferForOfflineModel()
545 if (offlineModels.size() != deviceBuffers.size()) { in CopyOfflineModelToDevice()
559 deviceBuffer = deviceBuffers[i].data; in CopyOfflineModelToDevice()
560 deviceBufferSize = deviceBuffers[i].length; in CopyOfflineModelToDevice()
589 size_t numOfflineModel = deviceBuffers.size(); in PrepareOfflineModel()
603 for (const Buffer& deviceBuffer : deviceBuffers) { in PrepareOfflineModel()
610 deviceBuffers.clear(); in PrepareOfflineModel()
641 std::vector<Buffer> deviceBuffers; in PrepareOfflineModel() local
[all …]
H A Dhdi_device_v2_1.cpp506 deviceBuffers.clear(); in AllocateDeviceBufferForOfflineModel()
523 deviceBuffers.clear(); in AllocateDeviceBufferForOfflineModel()
531 deviceBuffers.emplace_back(modelBuffer); in AllocateDeviceBufferForOfflineModel()
540 if (offlineModels.size() != deviceBuffers.size()) { in CopyOfflineModelToDevice()
554 deviceBuffer = deviceBuffers[i].data; in CopyOfflineModelToDevice()
555 deviceBufferSize = deviceBuffers[i].length; in CopyOfflineModelToDevice()
584 size_t numOfflineModel = deviceBuffers.size(); in PrepareOfflineModel()
598 for (const Buffer& deviceBuffer : deviceBuffers) { in PrepareOfflineModel()
605 deviceBuffers.clear(); in PrepareOfflineModel()
636 std::vector<Buffer> deviceBuffers; in PrepareOfflineModel() local
[all …]
H A Dhdi_device_v2_0.h74 std::vector<Buffer>& deviceBuffers);
76 std::vector<Buffer>& deviceBuffers);
77 OH_NN_ReturnCode PrepareOfflineModel(std::vector<Buffer>& deviceBuffers,
H A Dhdi_device_v2_1.h74 std::vector<Buffer>& deviceBuffers);
76 std::vector<Buffer>& deviceBuffers);
77 OH_NN_ReturnCode PrepareOfflineModel(std::vector<Buffer>& deviceBuffers,