Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/
H A Ddevice_manager_test.cpp113 liteGraph.get()->all_tensors_.emplace_back(tp);
114 liteGraph.get()->sub_graphs_.emplace_back(subGraph);
135 liteGraph.get()->all_nodes_.emplace_back(nullptr);
151 liteGraph.get()->all_nodes_.emplace_back(node);
176 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
197 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
216 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
238 liteGraph.get()->all_nodes_.emplace_back(getNode(primitive));
2228 liteGraph.get()->all_tensors_.emplace_back(tp);
2229 liteGraph.get()->all_tensors_.emplace_back(nullptr);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dinner_model_test.cpp121 SetLiteGraph(liteGraph);
140 SetLiteGraph(liteGraph);
158 SetLiteGraph(liteGraph);
176 SetLiteGraph(liteGraph);
194 SetLiteGraph(liteGraph);
214 SetLiteGraph(liteGraph);
233 SetLiteGraph(liteGraph);
251 SetLiteGraph(liteGraph);
306 SetLiteGraph(liteGraph);
360 SetLiteGraph(liteGraph);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dinner_model_test.cpp121 SetLiteGraph(liteGraph);
140 SetLiteGraph(liteGraph);
158 SetLiteGraph(liteGraph);
176 SetLiteGraph(liteGraph);
194 SetLiteGraph(liteGraph);
249 SetLiteGraph(liteGraph);
301 SetLiteGraph(liteGraph);
397 SetLiteGraph(liteGraph);
479 SetLiteGraph(liteGraph);
675 SetLiteGraph(liteGraph);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/inner_kits/
H A Dneural_network_runtime_inner_test.cpp51 EXPECT_NE(nullptr, liteGraph);
53 delete liteGraph;
54 liteGraph = nullptr;
68 const void* liteGraph = nullptr; variable
90 EXPECT_NE(nullptr, liteGraph);
91 liteGraph->name_ = "testGraph";
125 EXPECT_NE(nullptr, liteGraph);
126 liteGraph->name_ = "testGraph";
129 delete liteGraph;
131 liteGraph = nullptr;
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dinner_model.cpp43 void operator()(MSLITE::LiteGraph* liteGraph) const in operator ()()
45 MindIR_LiteGraph_Destroy(&liteGraph); in operator ()()
75 OH_NN_ReturnCode ConstructNNTensorsFromLiteGraph(const MSLITE::LiteGraph* liteGraph, in ConstructNNTensorsFromLiteGraph() argument
85 if (maximumIndex >= liteGraph->all_tensors_.size()) { in ConstructNNTensorsFromLiteGraph()
92 nnTensor = ConstructNNTensorFromLiteGraphTensor(liteGraph->all_tensors_[i]); in ConstructNNTensorsFromLiteGraph()
112 OH_NN_ReturnCode InnerModel::BuildFromLiteGraph(const MSLITE::LiteGraph* liteGraph, in BuildFromLiteGraph() argument
116 if (liteGraph == nullptr) { in BuildFromLiteGraph()
132 …OH_NN_ReturnCode ret = ConstructNNTensorsFromLiteGraph(liteGraph, liteGraph->input_indices_, m_inp… in BuildFromLiteGraph()
139 ret = ConstructNNTensorsFromLiteGraph(liteGraph, liteGraph->output_indices_, m_outputTensors); in BuildFromLiteGraph()
145 m_liteGraph.reset(const_cast<MSLITE::LiteGraph*>(liteGraph), LiteGraphDeleter()); in BuildFromLiteGraph()
H A Dneural_network_runtime.cpp319 …seInputDimsFromExtensions(char* data, size_t dataSize, const mindspore::lite::LiteGraph* liteGraph, in ParseInputDimsFromExtensions() argument
326 size_t inputCount = liteGraph->input_indices_.size(); // LiteGraph输入个数 in ParseInputDimsFromExtensions()
327 size_t allTensorSize = liteGraph->all_tensors_.size(); // LiteGraph所有tensor个数 in ParseInputDimsFromExtensions()
332 if (liteGraph->input_indices_[i] >= allTensorSize) { in ParseInputDimsFromExtensions()
334 liteGraph->input_indices_[i]); in ParseInputDimsFromExtensions()
339 mindspore::lite::TensorPtr tensor = liteGraph->all_tensors_[liteGraph->input_indices_[i]]; in ParseInputDimsFromExtensions()
369 const mindspore::lite::LiteGraph* liteGraph, ExtensionConfig& extensionConfig) in ParseDynamicDimsFromExtensions() argument
389 inputDims[0].first, inputDims[0].second, liteGraph, extensionConfig, dynamicCount); in ParseDynamicDimsFromExtensions()
494 NNRT_API OH_NN_ReturnCode OH_NNModel_BuildFromLiteGraph(OH_NNModel *model, const void *liteGraph, in OH_NNModel_BuildFromLiteGraph() argument
502 if (liteGraph == nullptr) { in OH_NNModel_BuildFromLiteGraph()
[all …]
H A Dlite_graph_to_hdi_model_v1_0.h26 OHOS::HDI::Nnrt::V1_0::Model *LiteGraph_To_HDIModel(const mindspore::lite::LiteGraph *liteGraph,
H A Dlite_graph_to_hdi_model_v2_0.h26 OHOS::HDI::Nnrt::V2_0::Model *LiteGraph_To_HDIModel(const mindspore::lite::LiteGraph *liteGraph,
H A Dlite_graph_to_hdi_model_v2_1.h26 OHOS::HDI::Nnrt::V2_1::Model *LiteGraph_To_HDIModel(const mindspore::lite::LiteGraph *liteGraph,
H A Dhdi_device_v2_0.cpp93 OH_NN_ReturnCode IsOfflineModel(std::shared_ptr<const mindspore::lite::LiteGraph> liteGraph, bool& … in IsOfflineModel() argument
96 if (liteGraph == nullptr) { in IsOfflineModel()
101 if (liteGraph->all_nodes_.size() == 0) { in IsOfflineModel()
107 if (liteGraph->all_nodes_.size() > 1) { in IsOfflineModel()
112 const mindspore::lite::LiteGraph::Node* pNode = liteGraph->all_nodes_[0]; in IsOfflineModel()
H A Dhdi_device_v2_1.cpp93 OH_NN_ReturnCode IsOfflineModel(std::shared_ptr<const mindspore::lite::LiteGraph> liteGraph, bool& … in IsOfflineModel() argument
96 if (liteGraph == nullptr) { in IsOfflineModel()
101 if (liteGraph->all_nodes_.size() == 0) { in IsOfflineModel()
107 if (liteGraph->all_nodes_.size() > 1) { in IsOfflineModel()
112 const mindspore::lite::LiteGraph::Node* pNode = liteGraph->all_nodes_[0]; in IsOfflineModel()
H A Dlite_graph_to_hdi_model_v1_0.cpp1000 OHOS::HDI::Nnrt::V1_0::Model *LiteGraph_To_HDIModel(const mindspore::lite::LiteGraph *liteGraph, in LiteGraph_To_HDIModel() argument
1003 if (liteGraph == nullptr) { in LiteGraph_To_HDIModel()
1015 for (auto node : liteGraph->all_nodes_) { in LiteGraph_To_HDIModel()
1045 for (auto tensor : liteGraph->all_tensors_) { in LiteGraph_To_HDIModel()
1065 for (auto graph : liteGraph->sub_graphs_) { in LiteGraph_To_HDIModel()
1079 retModel->name = liteGraph->name_; in LiteGraph_To_HDIModel()
1080 retModel->inputIndex = liteGraph->input_indices_; in LiteGraph_To_HDIModel()
1081 retModel->outputIndex = liteGraph->output_indices_; in LiteGraph_To_HDIModel()
H A Dlite_graph_to_hdi_model_v2_0.cpp1001 OHOS::HDI::Nnrt::V2_0::Model *LiteGraph_To_HDIModel(const mindspore::lite::LiteGraph *liteGraph, in LiteGraph_To_HDIModel() argument
1004 if (liteGraph == nullptr) { in LiteGraph_To_HDIModel()
1016 for (auto node : liteGraph->all_nodes_) { in LiteGraph_To_HDIModel()
1046 for (auto tensor : liteGraph->all_tensors_) { in LiteGraph_To_HDIModel()
1066 for (auto graph : liteGraph->sub_graphs_) { in LiteGraph_To_HDIModel()
1080 retModel->name = liteGraph->name_; in LiteGraph_To_HDIModel()
1081 retModel->inputIndex = liteGraph->input_indices_; in LiteGraph_To_HDIModel()
1082 retModel->outputIndex = liteGraph->output_indices_; in LiteGraph_To_HDIModel()
H A Dinner_model.h35 OH_NN_ReturnCode BuildFromLiteGraph(const mindspore::lite::LiteGraph* liteGraph,
H A Dnncompiler.h72 OH_NN_ReturnCode IsSupportedModel(const std::shared_ptr<mindspore::lite::LiteGraph>& liteGraph,
H A Dlite_graph_to_hdi_model_v2_1.cpp1805 OHOS::HDI::Nnrt::V2_1::Model *LiteGraph_To_HDIModel(const mindspore::lite::LiteGraph *liteGraph, in LiteGraph_To_HDIModel() argument
1808 if (liteGraph == nullptr) { in LiteGraph_To_HDIModel()
1820 for (auto node : liteGraph->all_nodes_) { in LiteGraph_To_HDIModel()
1851 for (auto tensor : liteGraph->all_tensors_) { in LiteGraph_To_HDIModel()
1871 for (auto graph : liteGraph->sub_graphs_) { in LiteGraph_To_HDIModel()
1885 retModel->name = liteGraph->name_; in LiteGraph_To_HDIModel()
1886 retModel->inputIndex = liteGraph->input_indices_; in LiteGraph_To_HDIModel()
1887 retModel->outputIndex = liteGraph->output_indices_; in LiteGraph_To_HDIModel()
H A Dnncompiler.cpp261 …turnCode NNCompiler::IsSupportedModel(const std::shared_ptr<mindspore::lite::LiteGraph>& liteGraph, in IsSupportedModel() argument
265 OH_NN_ReturnCode ret = m_device->GetSupportedOperation(liteGraph, supportedList); in IsSupportedModel()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/executor/
H A Dexecutor_test.cpp35 MSLITE::LiteGraph* liteGraph = new (std::nothrow) MSLITE::LiteGraph(); in BuildLiteGraph() local
36 if (liteGraph == nullptr) { in BuildLiteGraph()
40 liteGraph->name_ = "testGraph"; in BuildLiteGraph()
41 liteGraph->input_indices_.emplace_back(0); in BuildLiteGraph()
42 liteGraph->output_indices_.emplace_back(1); in BuildLiteGraph()
45 for (size_t indexInput = 0; indexInput < liteGraph->input_indices_.size(); ++indexInput) { in BuildLiteGraph()
47 void* liteGraphTensor1 = MSLITE::MindIR_Tensor_Create(liteGraph->name_, in BuildLiteGraph()
49 liteGraph->all_tensors_.emplace_back(liteGraphTensor1); in BuildLiteGraph()
54 void* liteGraphTensor2 = MSLITE::MindIR_Tensor_Create(liteGraph->name_, in BuildLiteGraph()
56 liteGraph->all_tensors_.emplace_back(liteGraphTensor2); in BuildLiteGraph()
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/executor/
H A Dexecutor_test.cpp35 MSLITE::LiteGraph* liteGraph = new (std::nothrow) MSLITE::LiteGraph(); in BuildLiteGraph() local
36 if (liteGraph == nullptr) { in BuildLiteGraph()
40 liteGraph->name_ = "testGraph"; in BuildLiteGraph()
41 liteGraph->input_indices_.emplace_back(0); in BuildLiteGraph()
42 liteGraph->output_indices_.emplace_back(1); in BuildLiteGraph()
45 for (size_t indexInput = 0; indexInput < liteGraph->input_indices_.size(); ++indexInput) { in BuildLiteGraph()
47 void* liteGraphTensor1 = MSLITE::MindIR_Tensor_Create(liteGraph->name_, in BuildLiteGraph()
49 liteGraph->all_tensors_.emplace_back(liteGraphTensor1); in BuildLiteGraph()
54 void* liteGraphTensor2 = MSLITE::MindIR_Tensor_Create(liteGraph->name_, in BuildLiteGraph()
56 liteGraph->all_tensors_.emplace_back(liteGraphTensor2); in BuildLiteGraph()
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/interfaces/innerkits/c/
H A Dneural_network_runtime_inner.h75 OH_NN_ReturnCode OH_NNModel_BuildFromLiteGraph(OH_NNModel *model, const void *liteGraph,
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/hdi_device/
H A Dhdi_device_test.cpp99 void LiteGraphDeleter(mindspore::lite::LiteGraph* liteGraph) in LiteGraphDeleter() argument
101 MindIR_LiteGraph_Destroy(&liteGraph); in LiteGraphDeleter()