Searched refs:distributed_train_step (Results 1 – 2 of 2) sorted by relevance
159 def distributed_train_step(dist_inputs): function171 accuracy, loss_value = distributed_train_step(next(iterator))
89 def distributed_train_step(self, batch_dataset, model, loss_fn, optimizer, member in CustomMnistBenchmark159 total_loss += self.distributed_train_step(batch_dataset, model,