diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-11 15:07:48 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-11 15:19:32 -0700 |
commit | 06ff12d06e85888701a2dba441e982e34a7db6ec (patch) | |
tree | a35e5efcb96e37c10b37dd8f74e1dda61a3566e3 /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | 640e0baf6e69b037ecc8c3044a11441f18afd180 (diff) |
Expose MaybeGetMinimumShape for use in cost estimators other than OpLevelCostEstimator.
PiperOrigin-RevId: 196315239
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h')
-rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.h b/tensorflow/core/grappler/costs/op_level_cost_estimator.h index 35649f7ee9..d384f57279 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -30,6 +30,8 @@ namespace grappler { bool GetTensorShapeProtoFromTensorProto(const TensorProto& tensor_proto, TensorShapeProto* tensor_shape_proto); +TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, + int rank, bool* found_unknown_shapes); class OpLevelCostEstimator { public: |