diff options
author | 2018-02-22 15:34:50 -0800 | |
---|---|---|
committer | 2018-02-22 15:38:52 -0800 | |
commit | 848c53fb11cab2631695cdb6c38bbdfeee972a75 (patch) | |
tree | 9c9ae9d599edaf3c72e8194c24d159ad30920653 /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | 6006f46dd7531b112360b831aa61de6c46618166 (diff) |
Implement the logic to parse TensorProto (the tensor value for input or filter shape info) in op_level_cost_estimator.
PiperOrigin-RevId: 186685409
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h')
-rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.h | 3 |
1 files changed, 3 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 a292e5e97f..7bb530fe31 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -28,6 +28,9 @@ limitations under the License. namespace tensorflow { namespace grappler { +bool GetTensorShapeProtoFromTensorProto(const TensorProto& tensor_proto, + TensorShapeProto* tensor_shape_proto); + class OpLevelCostEstimator { public: OpLevelCostEstimator(); |