From 18295c5ad780e40d1b5a99eb7232498558e81cd5 Mon Sep 17 00:00:00 2001 From: Brian Atkinson Date: Tue, 28 Jan 2020 17:12:33 -0800 Subject: [PATCH] Make //tensorflow/c:kernels_test IWYU clean. PiperOrigin-RevId: 292046262 Change-Id: I84717d56226875628c2e8172455cb9492117973e --- tensorflow/c/BUILD | 2 +- tensorflow/c/kernels_test.cc | 19 ++++++++++++++++++- 2 files changed, 19 insertions(+), 2 deletions(-) diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 268ca0086f9..8a7c72d0603 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -640,7 +640,7 @@ tf_cuda_cc_test( "//tensorflow/core:test", "//tensorflow/core:test_main", "//tensorflow/core/kernels:ops_testutil", - "//third_party/eigen3", + "@com_google_absl//absl/container:inlined_vector", ], ) diff --git a/tensorflow/c/kernels_test.cc b/tensorflow/c/kernels_test.cc index 0a363874084..a78521c190b 100644 --- a/tensorflow/c/kernels_test.cc +++ b/tensorflow/c/kernels_test.cc @@ -18,19 +18,36 @@ limitations under the License. #include "tensorflow/c/kernels.h" -#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" +#include +#include +#include + +#include +#include + +#include "absl/container/inlined_vector.h" #include "tensorflow/c/c_api.h" +#include "tensorflow/c/tf_datatype.h" +#include "tensorflow/c/tf_status.h" +#include "tensorflow/c/tf_tensor.h" +#include "tensorflow/core/framework/allocator.h" #include "tensorflow/core/framework/attr_value.pb.h" +#include "tensorflow/core/framework/device_base.h" #include "tensorflow/core/framework/kernel_def.pb.h" #include "tensorflow/core/framework/node_def.pb.h" #include "tensorflow/core/framework/node_def_builder.h" #include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op_kernel.h" +#include "tensorflow/core/framework/tensor.h" +#include "tensorflow/core/framework/tensor_types.h" #include "tensorflow/core/framework/types.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/kernels/ops_testutil.h" #include "tensorflow/core/lib/core/status_test_util.h" +#include "tensorflow/core/platform/env.h" +#include "tensorflow/core/platform/status.h" #include "tensorflow/core/platform/test.h" +#include "tensorflow/core/platform/types.h" struct MyCustomKernel { bool created;