Skip to content

Commit 428ee71

Browse files
Neural-Link Teamtensorflow-copybara
authored andcommitted
Add an async_node_hash_map for asynchronous hash map update.
PiperOrigin-RevId: 368262096
1 parent bcacce4 commit 428ee71

File tree

7 files changed

+749
-0
lines changed

7 files changed

+749
-0
lines changed

WORKSPACE

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,10 @@ workspace(name = "org_tensorflow_neural_structured_learning")
22

33
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
44

5+
# farmhash
6+
load("//research/carls/third_party/farmhash:workspace.bzl", farmhash = "repo")
7+
farmhash()
8+
59
# absl
610
http_archive(
711
name = "com_google_absl",

research/carls/base/BUILD

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,3 +160,28 @@ cc_test(
160160
"@tensorflow_solib//:framework_lib",
161161
],
162162
)
163+
164+
cc_library(
165+
name = "async_node_hash_map",
166+
hdrs = ["async_node_hash_map.h"],
167+
deps = [
168+
"@com_google_absl//absl/container:flat_hash_map",
169+
"@com_google_absl//absl/container:node_hash_map",
170+
"@com_google_absl//absl/container:raw_hash_set",
171+
"@com_google_absl//absl/synchronization",
172+
"@tensorflow_includes//:includes",
173+
"@tensorflow_solib//:framework_lib",
174+
],
175+
)
176+
177+
cc_test(
178+
name = "async_node_hash_map_test",
179+
srcs = ["async_node_hash_map_test.cc"],
180+
deps = [
181+
":async_node_hash_map",
182+
"@com_google_absl//absl/strings",
183+
"@com_google_googletest//:gtest",
184+
"@com_google_googletest//:gtest_main",
185+
"@farmhash_archive//:farmhash",
186+
],
187+
)

0 commit comments

Comments
 (0)