Searched refs:ParallelParameterContextRestoreShape (Results 1 – 3 of 3) sorted by relevance
137 …void ParallelParameterContextRestoreShape(const FuncGraphPtr &func_graph, const ParameterPtr ¶…
231 void ParallelContext::ParallelParameterContextRestoreShape(const FuncGraphPtr &func_graph, in ParallelParameterContextRestoreShape() function in mindspore::parallel::ParallelContext
524 context->ParallelParameterContextRestoreShape(func_graph, param_node, abs_ref); in AbstractSpecializeAction()