Searched refs:get_batch_dimension (Results 1 – 2 of 2) sorted by relevance
503 def get_batch_dimension(iterator): function
633 batch_dimension = distributed_training_utils.get_batch_dimension(iterator)