diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2017-11-06 11:12:11 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-06 11:17:02 -0800 |
| commit | 4a0eb28a01f72c8182cf0895c7817a4e0137f8f7 (patch) | |
| tree | 77173572d8e7ca8ef4976a2e59e116a952d70e27 /tensorflow/core/grappler/costs/op_level_cost_estimator.cc | |
| parent | 0556834abd8a994012d15ad081a850d24ce8fbdd (diff) | |
Improved encoding on shapes in grappler.
PiperOrigin-RevId: 174733491
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.cc')
| -rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc index a2fa847df2..bd84331b67 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc @@ -98,7 +98,7 @@ TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, } } else { for (int i = 0; i < shape.dim_size(); i++) { - if (shape.dim(i).size() == -1) { + if (shape.dim(i).size() < 0) { *found_unknown_shapes = true; VLOG(2) << "Use minimum dim size 1 because the shape is unknown."; // The size of each dimension is at least 1, if unknown. |