diff --git a/tensorflow/core/grappler/costs/cost_estimator.h b/tensorflow/core/grappler/costs/cost_estimator.h index 2efeebb5c53..95c3a84df51 100644 --- a/tensorflow/core/grappler/costs/cost_estimator.h +++ b/tensorflow/core/grappler/costs/cost_estimator.h @@ -17,6 +17,8 @@ limitations under the License. #define TENSORFLOW_CORE_GRAPPLER_COSTS_COST_ESTIMATOR_H_ #include +#include + #include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/protobuf/config.pb.h" diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator_test.cc b/tensorflow/core/grappler/costs/op_level_cost_estimator_test.cc index dfd4784dbf4..aa68f80fca5 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator_test.cc +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator_test.cc @@ -14,6 +14,9 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/core/grappler/costs/op_level_cost_estimator.h" + +#include + #include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/framework/attr_value_util.h" #include "tensorflow/core/framework/tensor.h" diff --git a/tensorflow/core/grappler/costs/virtual_placer.h b/tensorflow/core/grappler/costs/virtual_placer.h index 8761f40f707..8e8866c6d27 100644 --- a/tensorflow/core/grappler/costs/virtual_placer.h +++ b/tensorflow/core/grappler/costs/virtual_placer.h @@ -16,6 +16,8 @@ limitations under the License. #ifndef TENSORFLOW_CORE_GRAPPLER_COSTS_VIRTUAL_PLACER_H_ #define TENSORFLOW_CORE_GRAPPLER_COSTS_VIRTUAL_PLACER_H_ +#include + #include "tensorflow/core/platform/types.h" #include "tensorflow/core/protobuf/device_properties.pb.h" diff --git a/tensorflow/core/grappler/graph_analyzer/gen_node.h b/tensorflow/core/grappler/graph_analyzer/gen_node.h index faec9ecad88..413736f69c8 100644 --- a/tensorflow/core/grappler/graph_analyzer/gen_node.h +++ b/tensorflow/core/grappler/graph_analyzer/gen_node.h @@ -18,6 +18,7 @@ limitations under the License. #include #include +#include #include #include "tensorflow/core/framework/graph.pb.h" diff --git a/tensorflow/core/grappler/graph_analyzer/subgraph.h b/tensorflow/core/grappler/graph_analyzer/subgraph.h index 4de31d5dfa2..7d3494cdc43 100644 --- a/tensorflow/core/grappler/graph_analyzer/subgraph.h +++ b/tensorflow/core/grappler/graph_analyzer/subgraph.h @@ -18,6 +18,7 @@ limitations under the License. #include #include +#include #include "tensorflow/core/grappler/graph_analyzer/gen_node.h" #include "tensorflow/core/grappler/graph_analyzer/map_tools.h" diff --git a/tensorflow/core/grappler/utils/symbolic_shapes.cc b/tensorflow/core/grappler/utils/symbolic_shapes.cc index 1666de4b801..978a0c1554b 100644 --- a/tensorflow/core/grappler/utils/symbolic_shapes.cc +++ b/tensorflow/core/grappler/utils/symbolic_shapes.cc @@ -14,6 +14,9 @@ limitations under the License. ==============================================================================*/ #include "tensorflow/core/grappler/utils/symbolic_shapes.h" + +#include + #include "tensorflow/core/util/bcast.h" namespace tensorflow {