def _test_step(self, inputs): return strategy_run(self._strategy, self.test_batch, (inputs, ))
def _simple_eval_step(self, batch): return local_results( strategy_run(self._strategy, self.simple_eval_batch, (batch, )), self._strategy)
def _train_step_on_batches(self, batches): strategy_run(self._strategy, self.train_batches, batches)
def _eval_step(self, batch): strategy_run(self._strategy, self.eval_batch, (batch, ))
def _train_step(self, batch): strategy_run(self._strategy, self.train_batch, (batch, ))