Always relax shapes in common.function() by default.
Always relax shapes in common.function() by default.

PiperOrigin-RevId: 249146585
Change-Id: I91472114a603d2af21b7f8d34fa4f14219e47279

tensorflow/agents,tensorflow/agents