Searched refs:depend_inputs (Results 1 – 5 of 5) sorted by relevance
50 …AnfNodePtrList depend_inputs = {NewValueNode(prim::kPrimDepend), original_inputs[input_idx], origi… in ProcessNode() local51 auto depend_cnode = func_graph->NewCNode(depend_inputs); in ProcessNode()
698 auto &depend_inputs = inputs[0]->cast<CNodePtr>()->inputs(); in operator() local699 if (depend_inputs.size() != 3) { in operator()703 std::vector<AnfNodePtr> new_inputs({depend_inputs[1]}); in operator()709 … auto new_node = node->func_graph()->NewCNode({depend_inputs[0], new_call_node, depend_inputs[2]}); in operator()
395 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), first_input, in NewRecomputedNode() local397 auto depend_node = graph->NewCNode(depend_inputs); in NewRecomputedNode()
90 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), add, add_second}; in TEST_F() local91 auto depend = kernel_graph->NewCNode(depend_inputs); in TEST_F()
1682 auto depend_inputs = node->cast<CNodePtr>()->inputs(); in GetRealOpNode() local1683 if (depend_inputs.size() != 3) { // "Depend" primitive have 3 inputs in GetRealOpNode()1688 return GetRealOpNode(depend_inputs[1]); in GetRealOpNode()