Move TFMOT Library Code Off Private APIs

[alanchiao]

TFMOT currently has dependencies on TF and tf.keras private APIs. This is a focused effort to completely move off of them. We don't want TFMOT releases to stop working with newer TF releases.

Most recently, tfmot.sparsity.keras.PruningSummaries has stopped working with tf-nightly because of a dependence on a private _log_metrics API. If we can move this to use the previously available tf.summary public API, then our next release will work with all > 1.14.0 releases.

If we don't address this now, it'll eventually start to slow down releases as we may want to delay them to ensure all the techniques work with the next TF release(s).

For pruning and quantization, I've previously removed all (or nearly all) private APIs we use that we know how to fix without guidance from TF / tf.keras. Below, I'll put together a list of the ones that remain in order of priority and tackle them one-by-one.

[alanchiao]

FYI @akarmi, @arovir01 for weight clustering, in combination with #172

[akarmi]

FYI @akarmi, @arovir01 for weight clustering, in combination with #172

Thanks. We will check clustering for such issues.