diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2018-05-15 18:15:32 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-15 18:18:09 -0700 |
| commit | 5c1441b0743c152ad523a8f04a410f39bbc6698f (patch) | |
| tree | fb007657a95c2e5f3e385136a8f6d1d9c10b37a1 /tensorflow/core/protobuf | |
| parent | 734e996343f01e3be8193e9231c2a36bd3d796e2 (diff) | |
Optimize batch normalization when possible
PiperOrigin-RevId: 196762618
Diffstat (limited to 'tensorflow/core/protobuf')
| -rw-r--r-- | tensorflow/core/protobuf/rewriter_config.proto | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/core/protobuf/rewriter_config.proto b/tensorflow/core/protobuf/rewriter_config.proto index 1f9b0c51c1..ed2ba1feae 100644 --- a/tensorflow/core/protobuf/rewriter_config.proto +++ b/tensorflow/core/protobuf/rewriter_config.proto @@ -47,8 +47,11 @@ message RewriterConfig { // result using constants. Toggle constant_folding = 3; // Shape optimizations (default is OFF) - // Simplify computations made on shapes; + // Simplify computations made on shapes. Toggle shape_optimization = 13; + // Remapping (default is OFF) + // Remap subgraphs onto more efficient implementations. + Toggle remapping = 14; // Arithmetic optimizations (default is ON) // e.g. Simplify arithmetic ops; merge ops with same value (like constants). Toggle arithmetic_optimization = 7; |