tfp.experimental.nn.util.variables_save

Saves structure of tf.Variables to filename.