Searched refs:parameter_ptr (Results 1 – 3 of 3) sorted by relevance
175 auto parameter_ptr = cnode->input(1)->cast<ParameterPtr>(); in HandleReceiveParam() local176 MS_EXCEPTION_IF_NULL(parameter_ptr); in HandleReceiveParam()177 auto accu_parameter = FindGradAccuParameter(parameters, parameter_ptr->name()); in HandleReceiveParam()206 auto parameter_ptr = parameter->cast<ParameterPtr>(); in AddVirtualAssignAdd() local207 auto accu_parameter = FindGradAccuParameter(parameters, parameter_ptr->name()); in AddVirtualAssignAdd()
192 auto parameter_ptr = p->cast<ParameterPtr>(); in DumpParams() local193 if (parameter_ptr == nullptr) { in DumpParams()196 buffer << "%para" << para << "_" << parameter_ptr->name() << " : "; in DumpParams()206 buffer << " : IsWeight:" << std::boolalpha << AnfAlgo::IsParameterWeight(parameter_ptr); in DumpParams()
1643 auto parameter_ptr = parameter->cast<ParameterPtr>(); in SetSharedParameterFlag() local1644 if (!parameter_ptr) { in SetSharedParameterFlag()1656 auto tensor_layout = parameter_ptr->user_data<TensorLayout>(); in SetSharedParameterFlag()1700 ParameterPtr parameter_ptr = parameter->cast<ParameterPtr>(); in SetParallelShape() local1701 MS_EXCEPTION_IF_NULL(parameter_ptr); in SetParallelShape()1702 parameter_ptr->set_user_data<TensorLayout>(std::make_shared<TensorLayout>(tensor_layout)); in SetParallelShape()