Skip to content

Commit 3c467c6

Browse files
SimpleML Teamcopybara-github
SimpleML Team
authored and
copybara-github
committed
Automated Code Change
PiperOrigin-RevId: 610341553
1 parent 0d546cf commit 3c467c6

File tree

3 files changed

+51
-51
lines changed

3 files changed

+51
-51
lines changed

‎tensorflow_decision_forests/tensorflow/ops/training/kernel.cc

Lines changed: 29 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ tensorflow::Status YggdrasilModelContainer::LoadModel(
6767
}
6868

6969
YDF_LOG(INFO) << "Loading model from " << model_path;
70-
return tf::OkStatus();
70+
return absl::OkStatus();
7171
}
7272

7373
// OP loading a model from disk to memory.
@@ -195,7 +195,7 @@ tf::Status FeatureSet::Link(
195195
}
196196
}
197197

198-
return tf::OkStatus();
198+
return absl::OkStatus();
199199
}
200200

201201
tf::Status FeatureSet::IterateFeatures(
@@ -235,46 +235,46 @@ tf::Status FeatureSet::IterateFeatures(
235235
TF_RETURN_IF_ERROR(lambda_hash(feature.second, feature.first));
236236
}
237237

238-
return tf::OkStatus();
238+
return absl::OkStatus();
239239
}
240240

241241
tf::Status FeatureSet::Unlink() {
242242
TF_RETURN_IF_ERROR(IterateFeatures(
243243
[](SimpleMLNumericalFeature::Resource* feature, const int feature_idx) {
244244
feature->Unref();
245-
return tf::OkStatus();
245+
return absl::OkStatus();
246246
},
247247
[](SimpleMLCategoricalStringFeature::Resource* feature,
248248
const int feature_idx) {
249249
feature->Unref();
250-
return tf::OkStatus();
250+
return absl::OkStatus();
251251
},
252252
[](SimpleMLCategoricalIntFeature::Resource* feature,
253253
const int feature_idx) {
254254
feature->Unref();
255-
return tf::OkStatus();
255+
return absl::OkStatus();
256256
},
257257
[](SimpleMLCategoricalSetStringFeature::Resource* feature,
258258
const int feature_idx) {
259259
feature->Unref();
260-
return tf::OkStatus();
260+
return absl::OkStatus();
261261
},
262262
[](SimpleMLCategoricalSetIntFeature::Resource* feature,
263263
const int feature_idx) {
264264
feature->Unref();
265-
return tf::OkStatus();
265+
return absl::OkStatus();
266266
},
267267
[](SimpleMLHashFeature::Resource* feature, const int feature_idx) {
268268
feature->Unref();
269-
return tf::OkStatus();
269+
return absl::OkStatus();
270270
}));
271271
numerical_features_.clear();
272272
categorical_string_features_.clear();
273273
categorical_int_features_.clear();
274274
categorical_set_string_features_.clear();
275275
categorical_set_int_features_.clear();
276276
hash_features_.clear();
277-
return tf::OkStatus();
277+
return absl::OkStatus();
278278
}
279279

280280
// Initialize a dataset (including the dataset's dataspec) from the linked
@@ -292,7 +292,7 @@ tf::Status FeatureSet::InitializeDatasetFromFeatures(
292292
if (num_examples == -1) {
293293
num_examples = observed_num_examples;
294294
num_batches = observed_num_batches;
295-
return tf::OkStatus();
295+
return absl::OkStatus();
296296
}
297297
if (num_examples != observed_num_examples) {
298298
return tf::Status(
@@ -301,7 +301,7 @@ tf::Status FeatureSet::InitializeDatasetFromFeatures(
301301
"different input features $0 != $1.",
302302
num_examples, observed_num_examples));
303303
}
304-
return tf::OkStatus();
304+
return absl::OkStatus();
305305
};
306306

307307
for (int feature_idx = 0; feature_idx < NumFeatures(); feature_idx++) {
@@ -346,7 +346,7 @@ tf::Status FeatureSet::InitializeDatasetFromFeatures(
346346
TF_RETURN_IF_ERROR(apply_guide(feature->feature_name(), col));
347347
TF_RETURN_IF_ERROR(set_num_examples(feature->indexed_data().size(),
348348
feature->NumBatches()));
349-
return tf::OkStatus();
349+
return absl::OkStatus();
350350
},
351351
[&](SimpleMLCategoricalIntFeature::Resource* feature,
352352
const int feature_idx) {
@@ -420,7 +420,7 @@ tf::Status FeatureSet::InitializeDatasetFromFeatures(
420420
dataset::UpdateComputeSpecDiscretizedNumerical(value, col, col_acc);
421421
}
422422
}
423-
return tf::OkStatus();
423+
return absl::OkStatus();
424424
},
425425
[&](SimpleMLCategoricalStringFeature::Resource* feature,
426426
const int feature_idx) {
@@ -432,7 +432,7 @@ tf::Status FeatureSet::InitializeDatasetFromFeatures(
432432
dataset::UpdateCategoricalStringColumnSpec(
433433
reverse_index[indexed_value], col, col_acc));
434434
}
435-
return tf::OkStatus();
435+
return absl::OkStatus();
436436
},
437437
[&](SimpleMLCategoricalIntFeature::Resource* feature,
438438
const int feature_idx) {
@@ -442,7 +442,7 @@ tf::Status FeatureSet::InitializeDatasetFromFeatures(
442442
TF_RETURN_IF_ERROR_FROM_ABSL_STATUS(
443443
dataset::UpdateCategoricalIntColumnSpec(value, col, col_acc));
444444
}
445-
return tf::OkStatus();
445+
return absl::OkStatus();
446446
},
447447
[&](SimpleMLCategoricalSetStringFeature::Resource* feature,
448448
const int feature_idx) {
@@ -452,7 +452,7 @@ tf::Status FeatureSet::InitializeDatasetFromFeatures(
452452
TF_RETURN_IF_ERROR_FROM_ABSL_STATUS(
453453
dataset::UpdateCategoricalStringColumnSpec(value, col, col_acc));
454454
}
455-
return tf::OkStatus();
455+
return absl::OkStatus();
456456
},
457457
[&](SimpleMLCategoricalSetIntFeature::Resource* feature,
458458
const int feature_idx) {
@@ -462,17 +462,17 @@ tf::Status FeatureSet::InitializeDatasetFromFeatures(
462462
TF_RETURN_IF_ERROR_FROM_ABSL_STATUS(
463463
dataset::UpdateCategoricalIntColumnSpec(value, col, col_acc));
464464
}
465-
return tf::OkStatus();
465+
return absl::OkStatus();
466466
},
467467
[&](SimpleMLHashFeature::Resource* feature, const int feature_idx) {
468468
// Nothing to do.
469-
return tf::OkStatus();
469+
return absl::OkStatus();
470470
}));
471471

