diff options
author | 2017-07-27 14:27:32 -0700 | |
---|---|---|
committer | 2017-07-27 14:31:43 -0700 | |
commit | 22651083406ca01ac9d481e3367a3510d25f88cd (patch) | |
tree | bda0f3289d50f383eb1e632a595ab97258e35162 /tensorflow/c/BUILD | |
parent | 613bf1c7c1f8dfceed34fc85f2c71dd00432651e (diff) |
C API: Groundwork for experimenting with TF_Tensor in device memory.
TF_Tensor objects are always backed by host memory. This commit lays
the groundwork for allowing TF_Tensor objects to refer to tensor data
on device (e.g., GPU) memory.
PiperOrigin-RevId: 163388079
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 507b2fe1f1..21b09865db 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -61,6 +61,7 @@ tf_cuda_library( "//tensorflow/cc:grad_ops", "//tensorflow/cc:scope_internal", "//tensorflow/core:core_cpu", + "//tensorflow/core:core_cpu_internal", "//tensorflow/core:framework", "//tensorflow/core:protos_all_cc", "//tensorflow/core:lib", |