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_
#include <cmath>
#include <unordered_map>
#include "tensorflow/core/lib/core/status.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 <unordered_set>
#include "tensorflow/core/framework/attr_value.pb.h"
#include "tensorflow/core/framework/attr_value_util.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_
#define TENSORFLOW_CORE_GRAPPLER_COSTS_VIRTUAL_PLACER_H_
#include <unordered_map>
#include "tensorflow/core/platform/types.h"
#include "tensorflow/core/protobuf/device_properties.pb.h"

View File

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

View File

@ -18,6 +18,7 @@ limitations under the License.
#include <initializer_list>
#include <set>
#include <unordered_set>
#include "tensorflow/core/grappler/graph_analyzer/gen_node.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 <unordered_map>
#include "tensorflow/core/util/bcast.h"
namespace tensorflow {