diff options
| author |  Max Galkin <maxgalkin@google.com> | 2018-05-04 18:17:45 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-05 08:42:06 -0700 |
| commit | 59f0618cedb8fd6ea989471f64ec66619f26b83e (patch) | |
| tree | 61c29be3c388e4ad4720ae4b06707a7c0294873b /tensorflow/core/grappler/costs/op_level_cost_estimator.cc | |
| parent | ad2e7f0503f3c9078124ed74ff5249c35cf7f267 (diff) | |
GuaranteeConst is a NoOp for the op_level_cost_estiamtor.
PiperOrigin-RevId: 195501990
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, 2 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc index b35873ce38..199b69452f 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc @@ -27,6 +27,7 @@ namespace grappler { constexpr int kOpsPerMac = 2; constexpr char kConst[] = "Const"; +constexpr char kGuaranteeConst[] = "GuaranteeConst"; constexpr char kConv2d[] = "Conv2D"; constexpr char kConv2dBackpropFilter[] = "Conv2DBackpropFilter"; constexpr char kConv2dBackpropInput[] = "Conv2DBackpropInput"; @@ -205,6 +206,7 @@ OpLevelCostEstimator::OpLevelCostEstimator() { {kBatchMatMul, wrap(&OpLevelCostEstimator::PredictBatchMatMul)}, {kNoOp, wrap(&OpLevelCostEstimator::PredictNoOp)}, + {kGuaranteeConst, wrap(&OpLevelCostEstimator::PredictNoOp)}, {kGather, wrap(&OpLevelCostEstimator::PredictGatherOrSlice)}, {kGatherV2, wrap(&OpLevelCostEstimator::PredictGatherOrSlice)}, |