From 268c8f477ecd664d351194dc505bff0a9ca0d64f Mon Sep 17 00:00:00 2001 From: Oleg Dzhimiev Date: Fri, 3 Apr 2020 11:49:03 -0600 Subject: [PATCH] Added headers from common_runtime/gpu/* --- .../bytedeco/tensorflow/presets/tensorflow.java | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tensorflow/src/main/java/org/bytedeco/tensorflow/presets/tensorflow.java b/tensorflow/src/main/java/org/bytedeco/tensorflow/presets/tensorflow.java index 6c35769ee50..b2811042acb 100644 --- a/tensorflow/src/main/java/org/bytedeco/tensorflow/presets/tensorflow.java +++ b/tensorflow/src/main/java/org/bytedeco/tensorflow/presets/tensorflow.java @@ -226,6 +226,21 @@ "tensorflow/core/common_runtime/eager/eager_operation.h", "tensorflow/core/common_runtime/eager/kernel_and_device.h", "tensorflow/core/common_runtime/eager/tensor_handle.h", + "tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h", + "tensorflow/core/common_runtime/gpu/gpu_cudamalloc_allocator.h", + "tensorflow/core/common_runtime/gpu/gpu_debug_allocator.h", + "tensorflow/core/common_runtime/gpu/gpu_device.h", + "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h", + "tensorflow/core/common_runtime/gpu/gpu_host_allocator.h", + "tensorflow/core/common_runtime/gpu/gpu_id.h", + "tensorflow/core/common_runtime/gpu/gpu_id_manager.h", + "tensorflow/core/common_runtime/gpu/gpu_id_utils.h", + "tensorflow/core/common_runtime/gpu/gpu_init.h", + "tensorflow/core/common_runtime/gpu/gpu_managed_allocator.h", + "tensorflow/core/common_runtime/gpu/gpu_mem_allocator.h", + "tensorflow/core/common_runtime/gpu/gpu_process_state.h", + "tensorflow/core/common_runtime/gpu/gpu_stream_util.h", + "tensorflow/core/common_runtime/gpu/gpu_util.h", "tensorflow/c/eager/c_api.h", // "tensorflow/c/eager/c_api_experimental.h", "tensorflow/c/eager/c_api_internal.h",