aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/graph
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-09-13 20:08:15 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-13 20:12:38 -0700
commit4a665550dacdb5e162e71b4afe039de178ffc49f (patch)
treeee3ee08e6beced5bb57075bf25e15ef40d09efdf /tensorflow/core/graph
parent6dd278831a62be829ce6f15039e5b6b368b3727c (diff)
Add missing #include to mkl_layout_pass.cc.
PiperOrigin-RevId: 212921868
Diffstat (limited to 'tensorflow/core/graph')
-rw-r--r--tensorflow/core/graph/mkl_layout_pass.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/graph/mkl_layout_pass.cc b/tensorflow/core/graph/mkl_layout_pass.cc
index 2e644fe987..f5b0105862 100644
--- a/tensorflow/core/graph/mkl_layout_pass.cc
+++ b/tensorflow/core/graph/mkl_layout_pass.cc
@@ -28,6 +28,7 @@ limitations under the License.
#include "tensorflow/core/common_runtime/function.h"
#include "tensorflow/core/common_runtime/optimization_registry.h"
#include "tensorflow/core/framework/node_def_util.h"
+#include "tensorflow/core/framework/tensor.pb.h"
#include "tensorflow/core/graph/algorithm.h"
#include "tensorflow/core/graph/graph.h"
#include "tensorflow/core/graph/node_builder.h"