Home
last modified time | relevance | path

Searched refs:all_nodes_ (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/lite/src/common/
Dgraph_util.cc36 auto node_size = model->graph_.all_nodes_.size(); in GetGraphInputNodes()
38 auto node = model->graph_.all_nodes_[j]; in GetGraphInputNodes()
55 auto node_size = model->graph_.all_nodes_.size(); in GetGraphOutputNodes()
57 auto node = model->graph_.all_nodes_[j]; in GetGraphOutputNodes()
73 auto nodes_size = model->graph_.all_nodes_.size(); in GetLinkedPostNodeIdx()
75 auto node = model->graph_.all_nodes_[i]; in GetLinkedPostNodeIdx()
/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir_nnrt_lite_graph.cc34 for (size_t idx = 0; idx < graph->all_nodes_.size(); idx++) { in MindIR_LiteGraph_Destroy()
35 if (graph->all_nodes_[idx] != nullptr) { in MindIR_LiteGraph_Destroy()
37 static_cast<schema::Primitive *>(graph->all_nodes_[idx]->primitive_)); in MindIR_LiteGraph_Destroy()
38 delete graph->all_nodes_[idx]; in MindIR_LiteGraph_Destroy()
Dmindir_nnrt_lite_graph_to_model.cc88 MS_LOG(INFO) << "Start converting nodes, vector size = " << lite_graph->all_nodes_.size(); in MindIR_LiteGraph_To_Model()
89 nodes.reserve(lite_graph->all_nodes_.size()); in MindIR_LiteGraph_To_Model()
90 for (auto node : lite_graph->all_nodes_) { in MindIR_LiteGraph_To_Model()
/third_party/mindspore/mindspore/lite/src/
Dsub_graph_split.cc70 this->model_->graph_.all_nodes_.begin(), this->model_->graph_.all_nodes_.end(), in UpdateOfflineParallelFlag()
85 auto head_node = model_->graph_.all_nodes_.at(heads.front()); in CheckIsParallelSubGraph()
86 auto end_node = model_->graph_.all_nodes_.at(ends.front()); in CheckIsParallelSubGraph()
98 …if (GetPrimitiveType(model_->graph_.all_nodes_.at(input_node_index)->primitive_, SCHEMA_VERSION::S… in CheckIsParallelSubGraph()
110 …if (GetPrimitiveType(model_->graph_.all_nodes_.at(output_node_index)->primitive_, SCHEMA_VERSION::… in CheckIsParallelSubGraph()
224 int partial_index = static_cast<int>(model_->graph_.all_nodes_.size()); in ConvertSubGraphToModel()
253 LiteGraph::Node *cur_node = model_->graph_.all_nodes_[node_index]; in ConvertSubGraphToModel()
267 LiteGraph::Node *head_node = model_->graph_.all_nodes_[head_index]; in ConvertSubGraphToModel()
293 LiteGraph::Node *end_node = model_->graph_.all_nodes_[end_index]; in ConvertSubGraphToModel()
300 model_->graph_.all_nodes_.push_back(std::move(new_partial_node)); in ConvertSubGraphToModel()
[all …]
Dlite_model.cc72 for (size_t index = 0; index < this->graph_.all_nodes_.size(); ++index) { in ConvertAttrToTensors()
74 int status = ConvertAttrs(this->graph_.all_nodes_[index], &dst_tensors); in ConvertAttrToTensors()
93 this->graph_.all_nodes_[index]->input_indices_.push_back(cur_all_tensors_size++); in ConvertAttrToTensors()
106 auto nodes_size = this->graph_.all_nodes_.size(); in Free()
108 auto node = this->graph_.all_nodes_[i]; in Free()
132 auto nodes_size = this->graph_.all_nodes_.size(); in Destroy()
134 auto node = this->graph_.all_nodes_[i]; in Destroy()
138 this->graph_.all_nodes_.clear(); in Destroy()
203 for (auto &node : this->graph_.all_nodes_) { in NodeVerify()
232 auto node_size = this->graph_.all_nodes_.size(); in SubGraphVerify()
Dlite_model.h94 this->all_nodes_.push_back(node); in ConvertNodes()
135 this->graph_.all_nodes_.push_back(node); in ConvertNodes()
Dscheduler.cc691 for (auto &iter : src_model_->graph_.all_nodes_) { in NodeInputIsPartial()
733 auto node = src_model_->graph_.all_nodes_[node_index]; in InferSubGraphShape()
1266 auto node = src_model_->graph_.all_nodes_[node_index]; in SubGraphPreferDataType()
1382 for (auto &node : src_model_->graph_.all_nodes_) { in IsControlFlowPattern()
1421 auto node = src_model_->graph_.all_nodes_[node_index]; in ScheduleSubGraphToKernels()
1657 for (auto &node : src_model_->graph_.all_nodes_) { in InferSwitchShape()
1681 for (auto &iter : src_model_->graph_.all_nodes_) { in NodeInputIsSwitch()
Dlite_session.cc307 auto in_node = model->graph_.all_nodes_[in_node_index]; in InitGraphInputMap()
343 auto out_node = model->graph_.all_nodes_[out_node_index]; in InitGraphOutputNodeMap()
/third_party/flutter/skia/third_party/externals/spirv-tools/source/util/
Dhuffman_codec.h50 all_nodes_.reserve(3 * hist.size()); in HuffmanCodec()
277 all_nodes_.emplace_back(new Node()); in CreateNode()
278 all_nodes_.back()->id = next_node_id_++; in CreateNode()
279 return all_nodes_.back().get(); in CreateNode()
286 std::vector<std::unique_ptr<Node>> all_nodes_; variable
/third_party/mindspore/mindspore/lite/micro/coder/
Dgraph.cc135 auto in_node = model_->all_nodes_.at(in_node_index); in InitGraphInOutTensors()
161 auto *out_node = model_->all_nodes_.at(out_node_index); in InitGraphInOutTensors()
250 …std::for_each(model_->all_nodes_.begin(), model_->all_nodes_.end(), [this, target](const Model::No… in DumpUnSupportLayer()
Dsession.cc237 uint32_t nodes_size = model->all_nodes_.size(); in CreateOpCoders()
240 const auto *node = model->all_nodes_.at(i); in CreateOpCoders()
/third_party/mindspore/mindspore/lite/src/delegate/nnrt/
Dnnrt_delegate.cc291 for (auto node : lite_graph.all_nodes_) { in ShallowCopyLiteGraph()
299 node_list.reserve(lite_graph.all_nodes_.size()); in ShallowCopyLiteGraph()
301 for (auto node : lite_graph.all_nodes_) { in ShallowCopyLiteGraph()
351 nnrt_lite_graph->all_nodes_ = node_list; in ShallowCopyLiteGraph()
/third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/include/
Dmodel.h43 NodePtrVector all_nodes_; member
/third_party/mindspore/mindspore/core/ir/
Dmanager.cc217 all_nodes_ = AnfNodeSet(); in Reset()
363 all_nodes_.clear(); in Clear()
461 all_nodes_.difference_update(fg->parameters()); in MaybeDropFuncGraphs()
526 auto insert_result = all_nodes_.insert(node); in AcquireNodes()
567 if (!all_nodes_.contains(node)) { in MaybeDropNodes()
597 (void)all_nodes_.erase(node); in MaybeDropNodes()
663 (void)all_nodes_.erase(source_return); in MoveAllCNodeDropGraph()
674 all_nodes_.difference_update(source->parameters()); in MoveAllCNodeDropGraph()
Dmanager.h314 AnfNodeSet &all_nodes() { return all_nodes_; } in all_nodes()
341 AnfNodeSet all_nodes_; // managed nodes variable
/third_party/mindspore/mindspore/lite/mindir/include/
Dmindir_lite_graph.h41 std::vector<Node *> all_nodes_; member
/third_party/mindspore/mindspore/lite/include/
Dmodel.h56 NodePtrVector all_nodes_; member
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_export.cc178 auto nodes = model->all_nodes_; in FindNode()