Searched refs:forward_graph (Results 1 – 3 of 3) sorted by relevance
408 auto forward_graph = *(graph_set.begin()); in ProcessAllreduceFusion() local409 MS_EXCEPTION_IF_NULL(forward_graph); in ProcessAllreduceFusion()410 forward_ret_ = forward_graph->get_return(); in ProcessAllreduceFusion()
675 FuncGraphPtr forward_graph = real_fn->func_graph(); in GenerateFuncGraph() local676 MS_EXCEPTION_IF_NULL(forward_graph); in GenerateFuncGraph()677 forward_graph->set_flag(FUNC_GRAPH_FLAG_DEFER_INLINE, true); in GenerateFuncGraph()680 TraceGuard g(std::make_shared<TraceGradOperation>(forward_graph->debug_info())); in GenerateFuncGraph()683 auto nparam = forward_graph->parameters().size(); in GenerateFuncGraph()703 TraceGuard guard(std::make_shared<TraceGradOperation>(forward_graph->debug_info())); in GenerateFuncGraph()704 …k_child = GetGrad(j, weights, forward_graph->parameters(), forward_graph->has_flag("enable_tuple_g… in GenerateFuncGraph()
179 FuncGraphPtr forward_graph = real_fn->func_graph(); in Run() local180 MS_EXCEPTION_IF_NULL(forward_graph); in Run()190 auto new_graph = forward_graph->GenerateGraph(graph_specialize_args_without_sens); in Run()