diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-22 18:02:04 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-22 18:09:58 -0800 |
commit | 491fb62d90f080d4daf32b5539ec9b4a2de71c6c (patch) | |
tree | e52fb937c5e65f8d57014448b8ea5fefb8658c37 /tensorflow/core/grappler/costs/op_level_cost_estimator.cc | |
parent | 8852be3ed15e11071d6807b61294d36168be693c (diff) |
Add cost estimator tests for the BiasAdd, ReLU, and Conv2D operations.
PiperOrigin-RevId: 186705930
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 983b6891f1..29ef317e46 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc @@ -245,6 +245,8 @@ OpLevelCostEstimator::OpLevelCostEstimator() { {"Add", Eigen::internal::functor_traits< Eigen::internal::scalar_sum_op<float>>::Cost}, {"ApproximateEqual", 1}, + {"BiasAdd", Eigen::internal::functor_traits< + Eigen::internal::scalar_sum_op<float>>::Cost}, {"Div", Eigen::internal::functor_traits< Eigen::internal::scalar_quotient_op<float>>::Cost}, {"Equal", 1}, |