From 22651083406ca01ac9d481e3367a3510d25f88cd Mon Sep 17 00:00:00 2001 From: Asim Shankar Date: Thu, 27 Jul 2017 14:27:32 -0700 Subject: 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 --- tensorflow/c/BUILD | 1 + 1 file changed, 1 insertion(+) (limited to 'tensorflow/c/BUILD') 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", -- cgit v1.2.3