Home
last modified time | relevance | path

Searched refs:m_device (Results 1 – 10 of 10) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnnbackend.cpp31 : m_device(device), in NNBackend()
36 m_device = nullptr; in ~NNBackend()
46 if (m_device == nullptr) { in GetBackendName()
59 ret = m_device->GetVendorName(vendorName); in GetBackendName()
66 ret = m_device->GetVersion(version); in GetBackendName()
78 if (m_device == nullptr) { in GetBackendType()
94 if (m_device == nullptr) { in GetBackendStatus()
99 OH_NN_ReturnCode ret = m_device->GetDeviceStatus(status); in GetBackendStatus()
219 if (m_device == nullptr) { in GetDevice()
222 return m_device; in GetDevice()
[all …]
H A Dnncompiler.cpp123 : m_device(device), in NNCompiler()
127 : m_device(device), in NNCompiler()
154 if (m_device == nullptr) { in SetCacheDir()
179 if (m_device == nullptr) { in SetPerformance()
207 if (m_device == nullptr) { in SetPriority()
235 if (m_device == nullptr) { in SetEnableFp16()
383 ret = m_device->PrepareModel(m_liteGraph, config, m_preparedModel); in NormalBuild()
386 ret = m_device->PrepareModel(m_metaGraph, config, m_preparedModel); in NormalBuild()
413 if (m_device == nullptr) { in Build()
504 m_device->ReleaseBuffer(buffers[i].data); in ReleaseBufferByDevice()
[all …]
H A Dnnexecutor.cpp33 m_device(device), in NNExecutor()
508 m_device->ReleaseBuffer(curBuffer); in SetInputTensorWithNewBuffer()
634 m_device->ReleaseBuffer(inputBuffer); in SetInput()
764 m_device->ReleaseBuffer(curBuffer); in SetOutput()
819 m_device->ReleaseBuffer(curBuffer); in SetOutputFromMemory()
862 m_device->ReleaseBuffer(deviceInputBuffer); in CreateInputMemory()
892 auto ret = m_device->ReleaseBuffer((*memory)->data); in DestroyInputMemory()
927 m_device->ReleaseBuffer(deviceOutputBuffer); in CreateOutputMemory()
957 auto ret = m_device->ReleaseBuffer((*memory)->data); in DestroyOutputMemory()
1084 m_device->ReleaseBuffer((it.second).tensor->GetBuffer()); in ~NNExecutor()
[all …]
H A Dnncompiled_cache.cpp48 if (m_device == nullptr) { in Save()
77 if (m_device == nullptr) { in Restore()
148 m_device = nnBackend->GetDevice(); in SetBackend()
149 if (m_device == nullptr) { in SetBackend()
370 char* ptr = static_cast<char*>(m_device->AllocateBuffer(fsize)); in ReadCacheModelFile()
381 m_device->ReleaseBuffer(ptr); in ReadCacheModelFile()
H A Dnnbackend.h58 std::shared_ptr<Device> m_device;
H A Dnncompiled_cache.h75 std::shared_ptr<Device> m_device {nullptr};
H A Dnncompiler.h80 std::shared_ptr<Device> m_device {nullptr};
H A Dnnexecutor.h97 std::shared_ptr<Device> m_device {nullptr};
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_executor/
H A Dnn_executor_test.cpp192 std::shared_ptr<Device> m_device {nullptr}; variable
223 std::shared_ptr<Device> m_device {nullptr}; variable
252 std::shared_ptr<Device> m_device {nullptr}; variable
281 std::shared_ptr<Device> m_device {nullptr}; variable
311 std::shared_ptr<Device> m_device {nullptr}; variable
342 std::shared_ptr<Device> m_device {nullptr}; variable
382 std::shared_ptr<Device> m_device {nullptr}; variable
422 std::shared_ptr<Device> m_device {nullptr}; variable
462 std::shared_ptr<Device> m_device {nullptr}; variable
487 std::shared_ptr<Device> m_device {nullptr}; variable
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp1289 std::shared_ptr<Device> m_device {nullptr}; variable
1321 std::shared_ptr<Device> m_device {nullptr}; variable
1351 std::shared_ptr<Device> m_device {nullptr}; variable
1381 std::shared_ptr<Device> m_device {nullptr}; variable
1489 std::shared_ptr<Device> m_device {nullptr}; variable
1516 std::shared_ptr<Device> m_device {nullptr}; variable
1542 std::shared_ptr<Device> m_device {nullptr}; variable
1703 std::shared_ptr<Device> m_device {nullptr}; variable
1808 std::shared_ptr<Device> m_device {nullptr}; variable
1882 std::shared_ptr<Device> m_device {nullptr}; variable
[all …]