import tensorflow as tf

                name="divisbility_condition"
            )

            less_than_condition = tf.less(
                x=cur_batch_size, y=group_size, name="less_than_condition"
            )
