Home
last modified time | relevance | path

Searched defs:partial_inputs (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dnode_pass.cc39 const auto &partial_inputs = partial_node->inputs(); in AddOutputAndCallerToMap() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dcall_graph_tuple_transform.h122 auto partial_inputs = node->cast<CNodePtr>()->inputs(); in TransformBranchNode() local
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc423 std::vector<AnfNodePtr> partial_inputs; in CreatePartialInBelongAnf() local
477 const FuncGraphPtr &sub_graph, std::vector<AnfNodePtr> *partial_inputs, in CreateParameterForPartialSubGraph()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc57 const auto &partial_inputs = node->cast<CNodePtr>()->inputs(); in GetFuncGraphFromPartial() local
76 const auto &partial_inputs = partial_node->cast<CNodePtr>()->inputs(); in FetchParameterBySwitchNode() local
104 const auto &partial_inputs = tuple_inputs[i]->cast<CNodePtr>()->inputs(); in FetchParameterBySwitchLayerNode() local
883 const auto &partial_inputs = partial_node->inputs(); in FetchValueNodeBySwitchNode() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dswitch_actor.cc96 auto partial_inputs = cnode->inputs(); in ParsePartialInput() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc753 …std::vector<AnfNodePtr> partial_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimPart… in CreateSwitchInput() local
798 std::vector<AnfNodePtr> partial_inputs = partial_node->inputs(); in CreateCallSwitchInputs() local
834 auto partial_inputs = return_input_cnode->inputs(); in ProcessNodeRetFunc() local
957 auto partial_inputs = partial_node->inputs(); in CreateSwitchOrPartialNode() local
1979 auto partial_inputs = partial_cnode->inputs(); in AddPartialParametersMap() local
/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc161 auto partial_inputs = partial_cnode->inputs(); in GetNextNodes() local