Searched refs:cnode_input (Results 1 – 4 of 4) sorted by relevance
368 auto cnode_input = (*it)->inputs(); in InsertCallNode() local369 cnode_input.insert(cnode_input.begin(), call_anf_prim_vnode); in InsertCallNode()370 (*it)->set_inputs(cnode_input); in InsertCallNode()637 auto cnode_input = input_cnode->inputs(); in ConvertInputCNode() local639 cnode_input.insert(cnode_input.begin(), call_anf_prim_vnode); in ConvertInputCNode()640 input_cnode->set_inputs(cnode_input); in ConvertInputCNode()897 auto cnode_input = cnode->inputs(); in CreatePartialCnode() local899 cnode_input.insert(cnode_input.begin(), partial_anf_prim_vnode); in CreatePartialCnode()900 cnode->set_inputs(cnode_input); in CreatePartialCnode()
613 auto cnode_input = input->cast<CNodePtr>(); in BuildAdjointForInput() local614 auto forged_adjoint = ForgeCNodeAdjoint(cnode_input); in BuildAdjointForInput()681 auto cnode_input = input->cast<CNodePtr>(); in BackPropagate() local682 if (cnode_input != nullptr && cnode_input->stop_gradient()) { in BackPropagate()
783 auto cnode_input = graph->GetBackendAnfByFrontAnf(attr_input); in CreateCallSwitchInputs() local784 auto switch_cnode = cnode_input->cast<CNodePtr>(); in CreateCallSwitchInputs()880 auto cnode_input = graph->GetBackendAnfByFrontAnf(attr_input); in CreateCallSwitchLayerInputs() local881 auto switch_layer_cnode = cnode_input->cast<CNodePtr>(); in CreateCallSwitchLayerInputs()948 auto cnode_input = graph->GetBackendAnfByFrontAnf(attr_input); in CreateSwitchOrPartialNode() local949 if (cnode_input == nullptr) { in CreateSwitchOrPartialNode()954 if (AnfAlgo::CheckPrimitiveType(cnode_input, prim::kPrimPartial)) { in CreateSwitchOrPartialNode()964 } else if (AnfAlgo::CheckPrimitiveType(cnode_input, prim::kPrimSwitch)) { in CreateSwitchOrPartialNode()966 } else if (AnfAlgo::CheckPrimitiveType(cnode_input, prim::kPrimSwitchLayer)) { in CreateSwitchOrPartialNode()969 MS_LOG(ERROR) << "CNode:" << cnode->DebugString() << " input[0]" << cnode_input->DebugString() in CreateSwitchOrPartialNode()
2231 auto cnode_input = node->input(i + 1); in InferShape() local2232 MS_EXCEPTION_IF_NULL(cnode_input); in InferShape()2245 … auto tuple_get_item_index = AnfAlgo::GetTupleGetItemOutIndex(cnode_input->cast<CNodePtr>()); in InferShape()2253 if (AnfAlgo::CheckPrimitiveType(cnode_input, prim::kPrimTupleGetItem)) { in InferShape()2261 auto tuple_get_item_indexk = AnfAlgo::GetTupleGetItemOutIndex(cnode_input->cast<CNodePtr>()); in InferShape()2264 …} else if (cnode_input->isa<CNode>() && AnfAlgo::GetCNodeName(cnode_input) == prim::kPrimReshape->… in InferShape()2265 (void)args_spec_list.emplace_back(cnode_input->abstract()); in InferShape()