Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dops_builder.cpp29 const std::unordered_map<uint32_t, uint32_t>& modelIDToGraphID) const in GetInputIndex()
33 [modelIDToGraphID](uint32_t index) { in GetInputIndex()
34 return modelIDToGraphID.at(index); in GetInputIndex()
39 … const std::unordered_map<uint32_t, uint32_t>& modelIDToGraphID) const in GetOutputIndex()
43 [modelIDToGraphID](uint32_t index) { in GetOutputIndex()
44 return modelIDToGraphID.at(index); in GetOutputIndex()
H A Dinner_model.cpp587 std::unordered_map<uint32_t, uint32_t> modelIDToGraphID; in Build() local
588 AddTensorsToLiteGraph(modelIDToGraphID); in Build()
590 OH_NN_ReturnCode ret = AddNodesToLiteGraph(modelIDToGraphID); in Build()
614 void InnerModel::AddTensorsToLiteGraph(std::unordered_map<uint32_t, uint32_t>& modelIDToGraphID) in AddTensorsToLiteGraph() argument
628 modelIDToGraphID[i] = graphID++; in AddTensorsToLiteGraph()
635 [modelIDToGraphID](uint32_t index) {return modelIDToGraphID.at(index);}); in AddTensorsToLiteGraph()
639 [modelIDToGraphID](uint32_t index) {return modelIDToGraphID.at(index);}); in AddTensorsToLiteGraph()
642 …ode InnerModel::AddNodesToLiteGraph(const std::unordered_map<uint32_t, uint32_t>& modelIDToGraphID) in AddNodesToLiteGraph() argument
659 op->GetInputIndex(node->input_indices_, modelIDToGraphID); in AddNodesToLiteGraph()
660 op->GetOutputIndex(node->output_indices_, modelIDToGraphID); in AddNodesToLiteGraph()
H A Dops_builder.h56 … const std::unordered_map<uint32_t, uint32_t>& modelIDToGraphID) const;
58 … const std::unordered_map<uint32_t, uint32_t>& modelIDToGraphID) const;
H A Dinner_model.h65 void AddTensorsToLiteGraph(std::unordered_map<uint32_t, uint32_t>& modelIDToGraphID);
66 …_NN_ReturnCode AddNodesToLiteGraph(const std::unordered_map<uint32_t, uint32_t>& modelIDToGraphID);