Estimate cost for element wise ops and a minimum compute cost for dum…

…my execution time. PiperOrigin-RevId: 158585464
TeeeJay · Jun 10, 2017 · 19b4ccd · 19b4ccd
1 parent ef8b5fd
commit 19b4ccd
Show file tree

Hide file tree

Showing 6 changed files with 356 additions and 103 deletions.
diff --git a/tensorflow/core/grappler/costs/BUILD b/tensorflow/core/grappler/costs/BUILD
@@ -232,6 +232,7 @@ cc_library(
  ":op_performance_data_cc",
  "//tensorflow/core:framework",
  "//tensorflow/core/grappler/clusters:utils",
+ "//third_party/eigen3",
  ],
 )
 

diff --git a/tensorflow/core/grappler/costs/analytical_cost_estimator_test.cc b/tensorflow/core/grappler/costs/analytical_cost_estimator_test.cc
@@ -102,7 +102,7 @@ TEST_F(AnalyticalCostEstimatorTest, SimpleTest) {
  Costs summary;
  TF_ASSERT_OK(estimator.PredictCosts(item.graph, &cost_graph, &summary));
 
- EXPECT_EQ(Costs::NanoSeconds(9108), summary.execution_time);
+ EXPECT_EQ(Costs::NanoSeconds(9136), summary.execution_time);
  EXPECT_FALSE(summary.inaccurate);
 }