diff options
author | 2017-04-13 15:44:29 -0800 | |
---|---|---|
committer | 2017-04-13 17:04:07 -0700 | |
commit | dd3a6d364a739496b864e61e9a93c21cbc1b5d1d (patch) | |
tree | 6fe163e06959207b586ea6ce6a5277eac90535ae /tensorflow/cc/training/coordinator.h | |
parent | c8399f61ea845b0c440d13407429a92f6a0591e3 (diff) |
Only record the cost graph in the queue runner: this ensures that the memory
usage remains bounded over time.
Change: 153123196
Diffstat (limited to 'tensorflow/cc/training/coordinator.h')
-rw-r--r-- | tensorflow/cc/training/coordinator.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/cc/training/coordinator.h b/tensorflow/cc/training/coordinator.h index 632418c5ca..0e01b19cd9 100644 --- a/tensorflow/cc/training/coordinator.h +++ b/tensorflow/cc/training/coordinator.h @@ -36,8 +36,8 @@ class RunnerInterface { public: virtual ~RunnerInterface() {} virtual Status Join() = 0; - virtual Status ExportRunMetadata(RunMetadata* metadata) const { - return Status(error::INVALID_ARGUMENT, "No RunMetadata to export."); + virtual Status ExportCostGraph(CostGraphDef* cost_graph) const { + return Status(error::INVALID_ARGUMENT, "No cost model to export."); } /// Returns true iff the runner is running, i.e. if it is trying to populate /// its queue. |