No public description

Tested: PiperOrigin-RevId: 659827895 Change-Id: I96460aacf7dfda14ea6a40865078c10a66766c51
google-deepmind · Aug 6, 2024 · 12cee6a · 12cee6a
1 parent 6a0dcf4
commit 12cee6a
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 13 deletions.
diff --git a/reverb/cc/ops/client.cc b/reverb/cc/ops/client.cc
@@ -67,7 +67,7 @@ class ClientHandleOp : public tensorflow::ResourceOpKernel<ClientResource> {
  private:
   tensorflow::Status CreateResource(ClientResource** ret) override {
     *ret = new ClientResource(server_address_);
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   std::string server_address_;

diff --git a/reverb/cc/ops/pattern_dataset.cc b/reverb/cc/ops/pattern_dataset.cc
@@ -159,7 +159,7 @@ class ReverbPatternDatasetOp : public tensorflow::data::UnaryDatasetOpKernel {
         std::vector<const tensorflow::data::DatasetBase*>* inputs)
         const override {
       inputs->push_back(input_);
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
     tensorflow::Status CheckExternalState() const override {
@@ -223,7 +223,7 @@ class ReverbPatternDatasetOp : public tensorflow::data::UnaryDatasetOpKernel {
                         },
                         output));
 
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
    private:
@@ -298,7 +298,7 @@ class ReverbPatternDatasetOp : public tensorflow::data::UnaryDatasetOpKernel {
           *out_tensors = data_.front();
           data_.pop_front();
         }
-        return tensorflow::OkStatus();
+        return absl::OkStatus();
       }
 
      protected:

diff --git a/reverb/cc/ops/timestep_dataset.cc b/reverb/cc/ops/timestep_dataset.cc
@@ -120,7 +120,7 @@ class ReverbTimestepDatasetOp : public tensorflow::data::DatasetOpKernel {
     tensorflow::Status InputDatasets(
         std::vector<const DatasetBase*>* inputs) const override {
       inputs->clear();
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
    protected:
@@ -170,7 +170,7 @@ class ReverbTimestepDatasetOp : public tensorflow::data::DatasetOpKernel {
           },
           output));
 
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
    private:
@@ -248,17 +248,17 @@ class ReverbTimestepDatasetOp : public tensorflow::data::DatasetOpKernel {
           *end_of_sequence = false;
           *out_tensors = Sampler::WithInfoTensors(*info, std::move(data));
           rate_limited_ = info->rate_limited();
-          return tensorflow::OkStatus();
+          return absl::OkStatus();
         } else if (sampler_options_.rate_limiter_timeout <
                        absl::InfiniteDuration() &&
                    errors::IsRateLimiterTimeout(status)) {
           *end_of_sequence = true;
-          return tensorflow::OkStatus();
+          return absl::OkStatus();
         } else if (absl::IsOutOfRange(status) &&
                    sampler_options_.max_samples > 0) {
           // `max_samples` samples have already been returned by the iterator.
           *end_of_sequence = true;
-          return tensorflow::OkStatus();
+          return absl::OkStatus();
         } else {
           return ToTensorflowStatus(status);
         }

diff --git a/reverb/cc/ops/trajectory_dataset.cc b/reverb/cc/ops/trajectory_dataset.cc
@@ -120,7 +120,7 @@ class ReverbTrajectoryDatasetOp : public tensorflow::data::DatasetOpKernel {
     tensorflow::Status InputDatasets(
         std::vector<const DatasetBase*>* inputs) const override {
       inputs->clear();
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
    protected:
@@ -170,7 +170,7 @@ class ReverbTrajectoryDatasetOp : public tensorflow::data::DatasetOpKernel {
           },
           output));
 
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
    private:
@@ -252,12 +252,12 @@ class ReverbTrajectoryDatasetOp : public tensorflow::data::DatasetOpKernel {
                        absl::InfiniteDuration() &&
                    errors::IsRateLimiterTimeout(FromTensorflowStatus(status))) {
           *end_of_sequence = true;
-          return tensorflow::OkStatus();
+          return absl::OkStatus();
         } else if (sampler_options_.max_samples > 0 &&
                    absl::IsOutOfRange(FromTensorflowStatus(status))) {
           // `max_samples` samples have already been returned by the iterator.
           *end_of_sequence = true;
-          return tensorflow::OkStatus();
+          return absl::OkStatus();
         } else {
           return status;
         }