if self.run_eagerly or self._distribution_strategy:
      output_dict = training_eager_v1.train_on_batch(
          self,
          x,
          y,