From 3dbf1d1e99726836d2955b5a9910b42d2b27081d Mon Sep 17 00:00:00 2001 From: JunqiHu Date: Thu, 2 Nov 2023 16:39:44 +0800 Subject: [PATCH] [Embedding] Remove private header. Signed-off-by: JunqiHu --- tensorflow/core/framework/embedding/hbm_dram_ssd_storage.h | 1 - tensorflow/core/framework/embedding/hbm_dram_storage.h | 1 - 2 files changed, 2 deletions(-) diff --git a/tensorflow/core/framework/embedding/hbm_dram_ssd_storage.h b/tensorflow/core/framework/embedding/hbm_dram_ssd_storage.h index 1056f4bbd78..4bc3b7d3aa2 100644 --- a/tensorflow/core/framework/embedding/hbm_dram_ssd_storage.h +++ b/tensorflow/core/framework/embedding/hbm_dram_ssd_storage.h @@ -7,7 +7,6 @@ #include "tensorflow/core/framework/embedding/single_tier_storage.h" #include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h" #include "tensorflow/core/platform/stream_executor.h" -#include "tensorflow/core/util/tensor_bundle/tensor_bundle.h" namespace tensorflow { using se::DeviceMemoryBase; diff --git a/tensorflow/core/framework/embedding/hbm_dram_storage.h b/tensorflow/core/framework/embedding/hbm_dram_storage.h index d058d95f05b..15f6271fb4f 100644 --- a/tensorflow/core/framework/embedding/hbm_dram_storage.h +++ b/tensorflow/core/framework/embedding/hbm_dram_storage.h @@ -22,7 +22,6 @@ limitations under the License. #include "tensorflow/core/framework/embedding/hbm_storage_iterator.h" #include "tensorflow/core/framework/embedding/intra_thread_copy_id_allocator.h" #include "tensorflow/core/platform/stream_executor.h" -#include "tensorflow/core/util/tensor_bundle/tensor_bundle.h" namespace tensorflow { using se::DeviceMemoryBase;