[XLA] Use absl::flat_hash_{map,set}::contains() instead of count().

contains() is a much better description of what we're trying to do!

Also change a few std containers over to absl containers so we can advantage of
this.

PiperOrigin-RevId: 227894609