diff options
author | 2016-10-26 07:41:34 -0800 | |
---|---|---|
committer | 2016-10-26 08:47:43 -0700 | |
commit | 1439238b82f370c3096aaf0b45eb23940eb66671 (patch) | |
tree | 93920a792d465566461ded694804fcfe16805536 /tensorflow/core/common_runtime | |
parent | e234faa089bcac15be8f9f0cfddeb9f24892d9c1 (diff) |
Cleaned up the cost model manager collection code.
Change: 137279411
Diffstat (limited to 'tensorflow/core/common_runtime')
-rw-r--r-- | tensorflow/core/common_runtime/simple_graph_execution_state.cc | 10 | ||||
-rw-r--r-- | tensorflow/core/common_runtime/simple_graph_execution_state.h | 16 |
2 files changed, 0 insertions, 26 deletions
diff --git a/tensorflow/core/common_runtime/simple_graph_execution_state.cc b/tensorflow/core/common_runtime/simple_graph_execution_state.cc index ff00ad5cfd..82d36b51b5 100644 --- a/tensorflow/core/common_runtime/simple_graph_execution_state.cc +++ b/tensorflow/core/common_runtime/simple_graph_execution_state.cc @@ -274,16 +274,6 @@ Status SimpleGraphExecutionState::InitBaseGraph( return Status::OK(); } -void SimpleGraphExecutionState::UpdateCostsFromStats(const StepStats& ss) { - mutex_lock l(mu_); - costs_.MergeFromStats(node_name_to_cost_id_map_, ss); -} - -void SimpleGraphExecutionState::MergeCostsFromGlobal(CostModel* costs) { - mutex_lock l(mu_); - costs->MergeFromGlobal(costs_); -} - Status SimpleGraphExecutionState::BuildGraph( const BuildGraphOptions& options, std::unique_ptr<SimpleClientGraph>* out) { VLOG(1) << "BuildGraph"; diff --git a/tensorflow/core/common_runtime/simple_graph_execution_state.h b/tensorflow/core/common_runtime/simple_graph_execution_state.h index 2a33d9e298..3b6ce23c75 100644 --- a/tensorflow/core/common_runtime/simple_graph_execution_state.h +++ b/tensorflow/core/common_runtime/simple_graph_execution_state.h @@ -133,22 +133,6 @@ class SimpleGraphExecutionState { Status BuildGraph(const BuildGraphOptions& options, std::unique_ptr<SimpleClientGraph>* out); - // Sums execution statistics in "ss" into the CostModel. - void UpdateCostsFromStats(const StepStats& ss); - - Microseconds TimeEstimate(const Node* n) { - mutex_lock l(mu_); // could use reader lock - return costs_.TimeEstimate(n); - } - - Bytes SizeEstimate(const Node* n, int output_slot) { - mutex_lock l(mu_); // could use reader lock - return costs_.SizeEstimate(n, output_slot); - } - - // Merge the cost model maintained by this graph_execution_state to 'costs'. - void MergeCostsFromGlobal(CostModel* costs); - // The graph returned by BuildGraph may contain only the pruned // graph, whereas some clients may want access to the full graph. const Graph* full_graph() { |