From ed3745935d2b7bdcb6c2062984e924df305e6417 Mon Sep 17 00:00:00 2001 From: Neural-Link Team Date: Tue, 18 Jun 2024 06:18:47 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 644361070 --- neural_structured_learning/BUILD | 2 +- research/carls/knowledge_bank/initializer_helper.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/neural_structured_learning/BUILD b/neural_structured_learning/BUILD index eda5fd8..23a4fec 100644 --- a/neural_structured_learning/BUILD +++ b/neural_structured_learning/BUILD @@ -15,8 +15,8 @@ # Description: # Build rules for TensorFlow Neural Structured Learning. -# Placeholder for internal Python strict compatibility macro. # Internal annotation for sync +# Placeholder for internal Python strict compatibility macro. load("@rules_license//rules:license.bzl", "license") package( diff --git a/research/carls/knowledge_bank/initializer_helper.cc b/research/carls/knowledge_bank/initializer_helper.cc index 6bab3f2..98d0691 100644 --- a/research/carls/knowledge_bank/initializer_helper.cc +++ b/research/carls/knowledge_bank/initializer_helper.cc @@ -68,7 +68,7 @@ EmbeddingVectorProto InitializeEmbedding( return result; } if (initializer.has_random_uniform_initializer()) { - absl::BitGen bit_gen; + absl::SharedBitGen bit_gen; const auto& init = initializer.random_uniform_initializer(); for (int i = 0; i < embedding_dimension; ++i) { result.add_value(absl::Uniform(bit_gen, init.low(), init.high())); @@ -76,7 +76,7 @@ EmbeddingVectorProto InitializeEmbedding( return result; } if (initializer.has_random_normal_initializer()) { - absl::BitGen bit_gen; + absl::SharedBitGen bit_gen; const auto& init = initializer.random_normal_initializer(); for (int i = 0; i < embedding_dimension; ++i) { result.add_value(