Adds faster variant of EPR.

PiperOrigin-RevId: 458478936
tensorflow · Jul 1, 2022 · 322587c · 322587c
1 parent 98e2f5e
commit 322587c
Show file tree

Hide file tree

Showing 3 changed files with 280 additions and 125 deletions.
diff --git a/tensorflow_model_optimization/python/core/common/keras/compression/algorithms/BUILD b/tensorflow_model_optimization/python/core/common/keras/compression/algorithms/BUILD
@@ -21,6 +21,7 @@ py_strict_test(
     timeout = "long",
     srcs = ["epr_test.py"],
     python_version = "PY3",
+    shard_count = 4,
     tags = ["requires-net:external"],
     deps = [
         ":epr",