| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | optimizer_kernel_factory.cc | 26 std::string cnode_name = AnfAlgo::GetCNodeName(kernel_node); in Matched() local
27 if (kNameToIdxMap.count(cnode_name) == 0) { in Matched()
28 MS_LOG(ERROR) << "Can't find index info for kernel " << cnode_name; in Matched()
32 auto input_name_to_idx = kNameToIdxMap.at(cnode_name).at("inputs"); in Matched()
38 MS_LOG(EXCEPTION) << cnode_name << " does not have input named " << name; in Matched()
49 auto output_name_to_idx = kNameToIdxMap.at(cnode_name).at("outputs"); in Matched()
55 MS_LOG(EXCEPTION) << cnode_name << " does not have output named " << name; in Matched()
|
| D | aggregation_kernel_factory.cc | 26 std::string cnode_name = AnfAlgo::GetCNodeName(kernel_node); in Matched() local
27 if (kNameToIdxMap.count(cnode_name) == 0) { in Matched()
28 MS_LOG(ERROR) << "Can't find index info for kernel " << cnode_name; in Matched()
32 auto input_name_to_idx = kNameToIdxMap.at(cnode_name).at("inputs"); in Matched()
38 MS_LOG(DEBUG) << cnode_name << " does not have input named " << name in Matched()
50 auto output_name_to_idx = kNameToIdxMap.at(cnode_name).at("outputs"); in Matched()
56 MS_LOG(DEBUG) << cnode_name << " does not have output named " << name in Matched()
|
| D | dense_grad_accum_kernel.h | 41 std::string cnode_name = AnfAlgo::GetCNodeName(kernel_node); in InitKernel() local
42 if (kNameToIdxMap.count(cnode_name) == 0 || kNameToIdxMap.at(cnode_name).count("inputs") == 0 || in InitKernel()
43 kNameToIdxMap.at(cnode_name).at("inputs").count("grad") == 0) { in InitKernel()
44 MS_LOG(EXCEPTION) << "Can't find index info of grad for kernel " << cnode_name; in InitKernel()
47 size_t cnode_grad_idx = kNameToIdxMap.at(cnode_name).at("inputs").at("grad"); in InitKernel()
|
| D | fed_avg_kernel.h | 57 std::string cnode_name = AnfAlgo::GetCNodeName(kernel_node); in InitKernel() local
58 if (kNameToIdxMap.count(cnode_name) == 0 || kNameToIdxMap.at(cnode_name).count("inputs") == 0 || in InitKernel()
59 kNameToIdxMap.at(cnode_name).at("inputs").count("weight") == 0) { in InitKernel()
60 MS_LOG(EXCEPTION) << "Can't find index info of weight for kernel " << cnode_name; in InitKernel()
63 cnode_weight_idx_ = kNameToIdxMap.at(cnode_name).at("inputs").at("weight"); in InitKernel()
77 name_ = cnode_name + "." + weight_node->fullname_with_scope(); in InitKernel()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | parallel_pass.cc | 53 std::string cnode_name = cnode->fullname_with_scope(); in SetParallelOpName() local
54 if (cnode_name.find(PARALLEL_NAME_SUFFIX) != std::string::npos) { in SetParallelOpName()
55 MS_LOG(DEBUG) << " : Skip splited cnode " << cnode_name; in SetParallelOpName()
64 MS_LOG(DEBUG) << " : Reached a parallel care node: " << cnode_name; in SetParallelOpName()
69 cnode->set_fullname_with_scope(cnode_name + PARALLEL_NAME_SUFFIX); in SetParallelOpName()
127 std::string cnode_name = cnode->fullname_with_scope(); in Run() local
128 OperatorInfoPtr parallel_operator = CreateParallelOperator(cnode, cnode_name, parallel_op_name); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | parameter_aggregator.cc | 367 std::string cnode_name = AnfAlgo::GetCNodeName(cnode); in JudgeRequiresAggr() local
368 if (kNameToIdxMap.count(cnode_name) == 0 || kNameToIdxMap.at(cnode_name).count("inputs") == 0 || in JudgeRequiresAggr()
369 kNameToIdxMap.at(cnode_name).at("inputs").count("weight") == 0) { in JudgeRequiresAggr()
370 MS_LOG(EXCEPTION) << "Can't find index info of weight for kernel " << cnode_name; in JudgeRequiresAggr()
373 size_t cnode_weight_idx = kNameToIdxMap.at(cnode_name).at("inputs").at("weight"); in JudgeRequiresAggr()
|
| D | executor.cc | 308 std::string cnode_name = AnfAlgo::GetCNodeName(cnode); in GetTrainableParamName() local
309 if (kNameToIdxMap.count(cnode_name) == 0) { in GetTrainableParamName()
312 const OptimParamNameToIndex &index_info = kNameToIdxMap.at(cnode_name); in GetTrainableParamName()
317 MS_LOG(EXCEPTION) << weight_idx << " input of " << cnode_name << " is not a Parameter."; in GetTrainableParamName()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/common/ |
| D | gllo_utils.h | 112 const std::string &cnode_name);
115 const std::string &cnode_name);
|
| D | gllo_utils.cc | 1073 const std::string &cnode_name) { in GenTransposeNode() argument
1076 auto perm_node = BuildIntVecParameterNode(func_graph, perm, cnode_name + "_perm"); in GenTransposeNode()
1086 cnode->set_fullname_with_scope(cnode_name); in GenTransposeNode()
1094 const std::string &cnode_name) { in GenGatherNode() argument
1099 auto indices_node = BuildIntVecParameterNode(func_graph, indices, cnode_name + "_indices"); in GenGatherNode()
1104 auto axis_node = BuildIntVecParameterNode(func_graph, {0}, cnode_name + "_indices"); in GenGatherNode()
1118 cnode->set_fullname_with_scope(cnode_name); in GenGatherNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.cc | 197 auto cnode_name = AnfAlgo::GetCNodeName(anf_node); in RunAttrPass() local
198 auto FPass = build_json_attr_pass_map_.find(cnode_name); in RunAttrPass()
421 auto cnode_name = AnfAlgo::GetCNodeName(node); in IsPlaceHolderInput() local
422 if (node_set.find(cnode_name) == node_set.end()) { in IsPlaceHolderInput()
431 MS_LOG(EXCEPTION) << "Cnode: " << cnode_name << "doesn't has attribute placeholder_index."; in IsPlaceHolderInput()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.h | 60 static void DoFusion(const FuncGraphPtr &func_graph, const std::string &cnode_name,
|
| D | parameter_server.cc | 497 std::string cnode_name = AnfAlgo::GetCNodeName(cnode); in GetEmbeddingTableParamPtr() local
498 …if (cnode_name == kEmbeddingLookupOpName || cnode_name == kGatherV2OpName || cnode_name == kSparse… in GetEmbeddingTableParamPtr()
523 std::string cnode_name = AnfAlgo::GetCNodeName(cnode); in CacheEmbeddingTableParamPtr() local
524 if (cnode_name != kGatherV2OpName && cnode_name != kSparseGatherV2OpName) { in CacheEmbeddingTableParamPtr()
|
| D | util.cc | 139 void Util::DoFusion(const FuncGraphPtr &func_graph, const std::string &cnode_name, in DoFusion() argument
149 if (AnfAlgo::GetCNodeName(node) == cnode_name) { in DoFusion()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.cc | 407 auto cnode_name = AnfAlgo::GetCNodeName(output_info.first); in GetRealNodeUsedList() local
408 if ((cnode_name == prim::kPrimDepend->name() && output_info.second == kDependAttachNodeIndex) || in GetRealNodeUsedList()
409 (cnode_name == prim::kPrimUpdateState->name())) { in GetRealNodeUsedList()
430 auto cnode_name = AnfAlgo::GetCNodeName(output_info.first); in GetRealNodeUsedListByOutputIdx() local
431 if ((cnode_name == prim::kPrimDepend->name() && output_info.second == kDependAttachNodeIndex) || in GetRealNodeUsedListByOutputIdx()
432 (cnode_name == prim::kPrimUpdateState->name())) { in GetRealNodeUsedListByOutputIdx()
436 if (cnode_name == prim::kPrimTupleGetItem->name()) { in GetRealNodeUsedListByOutputIdx()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | quantize_util.cc | 594 auto cnode_name = cnode->fullname_with_scope(); in CopyFuncGraph() local
595 auto old_cnode_iter = old_cnode_map.find(cnode_name); in CopyFuncGraph()
597 MS_LOG(ERROR) << "can not find node: " << cnode_name; in CopyFuncGraph()
606 MS_LOG(ERROR) << "Param node has no default parameter: " << cnode_name; in CopyFuncGraph()
611 MS_LOG(ERROR) << "Default param of param node is not a tensor info:" << cnode_name; in CopyFuncGraph()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | transpose_fusion.cc | 126 const std::string &cnode_name) { in GenTransposeNode() argument
132 cnode->set_fullname_with_scope(cnode_name); in GenTransposeNode()
|
| /third_party/mindspore/mindspore/lite/tools/anf_exporter/ |
| D | anf_exporter.cc | 791 std::string cnode_name = fb_node->name; in SetOpOutputNode() local
825 ms_tensor->name = cnode_name; in SetOpOutputNode()
828 ms_tensor->name = cnode_name + "_o:" + std::to_string(i); in SetOpOutputNode()
867 ms_tensor->name = cnode_name; in SetOpOutputNode()
|