Train a shard of a batch on a single TPU core. Args: *args: metrics values from previous steps. Returns: New summed metrics values and a train_op.
(self, *args)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected