Searched defs:partial_inputs (Results 1 – 8 of 8) sorted by relevance
105 const auto &partial_inputs = partial->inputs(); in operator() local
808 …fNodePtr ExpandVmapPrim::BindInAxis(const CNodePtr &vmap_app, const AnfNodePtrList &partial_inputs, in BindInAxis()862 …apPrim::PostProcessVmap(const AnfNodePtr &expanded_vmap_node, const AnfNodePtrList &partial_inputs, in PostProcessVmap()
425 std::vector<AnfNodePtr> partial_inputs; in CreatePartialInBelongAnf() local481 const FuncGraphPtr &sub_graph, std::vector<AnfNodePtr> *partial_inputs, in CreateParameterForPartialSubGraph()
85 const auto &partial_inputs = partial_node->inputs(); in ModifyOutputAndCallerToMap() local
1311 …std::vector<AnfNodePtr> partial_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimPart… in CreateSwitchInput() local1419 std::vector<AnfNodePtr> partial_inputs = partial_node->inputs(); in CreateCallSwitchInputs() local1456 auto partial_inputs = return_input_cnode->inputs(); in ProcessNodeRetFunc() local1582 auto partial_inputs = partial_node->inputs(); in CreateSwitchOrPartialNode() local2617 auto partial_inputs = partial_cnode->inputs(); in AddPartialParametersMap() local
183 auto partial_inputs = partial_cnode->inputs(); in GetNextNodes() local
2624 auto partial_inputs = partial->cast<CNodePtr>()->inputs(); in SupportInlinePartial() local
2800 AnfNodeWeakPtrList partial_inputs; in EvalPrim() local