I'm referring to what the authors actually claim they did in the paper. They operated on XLA-generated textual IR.

Cf. the second paragraph of 3.3.4 of https://storage.googleapis.com/deepmind-media/DeepMind.com/B...