A Learned Performance Model for Tensor Processing Units