Add missing includes.
PiperOrigin-RevId: 251258978
This commit is contained in:
parent
58012efe68
commit
a6e1a25eab
@ -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"
|
||||
|
||||
|
@ -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"
|
||||
|
@ -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"
|
||||
|
||||
|
@ -18,6 +18,7 @@ limitations under the License.
|
||||
|
||||
#include <map>
|
||||
#include <memory>
|
||||
#include <unordered_map>
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
|
@ -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"
|
||||
|
@ -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 {
|
||||
|
Loading…
Reference in New Issue
Block a user