Searched refs:total_step_num (Results 1 – 1 of 1) sorted by relevance
29 CallbackParam(int64_t epoch_num, int64_t cur_epoch_step, int64_t total_step_num) in CallbackParam() argument30 … : cur_epoch_num_(epoch_num), cur_epoch_step_num_(cur_epoch_step), cur_step_num_(total_step_num) {} in CallbackParam()