Searched refs:abs_ref (Results 1 – 3 of 3) sorted by relevance
136 auto abs_ref = args_spec_list[0]->cast<AbstractRefPtr>(); in InferImplGetRefKey() local137 MS_EXCEPTION_IF_NULL(abs_ref); in InferImplGetRefKey()138 return abs_ref->ref(); in InferImplGetRefKey()152 auto abs_ref = args_spec_list[0]->cast<AbstractRefPtr>(); in InferImplGetRefValue() local153 MS_EXCEPTION_IF_NULL(abs_ref); in InferImplGetRefValue()154 return abs_ref->ref(); in InferImplGetRefValue()
523 auto abs_ref = std::make_shared<abstract::AbstractRef>(abs_ref_key, abs_value); in AbstractSpecializeAction() local524 context->ParallelParameterContextRestoreShape(func_graph, param_node, abs_ref); in AbstractSpecializeAction()525 args_spec.push_back(abs_ref); in AbstractSpecializeAction()526 context->ParallelParameterContextCkptShape(func_graph, param_node, abs_ref); in AbstractSpecializeAction()1008 auto abs_ref = std::make_shared<abstract::AbstractRef>(abs_ref_key, abs_value); in SetMindIRGraphAction() local1009 broaded_args.push_back(abs_ref); in SetMindIRGraphAction()
406 auto abs_ref = abstract->cast<AbstractRefPtr>(); in Run() local407 … abstract = std::make_shared<AbstractRef>(abs_ref->ref_key()->Broaden(), abs_ref); in Run()