Skip to content

Commit 322587c

Browse files
Johannes Ballétensorflower-gardener
authored andcommitted
Adds faster variant of EPR.
PiperOrigin-RevId: 458478936
1 parent 98e2f5e commit 322587c

File tree

3 files changed

+280
-125
lines changed

3 files changed

+280
-125
lines changed

tensorflow_model_optimization/python/core/common/keras/compression/algorithms/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ py_strict_test(
2121
timeout = "long",
2222
srcs = ["epr_test.py"],
2323
python_version = "PY3",
24+
shard_count = 4,
2425
tags = ["requires-net:external"],
2526
deps = [
2627
":epr",

0 commit comments

Comments
 (0)