Enable check_types in nest operations for encoding_network.
Enable check_types in nest operations for encoding_network.

PiperOrigin-RevId: 303418075
Change-Id: Ib8846e76864a394ace89dded1600d364fedc1b70

tensorflow/agents,tensorflow/agents