required for sharing the model between training and serving layers.
required for sharing the model between training and serving layers.