Add missing includes.

PiperOrigin-RevId: 251258978
This commit is contained in:
A. Unique TensorFlower 2019-06-03 10:27:19 -07:00 committed by TensorFlower Gardener
parent 58012efe68
commit a6e1a25eab
6 changed files with 12 additions and 0 deletions

View File

@ -17,6 +17,8 @@ limitations under the License.
#define TENSORFLOW_CORE_GRAPPLER_COSTS_COST_ESTIMATOR_H_ #define TENSORFLOW_CORE_GRAPPLER_COSTS_COST_ESTIMATOR_H_
#include <cmath> #include <cmath>
#include <unordered_map>
#include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/lib/core/status.h"
#include "tensorflow/core/protobuf/config.pb.h" #include "tensorflow/core/protobuf/config.pb.h"

View File

@ -14,6 +14,9 @@ limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "tensorflow/core/grappler/costs/op_level_cost_estimator.h" #include "tensorflow/core/grappler/costs/op_level_cost_estimator.h"
#include <unordered_set>
#include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/framework/attr_value.pb.h"
#include "tensorflow/core/framework/attr_value_util.h" #include "tensorflow/core/framework/attr_value_util.h"
#include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor.h"

View File

@ -16,6 +16,8 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_GRAPPLER_COSTS_VIRTUAL_PLACER_H_ #ifndef TENSORFLOW_CORE_GRAPPLER_COSTS_VIRTUAL_PLACER_H_
#define TENSORFLOW_CORE_GRAPPLER_COSTS_VIRTUAL_PLACER_H_ #define TENSORFLOW_CORE_GRAPPLER_COSTS_VIRTUAL_PLACER_H_
#include <unordered_map>
#include "tensorflow/core/platform/types.h" #include "tensorflow/core/platform/types.h"
#include "tensorflow/core/protobuf/device_properties.pb.h" #include "tensorflow/core/protobuf/device_properties.pb.h"

View File

@ -18,6 +18,7 @@ limitations under the License.
#include <map> #include <map>
#include <memory> #include <memory>
#include <unordered_map>
#include <vector> #include <vector>
#include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/framework/graph.pb.h"

View File

@ -18,6 +18,7 @@ limitations under the License.
#include <initializer_list> #include <initializer_list>
#include <set> #include <set>
#include <unordered_set>
#include "tensorflow/core/grappler/graph_analyzer/gen_node.h" #include "tensorflow/core/grappler/graph_analyzer/gen_node.h"
#include "tensorflow/core/grappler/graph_analyzer/map_tools.h" #include "tensorflow/core/grappler/graph_analyzer/map_tools.h"

View File

@ -14,6 +14,9 @@ limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "tensorflow/core/grappler/utils/symbolic_shapes.h" #include "tensorflow/core/grappler/utils/symbolic_shapes.h"
#include <unordered_map>
#include "tensorflow/core/util/bcast.h" #include "tensorflow/core/util/bcast.h"
namespace tensorflow { namespace tensorflow {