load_from_hdfs adopts zero-copy method

tensorflow · Jul 14, 2022 · 4579ee6 · 4579ee6
1 parent bdd5da4
commit 4579ee6
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/...flow_recommenders_addons/dynamic_embedding/core/kernels/lookup_impl/lookup_table_op_cpu.h b/...flow_recommenders_addons/dynamic_embedding/core/kernels/lookup_impl/lookup_table_op_cpu.h
@@ -311,14 +311,10 @@ class TableWrapperOptimized final : public TableWrapperBase<K, V> {
 
     while (i < file_size) {
       TF_RETURN_IF_ERROR(reader.ReadNBytes(record_len, &content));
-      K k;
-      std::memcpy(&k, reinterpret_cast<uint8*>(content.data()), sizeof(K));
-
-      ValueType value_vec;
-      std::memcpy(value_vec.data(),
-                  reinterpret_cast<uint8*>(content.data()) + sizeof(K),
-                  value_len);
-      table_->insert_or_assign(k, value_vec);
+      K* k = reinterpret_cast<K*>(content.data());
+      ValueType* value_vec =
+          reinterpret_cast<ValueType*>(content.data() + sizeof(K));
+      table_->insert_or_assign(*k, *value_vec);
       i += record_len;
     }
     return Status::OK();