Home
last modified time | relevance | path

Searched refs:GetBatchNum (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/
Druntime_model.cc106 MS_LOG(INFO) << "Label number: " << davinci_model->GetBatchNum(); in InitLabel()
107 label_list_.resize(davinci_model->GetBatchNum()); in InitLabel()
Ddavinci_model.h57 uint32_t GetBatchNum() const { return batch_num_; } in GetBatchNum() function
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dfull_quant_quantizer.cc929 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()
Dfull_quant_quantizer.h175 size_t GetBatchNum() const { return data_pre_process_param_.calibrate_size; } in GetBatchNum() function