_noinline is not longer needed in _specialize_per_device
_noinline is not longer needed in _specialize_per_device

The corresponding Grappler bug has been fixed.

PiperOrigin-RevId: 259504851
Change-Id: I1f46229aed69a10d0c9d6c743a550359639fc3ad

deepmind/sonnet,deepmind/sonnet