Some minor simplifications.
Some minor simplifications.

PiperOrigin-RevId: 420715222
Change-Id: I8fe9c12ae34dc1348ff1a68c917a54d2e7296d9a

deepmind/reverb,deepmind/reverb,deepmind/reverb,deepmind/reverb