diff options
author | Geoffrey Irving <geoffreyi@google.com> | 2017-06-29 11:44:13 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-29 11:48:10 -0700 |
commit | e85d3df92deb9d717befdf173966a2913ac2aea0 (patch) | |
tree | 1c9e80f85c965e039121d8591daab95c62fe7041 /tensorflow/core/framework/op_kernel.h | |
parent | 9b11f458196f6f0528c9974238497a6c8b6da547 (diff) |
Prepare to remove a bunch of proto.h includes from tensorflow/core headers
The goal is to make kernels mostly independent of proto headers, which will let
us lock down our .so imports. This CL does not remove any actual headers, but
changes a bunch of files so that header removal is possible in a followup CL.
It also marks the headers that will be removed with
// TODO(b/62899350): Remove
RELNOTES: n/a
PiperOrigin-RevId: 160552878
Diffstat (limited to 'tensorflow/core/framework/op_kernel.h')
-rw-r--r-- | tensorflow/core/framework/op_kernel.h | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/tensorflow/core/framework/op_kernel.h b/tensorflow/core/framework/op_kernel.h index fd5c8fdbb8..1b716c5a5a 100644 --- a/tensorflow/core/framework/op_kernel.h +++ b/tensorflow/core/framework/op_kernel.h @@ -24,19 +24,19 @@ limitations under the License. #include "tensorflow/core/framework/cancellation.h" #include "tensorflow/core/framework/control_flow.h" #include "tensorflow/core/framework/device_base.h" -#include "tensorflow/core/framework/function.h" -#include "tensorflow/core/framework/graph.pb.h" -#include "tensorflow/core/framework/kernel_def.pb.h" +#include "tensorflow/core/framework/function.h" // TODO(b/62899350): Remove +#include "tensorflow/core/framework/graph.pb.h" // TODO(b/62899350): Remove +#include "tensorflow/core/framework/kernel_def.pb.h" // TODO(b/62899350): Remove #include "tensorflow/core/framework/kernel_def_builder.h" #include "tensorflow/core/framework/node_def_util.h" -#include "tensorflow/core/framework/op.h" +#include "tensorflow/core/framework/op.h" // TODO(b/62899350): Remove #include "tensorflow/core/framework/rendezvous.h" #include "tensorflow/core/framework/selective_registration.h" #include "tensorflow/core/framework/session_state.h" -#include "tensorflow/core/framework/step_stats.pb.h" +#include "tensorflow/core/framework/step_stats.pb.h" // TODO(b/62899350): Remove #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor_shape.h" -#include "tensorflow/core/framework/tensor_shape.pb.h" +#include "tensorflow/core/framework/tensor_shape.pb.h" // TODO(b/62899350): Remove #include "tensorflow/core/framework/tracking_allocator.h" #include "tensorflow/core/framework/types.h" #include "tensorflow/core/framework/types.pb.h" @@ -65,9 +65,13 @@ class TensorSliceReaderCacheWrapper; } // namespace checkpoint class AsyncOpKernel; +class FunctionCallFrame; +class FunctionLibraryRuntime; class OpKernelConstruction; // declared below class OpKernelContext; // declared below +class OpRegistryInterface; class ResourceMgr; +class ScopedStepContainer; class OpKernel { public: |