[Embedding] Fix compile problem.

Signed-off-by: JunqiHu <[email protected]>
DeepRec-AI · Oct 20, 2023 · c93f0ea · c93f0ea
1 parent 88748e5
commit c93f0ea
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/tensorflow/core/framework/embedding/embedding_var_ckpt_data.h b/tensorflow/core/framework/embedding/embedding_var_ckpt_data.h
@@ -20,7 +20,6 @@ limitations under the License.
 namespace tensorflow {
 class BundleWriter;
 namespace {
-  const int kSavedPartitionNum = 1000;
   const int kDramFlagOffset = 49;
 }
 

diff --git a/tensorflow/core/framework/embedding/kv_interface.h b/tensorflow/core/framework/embedding/kv_interface.h
@@ -23,6 +23,7 @@ limitations under the License.
 namespace tensorflow {
 namespace {
 const char* kInferenceMode = "INFERENCE_MODE";
+const int kSavedPartitionNum = 1000;
 }
 
 template <class K, class V>