experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Another IWYU run over //tensorflow/c:kernels_test_gpu with cuda enabled.

Browse Source 
#ifdef's were hiding code, and that hid the need for includes.

PiperOrigin-RevId: 294335591
Change-Id: I90ecadea8307ea1ed8934382cb70c120da20034e
This commit is contained in:
Brian Atkinson 2020-02-10 16:51:00 -08:00 committed by TensorFlower Gardener
parent eb271644d8
commit 432ef2bee2
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/c/BUILD
View File

@ -647,12 +647,14 @@ tf_cuda_cc_test(
deps = [
":c_api",
":kernels",
"//tensorflow/core:core_cpu",
"//tensorflow/core:framework",
"//tensorflow/core:lib",
"//tensorflow/core:protos_all_cc",
"//tensorflow/core:test",
"//tensorflow/core:test_main",
"//tensorflow/core/kernels:ops_testutil",
"//third_party/eigen3",
"@com_google_absl//absl/container:inlined_vector",
],
)

4
tensorflow/c/kernels_test.cc
View File

@ -24,12 +24,16 @@ limitations under the License.
#include <memory>
#include <string>
#include <utility>
#include "absl/container/inlined_vector.h"
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#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/common_runtime/device.h"
#include "tensorflow/core/common_runtime/device_factory.h"
#include "tensorflow/core/framework/allocator.h"
#include "tensorflow/core/framework/attr_value.pb.h"
#include "tensorflow/core/framework/device_base.h"