diff options
Diffstat (limited to 'tensorflow/core/graph/graph_partition.h')
| -rw-r--r-- | tensorflow/core/graph/graph_partition.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/graph/graph_partition.h b/tensorflow/core/graph/graph_partition.h index 95f9eae251..4ae0133977 100644 --- a/tensorflow/core/graph/graph_partition.h +++ b/tensorflow/core/graph/graph_partition.h @@ -19,6 +19,7 @@ limitations under the License. #include <functional> #include <string> #include <unordered_map> +#include <vector> #include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/graph/costmodel.h" |