Searched refs:GetBatchNum (Results 1 – 4 of 4) sorted by relevance
106 MS_LOG(INFO) << "Label number: " << davinci_model->GetBatchNum(); in InitLabel()107 label_list_.resize(davinci_model->GetBatchNum()); in InitLabel()
57 uint32_t GetBatchNum() const { return batch_num_; } in GetBatchNum() function
929 for (size_t calib_index = 0; calib_index < calibrator_->GetBatchNum(); calib_index++) { in DoInference()1032 for (size_t i = 0; i < calibrator_->GetBatchNum(); i++) { in Int8Inference()1055 for (size_t i = 0; i < calibrator_->GetBatchNum(); i++) { in BiasCorrection()1079 if (calibrator_->GetBatchNum() == 0) { in BiasCorrection()1085 [this](float &data) { data = data / calibrator_->GetBatchNum(); }); in BiasCorrection()1195 for (size_t i = 0; i < calibrator_->GetBatchNum(); i++) { in CollectDataFrequency()
175 size_t GetBatchNum() const { return data_pre_process_param_.calibrate_size; } in GetBatchNum() function