Searched refs:weight_name (Results 1 – 9 of 9) sorted by relevance
65 const std::string &weight_name = weight.first; in StoreModelByIterNum() local66 if (stored_model.count(weight_name) == 0) { in StoreModelByIterNum()67 MS_LOG(ERROR) << "The stored model has no weight " << weight_name; in StoreModelByIterNum()71 MS_ERROR_IF_NULL_WO_RET_VAL(stored_model[weight_name]); in StoreModelByIterNum()72 MS_ERROR_IF_NULL_WO_RET_VAL(stored_model[weight_name]->addr); in StoreModelByIterNum()75 void *dst_addr = stored_model[weight_name]->addr; in StoreModelByIterNum()76 size_t dst_size = stored_model[weight_name]->size; in StoreModelByIterNum()125 const std::string weight_name = weight.first; in AssignNewModelMemory() local139 memory_register->RegisterArray(weight_name, &weight_data, weight_size); in AssignNewModelMemory()
111 const std::string &weight_name = weight_full_names_[i]; in Launch() local112 if (feature_map.count(weight_name) == 0) { in Launch()113 MS_LOG(EXCEPTION) << "The weights for " << weight_name << " is not pulled from server."; in Launch()116 …memcpy_s(inputs[i]->addr, inputs[i]->size, feature_map[weight_name].addr, feature_map[weight_name]… in Launch()162 for (const std::string &weight_name : weight_full_names_) { in BuildPullWeightReq() local163 auto fbs_weight_name = fbb->CreateString(weight_name); in BuildPullWeightReq()
104 std::string weight_name = weight_node->fullname_with_scope(); in Init() local105 MS_LOG(INFO) << "Parameter name is " << weight_name; in Init()106 weight_full_names_.push_back(weight_name); in Init()128 const std::string &weight_name = weight_full_names_[i]; in BuildUpdateModelReq() local129 auto fbs_weight_fullname = fbb->CreateString(weight_name); in BuildUpdateModelReq()
121 std::string weight_name = weight_node->fullname_with_scope(); in Init() local122 MS_LOG(INFO) << "Parameter name is " << weight_name; in Init()123 weight_name_to_input_idx_.insert(std::make_pair(weight_name, i)); in Init()
149 const std::string &weight_name = weight_full_names_[i]; in BuildPushWeightReq() local150 auto fbs_weight_fullname = fbb->CreateString(weight_name); in BuildPushWeightReq()
154 const std::string &weight_name = GetValue<std::string>(weight_name_value_node->value()); in DoFusion() local155 weight_names.push_back(weight_name); in DoFusion()
64 std::string weight_name() { return weight_name_; } in weight_name() function
526 std::string weight_name; in GetWeightInfo() local530 weight_name = weight_node->cast<CNodePtr>()->input(1)->cast<ParameterPtr>()->name(); in GetWeightInfo()534 weight_name = para->name(); in GetWeightInfo()558 (*fake_quant_table)[weight_name] = std::make_pair(nullptr, "input"); in GetWeightInfo()588 (*fake_quant_table)[weight_name] = std::make_pair(quant_op->adapter(), fakequant_min_node_name); in GetWeightInfo()
124 … const std::string &instance_name, const std::string &weight_name) { in CreateMirrorInput() argument146 if (param_ptr->name().find(weight_name) != std::string::npos && in CreateMirrorInput()161 …XCEPTION) << "You should define `accu_grads` when use " << op_name << " parameter:" << weight_name; in CreateMirrorInput()