How to make sure tensorflow2 is using all cpu cores for tf.keras?

Question

My tf.keras model is :

def build_model(
    input_dim:int,
    tag_size:int,
    ):
    """
    """

    model = tf.ker