aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/framework/shape_inference.h
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <bsteiner@google.com>2017-11-06 11:12:11 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-11-06 11:17:02 -0800
commit4a0eb28a01f72c8182cf0895c7817a4e0137f8f7 (patch)
tree77173572d8e7ca8ef4976a2e59e116a952d70e27 /tensorflow/core/framework/shape_inference.h
parent0556834abd8a994012d15ad081a850d24ce8fbdd (diff)
Improved encoding on shapes in grappler.
PiperOrigin-RevId: 174733491
Diffstat (limited to 'tensorflow/core/framework/shape_inference.h')
-rw-r--r--tensorflow/core/framework/shape_inference.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/framework/shape_inference.h b/tensorflow/core/framework/shape_inference.h
index d1b610d682..b68e6100df 100644
--- a/tensorflow/core/framework/shape_inference.h
+++ b/tensorflow/core/framework/shape_inference.h
@@ -71,6 +71,7 @@ class DimensionHandle {
friend class ShapeInferenceTestutil;
friend class ::tensorflow::ShapeRefinerTest;
friend class ShapeManager;
+ friend class ::tensorflow::grappler::GraphProperties;
// Intentionally copyable.
};