[Fix] Prefix bit should be in the upper bit.

tensorflow · Jan 6, 2025 · e75f868 · e75f868
1 parent e1be08e
commit e75f868
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/dynamic_embedding/movielens-1m-keras-with-horovod/movielens-1m-keras-with-horovod.py b/demo/dynamic_embedding/movielens-1m-keras-with-horovod/movielens-1m-keras-with-horovod.py
@@ -419,7 +419,7 @@ def call(self, features):
             boundaries=fea_info['boundaries'])(input_tensor)
       # To prepare for GPU table combined queries, use a prefix to distinguish different features in a table.
       if fea_info['ptype'] == 'user_occupation_label':
-        input_tensor_prefix_code = int(fea_info['code']) << 1
+        input_tensor_prefix_code = int(fea_info['code']) << 48
       else:
         input_tensor_prefix_code = None
       if input_tensor_prefix_code is not None: