From ffe6ede215729f99764761c5acf6a3bdebf69ced Mon Sep 17 00:00:00 2001 From: Akshay Modi Date: Tue, 8 May 2018 17:27:33 -0700 Subject: [PATCH] Include tensorflow::DataType header file PiperOrigin-RevId: 195903041 --- tensorflow/python/eager/BUILD | 1 + tensorflow/python/eager/pywrap_tensor.h | 1 + 2 files changed, 2 insertions(+) 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" -- 2.7.4