Searched refs:func_output (Results 1 – 2 of 2) sorted by relevance
96 …GraphOutput(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const AnfNodePtr &func_output) { in InsertCastForGraphOutput() argument107 if (used_node != func_output) { in InsertCastForGraphOutput()139 auto func_output = func_graph->output(); in Run() local143 InsertCastForGraphOutput(func_graph, cnode, func_output); in Run()
707 auto func_output = func_graph->output(); in GetFuncGraphOutputNodes() local708 MS_EXCEPTION_IF_NULL(func_output); in GetFuncGraphOutputNodes()709 if (func_output->isa<CNode>()) { in GetFuncGraphOutputNodes()711 auto cnode = func_output->cast<CNodePtr>(); in GetFuncGraphOutputNodes()726 output_list->push_back(AnfAlgo::VisitKernel(func_output, 0).first); in GetFuncGraphOutputNodes()730 output_list->push_back(AnfAlgo::VisitKernel(func_output, 0).first); in GetFuncGraphOutputNodes()