use param_key to identify layers with shared parameters