tfm.utils.serialize_layer