Searched refs:batch_size_for_model_fn (Results 1 – 2 of 2) sorted by relevance
1689 batch_size_for_model_fn = None1691 batch_size_for_model_fn = self._ctx.batch_size_for_model_fn1693 if batch_size_for_model_fn is not None:1694 _add_item_to_params(params, _BATCH_SIZE_KEY, batch_size_for_model_fn)
483 def batch_size_for_model_fn(self): member in _InternalTPUContext