How to optimize tensorflow model graph as I need to split model weight tensor in each session run?

Question

There is many model weight tensor [512, 512 * 3] saved in checkpoint. In the serving graph, I need to split those tensor into 3 weight tensor [512 * 512] in each session run