From: Akshay Modi <nareshmodi@google.com>
Date: Wed, 9 May 2018 00:27:33 +0000 (-0700)
Subject: Include tensorflow::DataType header file
X-Git-Tag: upstream/v1.9.0_rc1~116^2^2~219
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=ffe6ede215729f99764761c5acf6a3bdebf69ced;p=platform%2Fupstream%2Ftensorflow.git

Include tensorflow::DataType header file

PiperOrigin-RevId: 195903041
---

diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD
index b3268c9..a0fc538 100644
--- a/tensorflow/python/eager/BUILD
+++ b/tensorflow/python/eager/BUILD
@@ -25,6 +25,7 @@ cc_library(
         "//tensorflow/c/eager:c_api_internal",
         "//tensorflow/c/eager:tape",
         "//tensorflow/core:lib",
+        "//tensorflow/core:protos_all_cc",
         "//tensorflow/python:ndarray_tensor",
         "//tensorflow/python:ndarray_tensor_bridge",
         "//tensorflow/python:numpy_lib",
diff --git a/tensorflow/python/eager/pywrap_tensor.h b/tensorflow/python/eager/pywrap_tensor.h
index 88982b0..bc042eb 100644
--- a/tensorflow/python/eager/pywrap_tensor.h
+++ b/tensorflow/python/eager/pywrap_tensor.h
@@ -16,6 +16,7 @@ limitations under the License.
 #define TENSORFLOW_PYTHON_EAGER_PYWRAP_TENSOR_H_
 
 #include "tensorflow/c/eager/c_api.h"
+#include "tensorflow/core/framework/types.pb.h"
 #include "tensorflow/core/platform/types.h"
 #include "tensorflow/python/lib/core/numpy.h"