tf.contrib.model_pruning.get_weights

View source on GitHub