Searched refs:all_ep_nums_ (Results 1 – 1 of 1) sorted by relevance
51 all_ep_nums_.reserve(32); in TestCallback()57 all_ep_nums_.push_back(cb_param.cur_epoch_num_); in DSBegin()63 all_ep_nums_.push_back(cb_param.cur_epoch_num_); in DSEpochBegin()69 all_ep_nums_.push_back(cb_param.cur_epoch_num_); in DSNStepBegin()75 all_ep_nums_.push_back(cb_param.cur_epoch_num_); in DSEnd()81 all_ep_nums_.push_back(cb_param.cur_epoch_num_); in DSEpochEnd()87 all_ep_nums_.push_back(cb_param.cur_epoch_num_); in DSNStepEnd()107 return std::vector<int64_t>(all_ep_nums_.begin(), all_ep_nums_.begin() + len); in all_ep_nums()114 std::vector<int64_t> all_step_nums_, all_ep_nums_; member in mindspore::dataset::test::TestCallback