Searched refs:record_batch_end (Results 1 – 3 of 3) sorted by relevance
25 self.record_batch_end(self.epoch_key(epoch, index))
66 def record_batch_end(self, key): member in TrainerBase
112 def record_batch_end(self, key): member in ParameterServerBase283 self.record_batch_end(self.param_key(param_loc))