Searched refs:_InternalTPUContext (Results 1 – 1 of 1) sorted by relevance
173 class _InternalTPUContext(object): class702 class _OneCoreTPUContext(_InternalTPUContext):747 return _InternalTPUContext(config, train_batch_size, eval_batch_size,