Searched refs:compute_batch_size (Results 1 – 3 of 3) sorted by relevance
581 batch_size = distribute.compute_batch_size(dataset)590 batch_size = distribute.compute_batch_size(dataset)596 batch_size = distribute.compute_batch_size(dataset)603 batch_size = distribute.compute_batch_size(dataset)610 batch_size = distribute.compute_batch_size(dataset)617 batch_size = distribute.compute_batch_size(dataset)624 batch_size = distribute.compute_batch_size(dataset)
477 def compute_batch_size(dataset): function524 return ged_ops.compute_batch_size(dataset._variant_tensor) # pylint: disable=protected-access
1059 batch_size = distribute.compute_batch_size(dataset)