diff options
author | Benoit Steiner <bsteiner@google.com> | 2017-07-26 12:24:33 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-26 12:28:54 -0700 |
commit | 51cf7d0954c15e628c35fa3cca6e2ba7587f72ea (patch) | |
tree | 30a855f9a13f41dcd4c8321f47331801f0022653 /tensorflow/core/protobuf | |
parent | caca1c52e6c42e1b58ffee79e10fa4fde6f287ad (diff) |
Updated the memory optimization config to introduce an explicit default value. This will make it possible change change the default behavior in the future by updating the meta optimizer code to interpret that default value differently (e.g we could assume default means heuristics). The default value remains OFF.
PiperOrigin-RevId: 163239483
Diffstat (limited to 'tensorflow/core/protobuf')
-rw-r--r-- | tensorflow/core/protobuf/rewriter_config.proto | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tensorflow/core/protobuf/rewriter_config.proto b/tensorflow/core/protobuf/rewriter_config.proto index a9a1993bdf..1f9e2472a4 100644 --- a/tensorflow/core/protobuf/rewriter_config.proto +++ b/tensorflow/core/protobuf/rewriter_config.proto @@ -34,13 +34,15 @@ message RewriterConfig { bool disable_model_pruning = 2; enum MemOptType { + // The default setting (currently disabled) + DEFAULT_MEM_OPT = 0; // Disabled in the meta-optimizer. - NO_MEM_OPT = 0; + NO_MEM_OPT = 1; // Driven by manual op-level annotations. - MANUAL = 1; + MANUAL = 2; // Driven by heuristics. The behavior of these heuristics is subject to // change. Currently includes an experimental recomputation heuristic. - HEURISTICS = 2; + HEURISTICS = 3; } // Configures memory optimization passes through the meta-optimizer. Has no // effect on manually requested memory optimization passes in the optimizers |