diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2018-01-26 14:02:33 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-26 14:06:25 -0800 |
| commit | ea25bf9558a79747d82220e493d4347901853976 (patch) | |
| tree | eda168812562c1233f7e939d113d03bad94fff16 /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
| parent | 1bd5b2e6fada5334b04e2db87cf246d1a83fa533 (diff) | |
Add op level memory usage estimation to the op_level_cost_estimator
PiperOrigin-RevId: 183441321
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 5f541ccf04..a292e5e97f 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -132,6 +132,8 @@ class OpLevelCostEstimator { Costs PredictConv2DBackpropFilter(const OpContext& op_context) const; Costs PredictMatMul(const OpContext& op_context) const; Costs PredictNoOp(const OpContext& op_context) const; + Costs PredictIdentity(const OpContext& op_context) const; + Costs PredictVariable(const OpContext& op_context) const; Costs PredictBatchMatMul(const OpContext& op_context) const; Costs PredictMetadata(const OpContext& op_context) const; |