aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/protobuf
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <bsteiner@google.com>2018-05-10 14:22:51 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-05-10 14:25:29 -0700
commit7a493376873e6c21a3fd8d0e04fa51057afaf7a8 (patch)
treec940823db91840837f46e1a4fb6aa88cc43a7226 /tensorflow/core/protobuf
parent48e436c091bad11a9a146a280a1cefbeff3ffc8e (diff)
Started work on a shape optimizer
PiperOrigin-RevId: 196170800
Diffstat (limited to 'tensorflow/core/protobuf')
-rw-r--r--tensorflow/core/protobuf/rewriter_config.proto3
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/protobuf/rewriter_config.proto b/tensorflow/core/protobuf/rewriter_config.proto
index 029b27cd04..1f9b0c51c1 100644
--- a/tensorflow/core/protobuf/rewriter_config.proto
+++ b/tensorflow/core/protobuf/rewriter_config.proto
@@ -46,6 +46,9 @@ message RewriterConfig {
// Statically infer the value of tensors when possible, and materialize the
// result using constants.
Toggle constant_folding = 3;
+ // Shape optimizations (default is OFF)
+ // Simplify computations made on shapes;
+ Toggle shape_optimization = 13;
// Arithmetic optimizations (default is ON)
// e.g. Simplify arithmetic ops; merge ops with same value (like constants).
Toggle arithmetic_optimization = 7;