[feature]split_model_pai.py supports both tf1 and tf2

[feature]split_model_pai.py supports both tf1 and tf2 #832