Searched refs:partial_inputs (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/common/ |
| D | func_graph_subgraph.cc | 423 std::vector<AnfNodePtr> partial_inputs; in CreatePartialInBelongAnf() local
425 …auto ret = CreateParameterForPartialSubGraph(func_graph, &partial_inputs, &partial_inputs_and_subg… in CreatePartialInBelongAnf()
448 partial_inputs.insert(partial_inputs.begin(), graph_value_node); in CreatePartialInBelongAnf()
449 auto partial_cnode = belong_anf_->NewCNode(partial_prim, partial_inputs); in CreatePartialInBelongAnf()
452 for (size_t i = 0; i < partial_inputs.size(); ++i) { in CreatePartialInBelongAnf()
453 const auto &input = partial_inputs.at(i); in CreatePartialInBelongAnf()
477 const FuncGraphPtr &sub_graph, std::vector<AnfNodePtr> *partial_inputs, in CreateParameterForPartialSubGraph() argument
480 MS_ASSERT(partial_inputs != nullptr && partial_inputs->empty()); in CreateParameterForPartialSubGraph()
504 partial_inputs->emplace_back(input); in CreateParameterForPartialSubGraph()
522 partial_inputs->emplace_back(new_parameter); in CreateParameterForPartialSubGraph()
[all …]
|
| D | func_graph_subgraph.h | 64 …ParameterForPartialSubGraph(const FuncGraphPtr &sub_graph, std::vector<AnfNodePtr> *partial_inputs,
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | node_pass.cc | 39 const auto &partial_inputs = partial_node->inputs(); in AddOutputAndCallerToMap() local
40 if (!IsPrimitive(partial_inputs.at(0), prim::kPrimPartial)) { in AddOutputAndCallerToMap()
43 auto switch_subgraph = GetValueNode<FuncGraphPtr>(partial_inputs.at(kPartialArgsIndex)); in AddOutputAndCallerToMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | call_graph_tuple_transform.h | 122 auto partial_inputs = node->cast<CNodePtr>()->inputs(); in TransformBranchNode() local
123 if (IsValueNode<FuncGraph>(partial_inputs[1])) { in TransformBranchNode()
124 FuncGraphPtr fg = GetValueNode<FuncGraphPtr>(partial_inputs[1]); in TransformBranchNode()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | control_node_parser.cc | 57 const auto &partial_inputs = node->cast<CNodePtr>()->inputs(); in GetFuncGraphFromPartial() local
58 return GetValueNode<FuncGraphPtr>(partial_inputs[1]); in GetFuncGraphFromPartial()
76 const auto &partial_inputs = partial_node->cast<CNodePtr>()->inputs(); in FetchParameterBySwitchNode() local
77 for (size_t j = kPartialInputStartPos; j < partial_inputs.size(); ++j) { in FetchParameterBySwitchNode()
78 if (CheckValidFuncGraphInput(partial_inputs[j])) { in FetchParameterBySwitchNode()
79 (void)parameters.emplace_back(partial_inputs[j]); in FetchParameterBySwitchNode()
104 const auto &partial_inputs = tuple_inputs[i]->cast<CNodePtr>()->inputs(); in FetchParameterBySwitchLayerNode() local
107 for (size_t j = kPartialInputStartPos; j < partial_inputs.size(); ++j) { in FetchParameterBySwitchLayerNode()
108 if (CheckValidFuncGraphInput(partial_inputs[j])) { in FetchParameterBySwitchLayerNode()
109 (void)parameters.emplace_back(partial_inputs[j]); in FetchParameterBySwitchLayerNode()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | switch_actor.cc | 96 auto partial_inputs = cnode->inputs(); in ParsePartialInput() local
97 if (partial_inputs.size() <= kPartialFuncGraphPos) { in ParsePartialInput()
101 auto func_graph = GetValueNode<FuncGraphPtr>(partial_inputs[kPartialFuncGraphPos]); in ParsePartialInput()
104 for (size_t j = kPartialInputStartPos; j < partial_inputs.size(); ++j) { in ParsePartialInput()
105 AddInput(partial_inputs[j], branch_id); in ParsePartialInput()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | graph_partition.cc | 161 auto partial_inputs = partial_cnode->inputs(); in GetNextNodes() local
162 std::reverse(partial_inputs.begin(), partial_inputs.end()); in GetNextNodes()
163 … (void)extend_inputs.insert(extend_inputs.end(), partial_inputs.begin(), partial_inputs.end()); in GetNextNodes()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 753 …std::vector<AnfNodePtr> partial_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimPart… in CreateSwitchInput() local
758 partial_inputs.emplace_back(graph->GetBackendAnfByFrontAnf(node_input)); in CreateSwitchInput()
769 partial_inputs.emplace_back(std::make_shared<ValueNode>(kernel_graph)); in CreateSwitchInput()
770 partial_inputs.emplace_back(graph->GetBackendAnfByFrontAnf(node_input)); in CreateSwitchInput()
772 auto partial_node = graph->NewCNode(partial_inputs); in CreateSwitchInput()
798 std::vector<AnfNodePtr> partial_inputs = partial_node->inputs(); in CreateCallSwitchInputs() local
801 (void)partial_inputs.emplace_back(graph->GetBackendAnfByFrontAnf(cnode->input(i))); in CreateCallSwitchInputs()
803 auto new_partial = graph->NewCNode(partial_inputs); in CreateCallSwitchInputs()
834 auto partial_inputs = return_input_cnode->inputs(); in ProcessNodeRetFunc() local
835 …call_inputs.insert(call_inputs.end(), partial_inputs.begin() + kFirstDataInputIndex, partial_input… in ProcessNodeRetFunc()
[all …]
|