search

tensorflow / model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
https://www.tensorflow.org/model_optimization
Apache License 2.0
1.49k stars 319 forks source link

Remove `tf.function` to prevent frequent amounts of retracing. #966

Closed copybara-service[bot] closed 2 years ago

copybara-service[bot] commented 2 years ago

Remove tf.function to prevent frequent amounts of retracing.

Outer level scopes should apply tf.function if desired.