Searched refs:_OneCoreTPUContext (Results 1 – 1 of 1) sorted by relevance
702 class _OneCoreTPUContext(_InternalTPUContext): class708 super(_OneCoreTPUContext, self).__init__(744 return _OneCoreTPUContext(config, train_batch_size, eval_batch_size,