blob: 46e5215132f7ce1db9f3e0f3d41fd4bc53479990 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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_
|