Searched refs:reg_weight (Results 1 – 3 of 3) sorted by relevance
45 float reg_weight = 1.0 - 1.0 / list_length(&pred->succ_list); in ppir_schedule_calc_sched_info() local46 if (extra_reg > reg_weight) in ppir_schedule_calc_sched_info()47 extra_reg = reg_weight; in ppir_schedule_calc_sched_info()
50 float reg_weight = 1.0f - 1.0f / list_length(&pred->succ_list); in schedule_calc_sched_info() local51 if (extra_reg > reg_weight) in schedule_calc_sched_info()52 extra_reg = reg_weight; in schedule_calc_sched_info()
99 reg_weight = initializer('Normal', shape=[num_classes * 4, self.rcnn_fc_out_channels][::-1],102 self.reg_scores = DenseNoTranpose(self.rcnn_fc_out_channels, num_classes * 4, reg_weight)