Pre-load vectors for Tensorflow ranking classifier pipeline

I was looking this exact thing. Has someone figured out a standard way of doing this that works? It would be extremely helpful to us since our data set is quite small