tf_agents/bandits/networks/heteroscedastic_q_network.py
tf_agents/networks/q_network.py

tf_agents/bandits/networks/heteroscedastic_q_network.py
tf_agents/networks/value_network.py

tf_agents/bandits/networks/heteroscedastic_q_network.py
tf_agents/networks/actor_distribution_network.py