Searched refs:cnode_inputs (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | spatial_node_adapter.cc | 117 auto cnode_inputs = cnode->inputs(); in AdapteNodeWithDynamicInput() local
118 if (cnode_inputs.size() >= kCnodeInputMinNum) { in AdapteNodeWithDynamicInput()
119 new_inputs.insert(new_inputs.end(), cnode_inputs.begin() + 1, cnode_inputs.end()); in AdapteNodeWithDynamicInput()
127 const std::vector<AnfNodePtr> replace_node = {cnode_inputs[0], make_tuple_cnode}; in AdapteNodeWithDynamicInput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 669 void SessionBasic::GetCNodeInfo(const CNodePtr &cnode, std::vector<AnfNodePtr> *cnode_inputs) const… in GetCNodeInfo()
671 MS_EXCEPTION_IF_NULL(cnode_inputs); in GetCNodeInfo()
675 cnode_inputs->push_back(std::make_shared<ValueNode>(std::make_shared<Primitive>(*prim))); in GetCNodeInfo()
680 cnode_inputs->push_back(std::make_shared<ValueNode>(new_fg)); in GetCNodeInfo()
684 …GetNewCNodeInputs(const CNodePtr &cnode, KernelGraph *graph, std::vector<AnfNodePtr> *cnode_inputs, in GetNewCNodeInputs() argument
689 MS_EXCEPTION_IF_NULL(cnode_inputs); in GetNewCNodeInputs()
698 (void)cnode_inputs->emplace_back(graph->GetBackendAnfByFrontAnf(anf)); in GetNewCNodeInputs()
701 cnode_inputs->push_back(NewValueNode(MakeValue(SizeToInt(input_idx)))); in GetNewCNodeInputs()
704 cnode_inputs->push_back((*other_graph_cnode)[anf]); in GetNewCNodeInputs()
710 (void)cnode_inputs->emplace_back(new_value_node); in GetNewCNodeInputs()
[all …]
|
| D | session_basic.h | 175 …reateCNodeInputs(const CNodePtr &cnode, KernelGraph *graph, std::vector<AnfNodePtr> *cnode_inputs);
177 void GetCNodeInfo(const CNodePtr &cnode, std::vector<AnfNodePtr> *cnode_inputs) const;
178 …GetNewCNodeInputs(const CNodePtr &cnode, KernelGraph *graph, std::vector<AnfNodePtr> *cnode_inputs,
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | infer_mindir.cc | 359 auto &cnode_inputs = node->inputs(); in EvalFuncGraphAbastract() local
360 if (func_inputs.size() != cnode_inputs.size() - 1) { in EvalFuncGraphAbastract()
362 … << " CNode:" << node->DebugString() << " input size:" << cnode_inputs.size(); in EvalFuncGraphAbastract()
365 infer_resut_[func_inputs[i]] = infer_resut_[cnode_inputs[i + 1]]; in EvalFuncGraphAbastract()
394 auto &cnode_inputs = node->inputs(); in EvalFuncGraphAbastract() local
395 if (func_inputs.size() != cnode_inputs.size() - 1) { in EvalFuncGraphAbastract()
397 << " CNode:" << node->DebugString() << " input size:" << cnode_inputs.size(); in EvalFuncGraphAbastract()
401 SaveNodeInferResult(func_inputs[i], infer_resut_[cnode_inputs[i + 1]]); in EvalFuncGraphAbastract()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | clean.cc | 641 std::vector<AnfNodePtr> cnode_inputs{inputs[0]}; in EraseTuple() local
642 (void)cnode_inputs.insert(cnode_inputs.end(), new_inputs.begin(), new_inputs.end()); in EraseTuple()
645 auto new_node = node->func_graph()->NewCNode(cnode_inputs); in EraseTuple()
657 std::vector<AnfNodePtr> cnode_inputs{inputs[0], inputs[1]}; in EraseTuple() local
658 (void)cnode_inputs.insert(cnode_inputs.end(), new_inputs.begin(), new_inputs.end()); in EraseTuple()
661 auto new_node = cnode->func_graph()->NewCNode(cnode_inputs); in EraseTuple()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | control_node_parser.cc | 585 const auto &cnode_inputs = cnode->inputs(); in FetchFuncGraphbyCallNode() local
587 for (size_t i = kSwitchTrueBranchPos; i < cnode_inputs.size(); ++i) { in FetchFuncGraphbyCallNode()
588 if (IsPrimitiveCNode(cnode_inputs[i], prim::kPrimPartial)) { in FetchFuncGraphbyCallNode()
589 (void)func_graphs.emplace_back(GetFuncGraphFromPartial(cnode_inputs[i])); in FetchFuncGraphbyCallNode()
590 } else if (IsValueNode<FuncGraph>(cnode_inputs[i])) { in FetchFuncGraphbyCallNode()
591 (void)func_graphs.emplace_back(GetValueNode<FuncGraphPtr>(cnode_inputs[i])); in FetchFuncGraphbyCallNode()
595 … AnfAlgo::CheckPrimitiveType(cnode_inputs[kSwitchLayerBranchPos], prim::kPrimMakeTuple)) { in FetchFuncGraphbyCallNode()
596 const auto &tuple_inputs = cnode_inputs[kSwitchLayerBranchPos]->cast<CNodePtr>()->inputs(); in FetchFuncGraphbyCallNode()
|