Searched refs:set_is_parameter (Results 1 – 6 of 6) sorted by relevance
73 mmcost_.set_is_parameter({false, false}); in TEST_F()127 ac_cost_.set_is_parameter({false, false}); in TEST_F()181 prelu_cost_.set_is_parameter({false, true}); in TEST_F()
75 Status set_is_parameter(const std::vector<bool> &is_parameter);
1511 Status OperatorInfo::set_is_parameter(const std::vector<bool> &is_parameter) { in set_is_parameter() function in mindspore::parallel::OperatorInfo1518 operator_cost()->set_is_parameter(is_parameter); in set_is_parameter()
305 if (operator_info->set_is_parameter(parameter_info) != SUCCESS) { in CreateTheOperatorInfo()793 if (tmp_identity_ptr->set_is_parameter(is_parameter) != SUCCESS) { in AugmentCostGraph()
62 void set_is_parameter(const std::vector<bool> &is_parameter);
26 void OperatorCost::set_is_parameter(const std::vector<bool> &is_parameter) { is_parameter_ = is_par… in set_is_parameter() function in mindspore::parallel::OperatorCost