aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/framework/memory_types.cc
diff options
context:
space:
mode:
authorGravatar Geoffrey Irving <geoffreyi@google.com>2017-06-23 12:55:53 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-06-23 12:59:28 -0700
commit6ada43366663210beb0159b8c1a67b26ebfe6cb7 (patch)
tree1f41bb5c9d2000cb4dd47645f57c181aef3fae3e /tensorflow/core/framework/memory_types.cc
parent0eff699d3087171cf35671d9d0bd6f8e79441ab3 (diff)
Prepare to not include node_def.proto.h in node_def_util.h
The goal is to make kernels mostly independent of proto headers, which will let us lock down our .so imports. This CL makes a bunch of .cc files either include node_def.proto.h themselves or not need the definition of NodeDef; a second CL will make node_def_util.h not include node_def.proto.h. RELNOTES: n/a PiperOrigin-RevId: 159982117
Diffstat (limited to 'tensorflow/core/framework/memory_types.cc')
-rw-r--r--tensorflow/core/framework/memory_types.cc1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/framework/memory_types.cc b/tensorflow/core/framework/memory_types.cc
index c1dde1504a..6a2eed94b9 100644
--- a/tensorflow/core/framework/memory_types.cc
+++ b/tensorflow/core/framework/memory_types.cc
@@ -18,6 +18,7 @@ limitations under the License.
#include <utility>
#include "tensorflow/core/framework/kernel_def.pb.h"
+#include "tensorflow/core/framework/node_def.pb.h"
#include "tensorflow/core/framework/node_def_util.h"
#include "tensorflow/core/framework/op_kernel.h"
#include "tensorflow/core/framework/types.h"