diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2017-10-12 13:01:17 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-12 13:04:51 -0700 |
| commit | 4efb8cf64bb2f0f0727430d38ca6e48a99e572f7 (patch) | |
| tree | 7490ff4fc60690bcdaca673cd8a47e338677ec99 /tensorflow/core/grappler/costs/op_level_cost_estimator.cc | |
| parent | deb72df13e588eda07968adc306c1a87416cf7fc (diff) | |
Register a dummy estimation function for _Send ops. This ensures that we don't get negative cost estimates for _Send ops.
PiperOrigin-RevId: 171999586
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 7a1295c91e..a2fa847df2 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc @@ -35,6 +35,7 @@ constexpr char kRefIdentity[] = "RefIdentity"; constexpr char kNoOp[] = "NoOp"; constexpr char kReshape[] = "Reshape"; constexpr char kRecv[] = "_Recv"; +constexpr char kSend[] = "_Send"; constexpr char kBatchMatMul[] = "BatchMatMul"; constexpr char kVariable[] = "Variable"; constexpr char kVariableV2[] = "VariableV2"; @@ -165,6 +166,7 @@ OpLevelCostEstimator::OpLevelCostEstimator() { {kNoOp, wrap(&OpLevelCostEstimator::PredictNoOp)}, {kReshape, wrap(&OpLevelCostEstimator::PredictNoOp)}, {kRecv, wrap(&OpLevelCostEstimator::PredictNoOp)}, + {kSend, wrap(&OpLevelCostEstimator::PredictNoOp)}, {kVariable, wrap(&OpLevelCostEstimator::PredictNoOp)}, {kVariableV2, wrap(&OpLevelCostEstimator::PredictNoOp)}, {kBatchMatMul, wrap(&OpLevelCostEstimator::PredictBatchMatMul)}, |