tf.compat.v1.train.generate_checkpoint_state_proto