Searched refs:batch_size_for_input_fn (Results 1 – 2 of 2) sorted by relevance
796 batch_size=ctx.batch_size_for_input_fn,868 batch_size=ctx.batch_size_for_input_fn,966 batch_size=ctx.batch_size_for_input_fn,2661 batch_size_for_input_fn = ctx.batch_size_for_input_fn2662 if batch_size_for_input_fn is not None:2664 batch_size_for_input_fn)
469 def batch_size_for_input_fn(self): member in _InternalTPUContext