Home
last modified time | relevance | path

Searched refs:backendID (Results 1 – 21 of 21) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/
H A Dbackend_manager.cpp63 if (backendID == static_cast<size_t>(0)) { in GetBackend()
68 iter = m_backends.find(backendID); in GetBackend()
86 if (backendID == static_cast<size_t>(0)) { in GetBackendName()
89 iter = m_backendNames.find(backendID); in GetBackendName()
114 size_t backendID = regBackend->GetBackendID(); in RegisterBackend() local
120 "backendID=%{public}zu", backendID); in RegisterBackend()
130 m_backends.emplace(backendID, regBackend); in RegisterBackend()
131 m_backendIDs.emplace_back(backendID); in RegisterBackend()
152 for (auto backendID : backendIDs) { in RemoveBackend() local
154 m_backends.erase(backendID); in RemoveBackend()
[all …]
H A Dbackend_manager.h36 std::shared_ptr<Backend> GetBackend(size_t backendID);
37 const std::string& GetBackendName(size_t backendID);
H A Dneural_network_core.cpp315 compilationImpl->backendID = deviceID; in OH_NNCompilation_SetDevice()
410 std::shared_ptr<Backend> backend = manager.GetBackend(compilation->backendID); in CreateCompiler()
412 LOGE("CreateCompiler failed, fail to get backend %{public}zu.", compilation->backendID); in CreateCompiler()
784 std::shared_ptr<Backend> backend = manager.GetBackend(compilationImpl->backendID); in OH_NNCompilation_Destroy()
1122 size_t backendID = tensorImpl->GetBackendID(); in OH_NNTensor_Destroy() local
1124 std::shared_ptr<Backend> backend = backendManager.GetBackend(backendID); in OH_NNTensor_Destroy()
1126 LOGE("OH_NNTensor_Destroy failed, passed invalid backend name %{public}zu.", backendID); in OH_NNTensor_Destroy()
1379 std::shared_ptr<Backend> backend = backendManager.GetBackend(compilationImpl->backendID); in OH_NNExecutor_Construct()
1449 size_t backendID = executorImpl->GetBackendID(); in OH_NNExecutor_Destroy() local
1451 std::shared_ptr<Backend> backend = backendManager.GetBackend(backendID); in OH_NNExecutor_Destroy()
[all …]
H A Dtensor.h27 Tensor(size_t backendID);
H A Dcompilation.h30 size_t backendID {0};
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_backend/
H A Dnn_backend_test.cpp79 size_t backendID = 1; variable
94 size_t backendID = 1; variable
107 size_t backendID = 1; variable
128 size_t backendID = 1; variable
151 size_t backendID = 1; variable
177 size_t backendID = 1; variable
203 size_t backendID = 1; variable
218 size_t backendID = 1; variable
238 size_t backendID = 1; variable
258 size_t backendID = 1; variable
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_compiler/
H A Dnn_compiler_test.cpp197 size_t backendID = 1; variable
217 size_t backendID = 1; variable
234 size_t backendID = 1; variable
254 size_t backendID = 1; variable
273 size_t backendID = 1; variable
298 size_t backendID = 1; variable
323 size_t backendID = 1; variable
352 size_t backendID = 1; variable
373 size_t backendID = 1; variable
391 size_t backendID = 1; variable
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_compiled_cache/
H A Dnn_compiled_cache_test.cpp118 size_t backendID = 1; in Creator() local
155 size_t backendID = 1; variable
182 size_t backendID = 2; in Creator2() local
223 size_t backendID = 1; variable
328 size_t backendID = 1; variable
350 size_t backendID = 1; variable
375 size_t backendID = 1; variable
397 size_t backendID = 1; variable
419 size_t backendID = 2; variable
441 size_t backendID = 3; variable
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnntensor.h26 explicit NNTensor2_0(size_t backendID) : m_backendID(backendID) {} in NNTensor2_0() argument
H A Dnncompiled_cache.cpp138 OH_NN_ReturnCode NNCompiledCache::SetBackend(size_t backendID) in SetBackend() argument
141 std::shared_ptr<Backend> backend = backendManager.GetBackend(backendID); in SetBackend()
143 …[NNCompiledCache] SetBackend failed, backend with backendID %{public}zu is not exist.", backendID); in SetBackend()
150 …"[NNCompiledCache] SetBackend failed, device with backendID %{public}zu is not exist.", backendID); in SetBackend()
154 m_backendID = backendID; in SetBackend()
H A Dnncompiler.h33 NNCompiler(std::shared_ptr<Device> device, size_t backendID);
34 NNCompiler(const void* model, std::shared_ptr<Device> device, size_t backendID);
H A Dnnbackend.cpp30 NNBackend::NNBackend(const std::shared_ptr<Device>& device, size_t backendID) in NNBackend() argument
32 m_backendID(backendID) {} in NNBackend()
H A Dnnbackend.h31 explicit NNBackend(const std::shared_ptr<Device>& device, size_t backendID);
H A Dnncompiled_cache.h52 OH_NN_ReturnCode SetBackend(size_t backendID);
H A Dnncompiler.cpp122 NNCompiler::NNCompiler(std::shared_ptr<Device> device, size_t backendID) in NNCompiler() argument
124 m_backendID(backendID) {} in NNCompiler()
126 NNCompiler::NNCompiler(const void* model, std::shared_ptr<Device> device, size_t backendID) in NNCompiler() argument
128 m_backendID(backendID) in NNCompiler()
H A Dnnexecutor.h28 NNExecutor(size_t backendID,
H A Dnnexecutor.cpp29 NNExecutor::NNExecutor(size_t backendID, std::shared_ptr<Device> device, std::shared_ptr<PreparedMo… in NNExecutor() argument
32 : m_backendID(backendID), in NNExecutor()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_core_test/
H A Dneural_network_core_test.h48 size_t backendID = 1; variable
H A Dneural_network_core_test.cpp156 size_t backendID = 4; in Creator4() local
183 std::shared_ptr<Backend> backend = std::make_unique<NNBackend>(device, backendID); in Creator4()
1530 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1557 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1584 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1613 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1629 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1657 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1673 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1702 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_tensor/
H A Dnn_tensor_test.cpp243 size_t backendID = 1; in Creator() local
266 std::shared_ptr<Backend> backend = std::make_unique<NNBackend>(device, backendID); in Creator()
367 size_t backendID = 2; in Creator2() local
369 std::shared_ptr<Backend> backend = std::make_unique<NNBackend>(nullptr, backendID); in Creator2()
410 size_t backendID = 3; in Creator3() local
433 std::shared_ptr<Backend> backend = std::make_unique<NNBackend>(device, backendID); in Creator3()
475 size_t backendID = 4; in Creator4() local
502 std::shared_ptr<Backend> backend = std::make_unique<NNBackend>(device, backendID); in Creator4()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_executor/
H A Dnn_executor_test.cpp927 size_t backendID = 1; variable
932 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
988 size_t backendID = 1; variable
993 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1048 size_t backendID = 1; variable
1053 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);
1155 size_t backendID = 1; variable
1160 std::unique_ptr<NNBackend> hdiDevice = std::make_unique<NNBackend>(device, backendID);