No public description

Googler · copybara-github · commit bcd64c52865c · 2024-02-29T00:19:09.000-08:00
PiperOrigin-RevId: 611369523
Change-Id: I47c9d9eba96f43f3133e842f688bc1241f4a8a1b
diff --git a/tensorflow_compression/cc/kernels/range_coding_kernels.cc b/tensorflow_compression/cc/kernels/range_coding_kernels.cc
@@ -144,7 +144,7 @@ Status CheckCdfShape(const TensorShape& data_shape,
         "The last dimension of `cdf` should be > 1: ", cdf_shape.DebugString());
   }
 
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 tensorflow::Status CheckCdfValues(int precision,
@@ -169,7 +169,7 @@ tensorflow::Status CheckCdfValues(int precision,
       }
     }
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 class RangeEncodeOp : public OpKernel {
@@ -265,7 +265,7 @@ class RangeEncodeOp : public OpKernel {
     }
 
     encoder.Finalize(output);
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   int precision_;
@@ -369,7 +369,7 @@ class RangeDecodeOp : public OpKernel {
 
       *data = decoder.Decode({cdf_slice, chip_size}, precision_);
     }
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   int precision_;
diff --git a/tensorflow_compression/cc/kernels/range_coding_kernels_test.cc b/tensorflow_compression/cc/kernels/range_coding_kernels_test.cc
@@ -130,7 +130,7 @@ class RangeCoderOpsTest : public OpsTestBase {
     *output = *GetOutput(0);
     inputs_.clear();
 
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   Status RunDecodeOp(int precision, absl::Span<const Tensor> input,
@@ -166,7 +166,7 @@ class RangeCoderOpsTest : public OpsTestBase {
     *output = *GetOutput(0);
     inputs_.clear();
 
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   void TestEncodeAndDecode(int precision, const Tensor& data,
diff --git a/tensorflow_compression/cc/kernels/range_coding_kernels_util.cc b/tensorflow_compression/cc/kernels/range_coding_kernels_util.cc
@@ -87,7 +87,7 @@ Status MergeAxes(const TensorShape& broadcast_shape,
   }
   merged_storage_shape.push_back(storage_stride);
 
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 }  // namespace tensorflow_compression
diff --git a/tensorflow_compression/cc/kernels/run_length_gamma_kernels_test.cc b/tensorflow_compression/cc/kernels/run_length_gamma_kernels_test.cc
@@ -73,7 +73,7 @@ class BitCodingOpsTest : public OpsTestBase {
     TF_RETURN_IF_ERROR(RunOpKernel());
     *output = *GetOutput(0);
     inputs_.clear();
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   Status RunDecodeOp(absl::Span<const Tensor> inputs, Tensor* output) {
@@ -91,7 +91,7 @@ class BitCodingOpsTest : public OpsTestBase {
     TF_RETURN_IF_ERROR(RunOpKernel());
     *output = *GetOutput(0);
     inputs_.clear();
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   void TestEncodeAndDecode(const Tensor& data_tensor) {
diff --git a/tensorflow_compression/cc/kernels/run_length_kernels_test.cc b/tensorflow_compression/cc/kernels/run_length_kernels_test.cc
@@ -78,7 +78,7 @@ class BitCodingOpsTest : public OpsTestBase {
     TF_RETURN_IF_ERROR(RunOpKernel());
     *output = *GetOutput(0);
     inputs_.clear();
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   Status RunDecodeOp(absl::Span<const Tensor> inputs, Tensor* output,
@@ -102,7 +102,7 @@ class BitCodingOpsTest : public OpsTestBase {
     TF_RETURN_IF_ERROR(RunOpKernel());
     *output = *GetOutput(0);
     inputs_.clear();
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   void TestEncodeAndDecode(const Tensor& data_tensor, const int run_length_code,
diff --git a/tensorflow_compression/cc/kernels/unbounded_index_range_coding_kernels.cc b/tensorflow_compression/cc/kernels/unbounded_index_range_coding_kernels.cc
@@ -59,7 +59,7 @@ tensorflow::Status CheckIndex(int64_t upper_bound, const Tensor& index) {
                                      upper_bound, "): value=", flat(i));
     }
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 tensorflow::Status CheckCdfSize(int64_t upper_bound, const Tensor& cdf_size) {
@@ -70,7 +70,7 @@ tensorflow::Status CheckCdfSize(int64_t upper_bound, const Tensor& cdf_size) {
                                      upper_bound, "]: value=", flat(i));
     }
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 tensorflow::Status CheckCdf(int precision, const Tensor& cdf,
@@ -96,7 +96,7 @@ tensorflow::Status CheckCdf(int precision, const Tensor& cdf,
       }
     }
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 // Assumes that CheckArgumentShapes().ok().
@@ -107,7 +107,7 @@ tensorflow::Status CheckArgumentValues(int precision, const Tensor& index,
   TF_RETURN_IF_ERROR(CheckIndex(cdf.dim_size(0), index));
   TF_RETURN_IF_ERROR(CheckCdfSize(cdf.dim_size(1), cdf_size));
   TF_RETURN_IF_ERROR(CheckCdf(precision, cdf, cdf_size));
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 tensorflow::Status CheckArgumentShapes(const Tensor& index, const Tensor& cdf,
@@ -131,7 +131,7 @@ tensorflow::Status CheckArgumentShapes(const Tensor& index, const Tensor& cdf,
         "should match the number of rows in 'cdf': offset.shape=",
         offset.shape(), ", cdf.shape=", cdf.shape());
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 class UnboundedIndexRangeEncodeOp : public OpKernel {
@@ -365,7 +365,7 @@ class UnboundedIndexRangeDecodeOp : public OpKernel {
       output(i) = value;
     }
 
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   int precision_;
diff --git a/tensorflow_compression/cc/kernels/unbounded_index_range_coding_kernels_test.cc b/tensorflow_compression/cc/kernels/unbounded_index_range_coding_kernels_test.cc
@@ -167,7 +167,7 @@ class UnboundedIndexRangeCoderOpsTest : public OpsTestBase {
     *output = *GetOutput(0);
     inputs_.clear();
 
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   Status RunDecodeOpDebug(int precision, int overflow_width,
@@ -205,7 +205,7 @@ class UnboundedIndexRangeCoderOpsTest : public OpsTestBase {
     *output = *GetOutput(0);
     inputs_.clear();
 
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   void TestEncodeAndDecode(int precision, int overflow_width,
diff --git a/tensorflow_compression/cc/kernels/y4m_dataset_kernels.cc b/tensorflow_compression/cc/kernels/y4m_dataset_kernels.cc
@@ -93,12 +93,10 @@ class Y4MDatasetOp : public DatasetOpKernel {
     string DebugString() const override { return "Y4MDatasetOp::Dataset"; }
 
     Status InputDatasets(vector<const DatasetBase*>* inputs) const override {
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
-    Status CheckExternalState() const override {
-      return tensorflow::OkStatus();
-    }
+    Status CheckExternalState() const override { return absl::OkStatus(); }
 
    protected:
     Status AsGraphDefInternal(SerializationContext* ctx,
@@ -107,7 +105,7 @@ class Y4MDatasetOp : public DatasetOpKernel {
       Node* filenames = nullptr;
       TF_RETURN_IF_ERROR(b->AddVector(filenames_, &filenames));
       TF_RETURN_IF_ERROR(b->AddDataset(this, {filenames}, output));
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
    private:
@@ -199,7 +197,7 @@ class Y4MDatasetOp : public DatasetOpKernel {
           // Exit if there are no more files to process.
           if (file_index_ >= dataset()->filenames_.size()) {
             *end_of_sequence = true;
-            return tensorflow::OkStatus();
+            return absl::OkStatus();
           }
 
           // Open next file.
@@ -227,7 +225,7 @@ class Y4MDatasetOp : public DatasetOpKernel {
         TF_RETURN_IF_ERROR(
             writer->WriteScalar(full_name("file_pos"), file_pos));
 
-        return tensorflow::OkStatus();
+        return absl::OkStatus();
       }
 
       Status RestoreInternal(
@@ -252,7 +250,7 @@ class Y4MDatasetOp : public DatasetOpKernel {
                                          chroma_format_));
           file_pos_ = file_pos;
         }
-        return tensorflow::OkStatus();
+        return absl::OkStatus();
       }
 
      private:
@@ -280,7 +278,7 @@ class Y4MDatasetOp : public DatasetOpKernel {
               std::memcpy(&header[offset], chunk.data(), pos + 1);
             }
             header.resize(offset + pos + 1);
-            return tensorflow::OkStatus();
+            return absl::OkStatus();
           }
           // We reached the end of the file, and the header is not complete.
           if (!status.ok()) {
@@ -394,7 +392,7 @@ class Y4MDatasetOp : public DatasetOpKernel {
               "Input file '", dataset()->filenames_[file_index],
               "' has 4:2:0 chroma format, but odd width or height.");
         }
-        return tensorflow::OkStatus();
+        return absl::OkStatus();
       }
 
       mutex mu_;

Original file line number	Diff line number	Diff line change
`@@ -144,7 +144,7 @@ Status CheckCdfShape(const TensorShape& data_shape,`
`144`	`144`	"The last dimension of `cdf` should be > 1: ", cdf_shape.DebugString());
`145`	`145`	`}`
`146`	`146`
`147`		`- return tensorflow::OkStatus();`
	`147`	`+ return absl::OkStatus();`
`148`	`148`	`}`
`149`	`149`
`150`	`150`	`tensorflow::Status CheckCdfValues(int precision,`
`@@ -169,7 +169,7 @@ tensorflow::Status CheckCdfValues(int precision,`
`169`	`169`	`}`
`170`	`170`	`}`
`171`	`171`	`}`
`172`		`- return tensorflow::OkStatus();`
	`172`	`+ return absl::OkStatus();`
`173`	`173`	`}`
`174`	`174`
`175`	`175`	`class RangeEncodeOp : public OpKernel {`
`@@ -265,7 +265,7 @@ class RangeEncodeOp : public OpKernel {`
`265`	`265`	`}`
`266`	`266`
`267`	`267`	`encoder.Finalize(output);`
`268`		`- return tensorflow::OkStatus();`
	`268`	`+ return absl::OkStatus();`
`269`	`269`	`}`
`270`	`270`
`271`	`271`	`int precision_;`
`@@ -369,7 +369,7 @@ class RangeDecodeOp : public OpKernel {`
`369`	`369`
`370`	`370`	`*data = decoder.Decode({cdf_slice, chip_size}, precision_);`
`371`	`371`	`}`
`372`		`- return tensorflow::OkStatus();`
	`372`	`+ return absl::OkStatus();`
`373`	`373`	`}`
`374`	`374`
`375`	`375`	`int precision_;`
Original file line number	Diff line number	Diff line change
`@@ -130,7 +130,7 @@ class RangeCoderOpsTest : public OpsTestBase {`
`130`	`130`	`output = GetOutput(0);`
`131`	`131`	`inputs_.clear();`
`132`	`132`
`133`		`- return tensorflow::OkStatus();`
	`133`	`+ return absl::OkStatus();`
`134`	`134`	`}`
`135`	`135`
`136`	`136`	`Status RunDecodeOp(int precision, absl::Span<const Tensor> input,`
`@@ -166,7 +166,7 @@ class RangeCoderOpsTest : public OpsTestBase {`
`166`	`166`	`output = GetOutput(0);`
`167`	`167`	`inputs_.clear();`
`168`	`168`
`169`		`- return tensorflow::OkStatus();`
	`169`	`+ return absl::OkStatus();`
`170`	`170`	`}`
`171`	`171`
`172`	`172`	`void TestEncodeAndDecode(int precision, const Tensor& data,`
Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,7 @@ Status MergeAxes(const TensorShape& broadcast_shape,`
`87`	`87`	`}`
`88`	`88`	`merged_storage_shape.push_back(storage_stride);`
`89`	`89`
`90`		`- return tensorflow::OkStatus();`
	`90`	`+ return absl::OkStatus();`
`91`	`91`	`}`
`92`	`92`
`93`	`93`	`} // namespace tensorflow_compression`
Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@ tensorflow::Status CheckIndex(int64_t upper_bound, const Tensor& index) {`
`59`	`59`	`upper_bound, "): value=", flat(i));`
`60`	`60`	`}`
`61`	`61`	`}`
`62`		`- return tensorflow::OkStatus();`
	`62`	`+ return absl::OkStatus();`
`63`	`63`	`}`
`64`	`64`
`65`	`65`	`tensorflow::Status CheckCdfSize(int64_t upper_bound, const Tensor& cdf_size) {`
`@@ -70,7 +70,7 @@ tensorflow::Status CheckCdfSize(int64_t upper_bound, const Tensor& cdf_size) {`
`70`	`70`	`upper_bound, "]: value=", flat(i));`
`71`	`71`	`}`
`72`	`72`	`}`
`73`		`- return tensorflow::OkStatus();`
	`73`	`+ return absl::OkStatus();`
`74`	`74`	`}`
`75`	`75`
`76`	`76`	`tensorflow::Status CheckCdf(int precision, const Tensor& cdf,`
`@@ -96,7 +96,7 @@ tensorflow::Status CheckCdf(int precision, const Tensor& cdf,`
`96`	`96`	`}`
`97`	`97`	`}`
`98`	`98`	`}`
`99`		`- return tensorflow::OkStatus();`
	`99`	`+ return absl::OkStatus();`
`100`	`100`	`}`
`101`	`101`
`102`	`102`	`// Assumes that CheckArgumentShapes().ok().`
`@@ -107,7 +107,7 @@ tensorflow::Status CheckArgumentValues(int precision, const Tensor& index,`
`107`	`107`	`TF_RETURN_IF_ERROR(CheckIndex(cdf.dim_size(0), index));`
`108`	`108`	`TF_RETURN_IF_ERROR(CheckCdfSize(cdf.dim_size(1), cdf_size));`
`109`	`109`	`TF_RETURN_IF_ERROR(CheckCdf(precision, cdf, cdf_size));`
`110`		`- return tensorflow::OkStatus();`
	`110`	`+ return absl::OkStatus();`
`111`	`111`	`}`
`112`	`112`
`113`	`113`	`tensorflow::Status CheckArgumentShapes(const Tensor& index, const Tensor& cdf,`
`@@ -131,7 +131,7 @@ tensorflow::Status CheckArgumentShapes(const Tensor& index, const Tensor& cdf,`
`131`	`131`	`"should match the number of rows in 'cdf': offset.shape=",`
`132`	`132`	`offset.shape(), ", cdf.shape=", cdf.shape());`
`133`	`133`	`}`
`134`		`- return tensorflow::OkStatus();`
	`134`	`+ return absl::OkStatus();`
`135`	`135`	`}`
`136`	`136`
`137`	`137`	`class UnboundedIndexRangeEncodeOp : public OpKernel {`
`@@ -365,7 +365,7 @@ class UnboundedIndexRangeDecodeOp : public OpKernel {`
`365`	`365`	`output(i) = value;`
`366`	`366`	`}`
`367`	`367`
`368`		`- return tensorflow::OkStatus();`
	`368`	`+ return absl::OkStatus();`
`369`	`369`	`}`
`370`	`370`
`371`	`371`	`int precision_;`
Original file line number	Diff line number	Diff line change
`@@ -167,7 +167,7 @@ class UnboundedIndexRangeCoderOpsTest : public OpsTestBase {`
`167`	`167`	`output = GetOutput(0);`
`168`	`168`	`inputs_.clear();`
`169`	`169`
`170`		`- return tensorflow::OkStatus();`
	`170`	`+ return absl::OkStatus();`
`171`	`171`	`}`
`172`	`172`
`173`	`173`	`Status RunDecodeOpDebug(int precision, int overflow_width,`
`@@ -205,7 +205,7 @@ class UnboundedIndexRangeCoderOpsTest : public OpsTestBase {`
`205`	`205`	`output = GetOutput(0);`
`206`	`206`	`inputs_.clear();`
`207`	`207`
`208`		`- return tensorflow::OkStatus();`
	`208`	`+ return absl::OkStatus();`
`209`	`209`	`}`
`210`	`210`
`211`	`211`	`void TestEncodeAndDecode(int precision, int overflow_width,`
Original file line number	Diff line number	Diff line change
`@@ -93,12 +93,10 @@ class Y4MDatasetOp : public DatasetOpKernel {`
`93`	`93`	`string DebugString() const override { return "Y4MDatasetOp::Dataset"; }`
`94`	`94`
`95`	`95`	`Status InputDatasets(vector<const DatasetBase> inputs) const override {`
`96`		`- return tensorflow::OkStatus();`
	`96`	`+ return absl::OkStatus();`
`97`	`97`	`}`
`98`	`98`
`99`		`- Status CheckExternalState() const override {`
`100`		`- return tensorflow::OkStatus();`
`101`		`- }`
	`99`	`+ Status CheckExternalState() const override { return absl::OkStatus(); }`
`102`	`100`
`103`	`101`	`protected:`
`104`	`102`	`Status AsGraphDefInternal(SerializationContext* ctx,`
`@@ -107,7 +105,7 @@ class Y4MDatasetOp : public DatasetOpKernel {`
`107`	`105`	`Node* filenames = nullptr;`
`108`	`106`	`TF_RETURN_IF_ERROR(b->AddVector(filenames_, &filenames));`
`109`	`107`	`TF_RETURN_IF_ERROR(b->AddDataset(this, {filenames}, output));`
`110`		`- return tensorflow::OkStatus();`
	`108`	`+ return absl::OkStatus();`
`111`	`109`	`}`
`112`	`110`
`113`	`111`	`private:`
`@@ -199,7 +197,7 @@ class Y4MDatasetOp : public DatasetOpKernel {`
`199`	`197`	`// Exit if there are no more files to process.`
`200`	`198`	`if (file_index_ >= dataset()->filenames_.size()) {`
`201`	`199`	`*end_of_sequence = true;`
`202`		`- return tensorflow::OkStatus();`
	`200`	`+ return absl::OkStatus();`
`203`	`201`	`}`
`204`	`202`
`205`	`203`	`// Open next file.`
`@@ -227,7 +225,7 @@ class Y4MDatasetOp : public DatasetOpKernel {`
`227`	`225`	`TF_RETURN_IF_ERROR(`
`228`	`226`	`writer->WriteScalar(full_name("file_pos"), file_pos));`
`229`	`227`
`230`		`- return tensorflow::OkStatus();`
	`228`	`+ return absl::OkStatus();`
`231`	`229`	`}`
`232`	`230`
`233`	`231`	`Status RestoreInternal(`
`@@ -252,7 +250,7 @@ class Y4MDatasetOp : public DatasetOpKernel {`
`252`	`250`	`chroma_format_));`
`253`	`251`	`file_pos_ = file_pos;`
`254`	`252`	`}`
`255`		`- return tensorflow::OkStatus();`
	`253`	`+ return absl::OkStatus();`
`256`	`254`	`}`
`257`	`255`
`258`	`256`	`private:`
`@@ -280,7 +278,7 @@ class Y4MDatasetOp : public DatasetOpKernel {`
`280`	`278`	`std::memcpy(&header[offset], chunk.data(), pos + 1);`
`281`	`279`	`}`
`282`	`280`	`header.resize(offset + pos + 1);`
`283`		`- return tensorflow::OkStatus();`
	`281`	`+ return absl::OkStatus();`
`284`	`282`	`}`
`285`	`283`	`// We reached the end of the file, and the header is not complete.`
`286`	`284`	`if (!status.ok()) {`
`@@ -394,7 +392,7 @@ class Y4MDatasetOp : public DatasetOpKernel {`
`394`	`392`	`"Input file '", dataset()->filenames_[file_index],`
`395`	`393`	`"' has 4:2:0 chroma format, but odd width or height.");`
`396`	`394`	`}`
`397`		`- return tensorflow::OkStatus();`
	`395`	`+ return absl::OkStatus();`
`398`	`396`	`}`
`399`	`397`
`400`	`398`	`mutex mu_;`