472472
TF_RETURN_IF_ERROR_FROM_ABSL_STATUS(dataset::FinalizeComputeSpec(
473473
guide, accumulator, dataset->mutable_data_spec()));
474474

475-
return tf::OkStatus();
475+
return absl::OkStatus();
476476
}
477477

478478
tf::Status FeatureSet::MoveExamplesFromFeaturesToDataset(
@@ -492,7 +492,7 @@ tf::Status FeatureSet::MoveExamplesFromFeaturesToDataset(
492492
"between features for feature $0 != $1. For feature $2.",
493493
dataset->nrow(), num_rows, feature->feature_name()));
494494
}
495-
return tf::OkStatus();
495+
return absl::OkStatus();
496496
};
497497

498498
TF_RETURN_IF_ERROR(IterateFeatures(
@@ -524,7 +524,7 @@ tf::Status FeatureSet::MoveExamplesFromFeaturesToDataset(
524524
*col_data->mutable_values() = std::move(*feature->mutable_data());
525525
}
526526
feature->mutable_data()->clear();
527-
return tf::OkStatus();
527+
return absl::OkStatus();
528528
},
529529
[&](SimpleMLCategoricalStringFeature::Resource* feature,
530530
const int feature_idx) -> tensorflow::Status {
@@ -550,7 +550,7 @@ tf::Status FeatureSet::MoveExamplesFromFeaturesToDataset(
550550
}
551551
// Note: Thread annotations don't work in lambdas.
552552
feature->non_mutex_protected_clear();
553-
return tf::OkStatus();
553+
return absl::OkStatus();
554554
},
555555
[&](SimpleMLCategoricalIntFeature::Resource* feature,
556556
const int feature_idx) -> tensorflow::Status {
@@ -573,7 +573,7 @@ tf::Status FeatureSet::MoveExamplesFromFeaturesToDataset(
573573
col_data->Add(value);
574574
}
575575
feature->mutable_data()->clear();
576-
return tf::OkStatus();
576+
return absl::OkStatus();
577577
},
578578
[&](SimpleMLCategoricalSetStringFeature::Resource* feature,
579579
const int feature_idx) -> tensorflow::Status {
@@ -612,7 +612,7 @@ tf::Status FeatureSet::MoveExamplesFromFeaturesToDataset(
612612
col_data->AddVector(tmp_value);
613613
}
614614
feature->non_mutex_protected_clear();
615-
return tf::OkStatus();
615+
return absl::OkStatus();
616616
},
617617
[&](SimpleMLCategoricalSetIntFeature::Resource* feature,
618618
const int feature_idx) -> tensorflow::Status {
@@ -664,7 +664,7 @@ tf::Status FeatureSet::MoveExamplesFromFeaturesToDataset(
664664
col_data->AddVector(tmp_value);
665665
}
666666
feature->non_mutex_protected_clear();
667-
return tf::OkStatus();
667+
return absl::OkStatus();
668668
},
669669
[&](SimpleMLHashFeature::Resource* feature,
670670
const int feature_idx) -> tensorflow::Status {
@@ -675,10 +675,10 @@ tf::Status FeatureSet::MoveExamplesFromFeaturesToDataset(
675675
dataset::VerticalDataset::HashColumn>(feature_idx));
676676
*col_data->mutable_values() = std::move(*feature->mutable_data());
677677
feature->mutable_data()->clear();
678-
return tf::OkStatus();
678+
return absl::OkStatus();
679679
}));
680680

681-
return tf::OkStatus();
681+
return absl::OkStatus();
682682
}
683683

684684
int FeatureSet::NumFeatures() const {
@@ -974,7 +974,7 @@ class SimpleMLModelTrainer : public tensorflow::OpKernel {
974974
feature_set.InitializeDatasetFromFeatures(ctx, guide_, dataset));
975975
TF_RETURN_IF_ERROR(
976976
feature_set.MoveExamplesFromFeaturesToDataset(ctx, dataset));
977-
return tf::OkStatus();
977+
return absl::OkStatus();
978978
}
979979

980980
bool HasTrainingExamples(tf::OpKernelContext* ctx) {

0 commit comments

Comments
 (0)