tfmot.sparsity.keras.compat.assign

View source on GitHub