aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/graph/costutil.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/graph/costutil.h')
-rw-r--r--tensorflow/core/graph/costutil.h19
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/core/graph/costutil.h b/tensorflow/core/graph/costutil.h
new file mode 100644
index 0000000000..46e5215132
--- /dev/null
+++ b/tensorflow/core/graph/costutil.h
@@ -0,0 +1,19 @@
+#ifndef TENSORFLOW_GRAPH_COSTUTIL_H_
+#define TENSORFLOW_GRAPH_COSTUTIL_H_
+
+#include <vector>
+#include "tensorflow/core/platform/port.h"
+
+namespace tensorflow {
+
+class CostModel;
+class Graph;
+
+// result[i] is an estimate of the longest execution path from
+// the node with id i to the sink node.
+std::vector<int64> LongestOutgoingPathCost(const Graph& graph,
+ const CostModel& cm);
+
+} // namespace tensorflow
+
+#endif // TENSORFLOW_GRAPH_COSTUTIL_H_