Home
last modified time | relevance | path

Searched refs:cnodes (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.cc473 bool AscendStreamAssign::FinetuneSubgraphExecOrder(vector<CNodePtr> *cnodes) { in FinetuneSubgraphExecOrder() argument
474 MS_EXCEPTION_IF_NULL(cnodes); in FinetuneSubgraphExecOrder()
475 auto hcom_pos = find_if(cnodes->begin(), cnodes->end(), in FinetuneSubgraphExecOrder()
477 if (hcom_pos == cnodes->end()) { in FinetuneSubgraphExecOrder()
482 vector<CNodePtr> ori_cnodes(cnodes->begin(), cnodes->end()); in FinetuneSubgraphExecOrder()
483 cnodes->clear(); in FinetuneSubgraphExecOrder()
490 auto last_input_pos = cnodes->end(); in FinetuneSubgraphExecOrder()
492 auto pos = find(cnodes->begin(), cnodes->end(), input); in FinetuneSubgraphExecOrder()
493 if (pos != cnodes->end()) { in FinetuneSubgraphExecOrder()
494 … last_input_pos = (last_input_pos == cnodes->end() || last_input_pos < pos) ? pos : last_input_pos; in FinetuneSubgraphExecOrder()
[all …]
Dascend_stream_assign.h173 bool FinetuneSubgraphExecOrder(vector<CNodePtr> *cnodes);
/third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/
Dnode_parser_registry_test.cc67 std::vector<CNodePtr> cnodes; in TEST_F() local
70 cnodes.push_back(node->cast<CNodePtr>()); in TEST_F()
73 ASSERT_EQ(cnodes.size(), 2); in TEST_F()
74 auto cnode = cnodes.front(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc121 CNodePtrList FindUniqueCacheEnable(const CNodePtrList &cnodes) { in FindUniqueCacheEnable() argument
122 size_t cnodes_size = cnodes.size(); in FindUniqueCacheEnable()
125 if (IsPrimitiveCNode(cnodes[i], prim::kPrimUnique)) { in FindUniqueCacheEnable()
126 auto unique_node = cnodes[i]; in FindUniqueCacheEnable()
443 CNodePtrList FindSparseGatherV2WithCache(const CNodePtrList &cnodes, const ParamSet &param_set) { in FindSparseGatherV2WithCache() argument
444 size_t cnodes_size = cnodes.size(); in FindSparseGatherV2WithCache()
447 if (IsPrimitiveCNode(cnodes[i], prim::kPrimSparseGatherV2) || in FindSparseGatherV2WithCache()
448 IsPrimitiveCNode(cnodes[i], prim::kPrimEmbeddingLookup)) { in FindSparseGatherV2WithCache()
449 auto load_node = cnodes[i]->input(1); in FindSparseGatherV2WithCache()
456 sparse_gather_v2_with_cache.push_back(cnodes[i]); in FindSparseGatherV2WithCache()
[all …]
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Danf_exporter.cc83 std::list<CNodePtr> cnodes{}; in GetOrderedCNodes() local
88 cnodes.push_back(cnode); in GetOrderedCNodes()
91 return cnodes; in GetOrderedCNodes()
355 auto cnodes = GetOrderedCNodes(func_graph); in InsertCallNode() local
356 for (auto it = cnodes.begin(); it != cnodes.end();) { in InsertCallNode()
364 it = cnodes.insert(it, call_cnode); in InsertCallNode()
375 return cnodes; in InsertCallNode()
381 auto cnodes = InsertCallNode(func_graph); in Anf2Fb() local
382 for (const auto &cnode : cnodes) { in Anf2Fb()
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dspatial_node_adapter.cc133 auto cnodes = func_graph->GetOrderedCnodes(); in AdapteSpatialNode() local
134 for (const auto &cnode : cnodes) { in AdapteSpatialNode()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_pad_adjust.cc103 auto cnodes = func_graph->GetOrderedCnodes(); in Adjust() local
104 for (auto &cnode : cnodes) { in Adjust()
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dquant_cast.cc132 auto cnodes = graph->GetOrderedCnodes(); in Run() local
133 for (auto &cnode : cnodes) { in Run()
Dfull_quant_quantizer.cc780 auto cnodes = funcGraph->GetOrderedCnodes(); in QuantNode() local
781 for (auto &cnode : cnodes) { in QuantNode()
872 auto cnodes = funcGraph->GetOrderedCnodes(); in PreProcess() local
873 for (auto &cnode : cnodes) { in PreProcess()
1087 auto cnodes = func_graph->GetOrderedCnodes(); in BiasCorrection() local
1088 for (auto &cnode : cnodes) { in BiasCorrection()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse_checker.cc296 void MemReuseChecker::ExportNormalOpIr(const std::vector<CNodePtr> &cnodes) { in ExportNormalOpIr() argument
304 for (const auto &node : cnodes) { in ExportNormalOpIr()
361 const auto &cnodes = graph->execution_order(); in CheckNormalIR() local
362 for (const auto &node : cnodes) { in CheckNormalIR()
411 ExportNormalOpIr(cnodes); in CheckNormalIR()
Dmem_reuse_checker.h53 void ExportNormalOpIr(const std::vector<CNodePtr> &cnodes);
/third_party/boost/libs/regex/src/
Dfileiter.cpp330 long cnodes = (_size + buf_size - 1) / buf_size; in open() local
333 if(cnodes > (long)((INT_MAX) / sizeof(pointer*))) in open()
341 _first = new pointer[(int)cnodes]; in open()
342 _last = _first + cnodes; in open()
343 std::memset(_first, 0, cnodes*sizeof(pointer)); in open()
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dexecutor.cc325 const auto &cnodes = func_graph->GetOrderedCnodes(); in InitParamAggregator() local
326 for (const auto &cnode : cnodes) { in InitParamAggregator()
/third_party/mindspore/mindspore/ccsrc/ps/
Dparameter_server.cc474 std::list<CNodePtr> cnodes = func_graph_->GetOrderedCnodes(); in GetCNode() local
475 for (CNodePtr cnode : cnodes) { in GetCNode()
493 auto cnodes = func_graph_->GetOrderedCnodes(); in GetEmbeddingTableParamPtr() local
495 for (auto cnode : cnodes) { in GetEmbeddingTableParamPtr()
520 auto cnodes = func_graph_->GetOrderedCnodes(); in CacheEmbeddingTableParamPtr() local
521 for (auto cnode : cnodes) { in CacheEmbeddingTableParamPtr()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/
Dallreduce_fusion.cc146 const auto &cnodes = allreduce_graph_.cnode_set(); in AddEdgeToGraph() local
147 for (auto &cnode : cnodes) { in AddEdgeToGraph()
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph.cc641 std::list<CNodePtr> cnodes; in GetOrderedCnodes() local
646 cnodes.push_back(cnode); in GetOrderedCnodes()
649 return cnodes; in GetOrderedCnodes()
Dfunc_graph_cloner.cc198 auto &cnodes = func_graph->func_graph_cnodes_index(); in CloneFuncGraphValueNodes() local
199 for (auto &cnode : cnodes) { in CloneFuncGraphValueNodes()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc82 const std::vector<CNodePtr> &cnodes) { in FindCandidateRecomputedNodes() argument
85 for (const auto &cnode : cnodes) { in FindCandidateRecomputedNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.h86 std::string ExtractGraphKernelName(const AnfNodePtrList &cnodes, const string &prefix = "", const s…
Dgraph_kernel_helper.cc266 auto cnodes = fg->GetOrderedCnodes(); in MixedNodesTransToGraph() local
267 for (const auto &n : cnodes) { in MixedNodesTransToGraph()
588 std::string ExtractGraphKernelName(const AnfNodePtrList &cnodes, const string &prefix, const string… in ExtractGraphKernelName() argument
593 for (const auto &node : cnodes) { in ExtractGraphKernelName()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.cc365 auto cnodes = sub_graph->GetOrderedCnodes(); in CheckSubGraph() local
366 if (cnodes.size() != cnode_num || nodes.size() != all_node_num) { in CheckSubGraph()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Ddecrease_transpose_algo.cc102 void SetTransType(const std::set<CNodePtr> &cnodes, FormatTransNodeType *trans_type) { in SetTransType() argument
105 for (auto &cnode : cnodes) { in SetTransType()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline.cc199 auto cnodes = fg->GetOrderedCnodes(); in GetCachedFuncGraph() local
200 for (auto cnode : cnodes) { in GetCachedFuncGraph()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_auto_monad.cc1594 const auto &cnodes = graph_->execution_order(); in GenerateExecuteOrder() local
1595 for (auto &cnode : cnodes) { in GenerateExecuteOrder()