Contributed Talk 2: A Learned Cost Model for the TPU