def METHOD_NAME(images):
    num_batches = tf.shape(images)[0]
    num_non_zero_batches = tf.math.count_nonzero(
        tf.math.count_nonzero(images, axis=[1, 2, 3]), dtype=tf.int32
    )
    return num_batches - num_non_zero_batches