Searched refs:current_epoch (Results 1 – 4 of 4) sorted by relevance
327 current_epoch = math.floor(i / step_per_epoch)328 decay_epoch, tmp_epoch = function(decay_epoch, current_epoch)373 current_epoch = math.floor(i / step_per_epoch)374 warmup_epoch, tmp_epoch = function(warmup_epoch, current_epoch)
192 void *current_epoch = currnet_epoch_tensor->device_address()->GetMutablePtr(); in SetOpMappingInfo() local195 if (current_epoch != nullptr && current_step != nullptr && steps_per_epoch != nullptr) { in SetOpMappingInfo()196 dump_info->set_step_id_addr(reinterpret_cast<uint64_t>(current_epoch)); in SetOpMappingInfo()
252 uint32_t current_epoch() const { return current_epoch_; } in current_epoch() function
274 *epoch_val = SizeToInt(graph->current_epoch()); in LoadCtrlInputTensor()287 graph->set_current_epoch(graph->current_epoch() + 1); in LoadCtrlInputTensor()