TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs - Phothilimthana, Mangpo; Abu-El-Haija, Sami; Cao, Kaidi; Fatemi, Bahare; Burrows, Michael; Mendis, Charith; Perozzi, Bryan (Advances in Neural Information Processing Systems, 2023)