TPU-Blocker: custom call memory operations in f64 being rewritten

[wsmoses]

While rewriting computation to not contain X{64|128} element types, XLA encountered an HLO for which this rewriting is not implemented: %custom-call.272 = f64[1520,3056]{1,0} custom-call(%add.2772),