Make Head v2 public as tf.estimator.*

PiperOrigin-RevId: 253680221

tensorflow/tools/api/golden/v1/tensorflow.estimator.-binary-class-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.BinaryClassHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.binary_class_head.BinaryClassHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'weight_column\', \'thresholds\', \'label_vocabulary\', \'loss_reduction\', \'loss_fn\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'sum_over_batch_size\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v1/tensorflow.estimator.-head.pbtxt

@@ -0,0 +1,40 @@
+path: "tensorflow.estimator.Head"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<class \'abc.abstractproperty\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<class \'abc.abstractproperty\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<class \'abc.abstractproperty\'>"
+  }
+  member_method {
+    name: "__init__"
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+}

tensorflow/tools/api/golden/v1/tensorflow.estimator.-logistic-regression-head.pbtxt

@@ -0,0 +1,43 @@
+path: "tensorflow.estimator.LogisticRegressionHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.LogisticRegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.RegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'weight_column\', \'loss_reduction\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'sum_over_batch_size\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v1/tensorflow.estimator.-multi-class-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.MultiClassHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.multi_class_head.MultiClassHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'n_classes\', \'weight_column\', \'label_vocabulary\', \'loss_reduction\', \'loss_fn\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'sum_over_batch_size\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v1/tensorflow.estimator.-multi-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.MultiHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.multi_head.MultiHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'heads\', \'head_weights\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v1/tensorflow.estimator.-multi-label-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.MultiLabelHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.multi_label_head.MultiLabelHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'n_classes\', \'weight_column\', \'thresholds\', \'label_vocabulary\', \'loss_reduction\', \'loss_fn\', \'classes_for_class_based_metrics\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'sum_over_batch_size\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v1/tensorflow.estimator.-poisson-regression-head.pbtxt

@@ -0,0 +1,43 @@
+path: "tensorflow.estimator.PoissonRegressionHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.PoissonRegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.RegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'label_dimension\', \'weight_column\', \'loss_reduction\', \'compute_full_loss\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'None\', \'sum_over_batch_size\', \'True\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v1/tensorflow.estimator.-regression-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.RegressionHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.RegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'label_dimension\', \'weight_column\', \'loss_reduction\', \'loss_fn\', \'inverse_link_fn\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'None\', \'sum_over_batch_size\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v1/tensorflow.estimator.pbtxt

@@ -16,6 +16,10 @@ tf_module {
     name: "BestExporter"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "BinaryClassHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "BoostedTreesClassifier"
     mtype: "<type \'type\'>"
@@ -92,6 +96,10 @@ tf_module {
     name: "GlobalStepWaiterHook"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "Head"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "LatestExporter"
     mtype: "<type \'type\'>"
@@ -112,10 +120,26 @@ tf_module {
     name: "LoggingTensorHook"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "LogisticRegressionHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "ModeKeys"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "MultiClassHead"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "MultiHead"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "MultiLabelHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "NanLossDuringTrainingError"
     mtype: "<type \'type\'>"
@@ -124,10 +148,18 @@ tf_module {
     name: "NanTensorHook"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "PoissonRegressionHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "ProfilerHook"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "RegressionHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "RunConfig"
     mtype: "<type \'type\'>"

tensorflow/tools/api/golden/v2/tensorflow.estimator.-binary-class-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.BinaryClassHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.binary_class_head.BinaryClassHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'weight_column\', \'thresholds\', \'label_vocabulary\', \'loss_reduction\', \'loss_fn\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'sum_over_batch_size\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.estimator.-head.pbtxt

@@ -0,0 +1,40 @@
+path: "tensorflow.estimator.Head"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<class \'abc.abstractproperty\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<class \'abc.abstractproperty\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<class \'abc.abstractproperty\'>"
+  }
+  member_method {
+    name: "__init__"
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.estimator.-logistic-regression-head.pbtxt

@@ -0,0 +1,43 @@
+path: "tensorflow.estimator.LogisticRegressionHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.LogisticRegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.RegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'weight_column\', \'loss_reduction\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'sum_over_batch_size\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.estimator.-multi-class-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.MultiClassHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.multi_class_head.MultiClassHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'n_classes\', \'weight_column\', \'label_vocabulary\', \'loss_reduction\', \'loss_fn\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'sum_over_batch_size\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.estimator.-multi-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.MultiHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.multi_head.MultiHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'heads\', \'head_weights\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.estimator.-multi-label-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.MultiLabelHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.multi_label_head.MultiLabelHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'n_classes\', \'weight_column\', \'thresholds\', \'label_vocabulary\', \'loss_reduction\', \'loss_fn\', \'classes_for_class_based_metrics\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'sum_over_batch_size\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\', \'keys\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.estimator.-poisson-regression-head.pbtxt

@@ -0,0 +1,43 @@
+path: "tensorflow.estimator.PoissonRegressionHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.PoissonRegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.RegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'label_dimension\', \'weight_column\', \'loss_reduction\', \'compute_full_loss\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'None\', \'sum_over_batch_size\', \'True\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.estimator.-regression-head.pbtxt

@@ -0,0 +1,42 @@
+path: "tensorflow.estimator.RegressionHead"
+tf_class {
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.regression_head.RegressionHead\'>"
+  is_instance: "<class \'tensorflow_estimator.python.estimator.head.base_head.Head\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "logits_dimension"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "loss_reduction"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'label_dimension\', \'weight_column\', \'loss_reduction\', \'loss_fn\', \'inverse_link_fn\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'None\', \'sum_over_batch_size\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "create_estimator_spec"
+    argspec: "args=[\'self\', \'features\', \'mode\', \'logits\', \'labels\', \'optimizer\', \'trainable_variables\', \'train_op_fn\', \'update_ops\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "loss"
+    argspec: "args=[\'self\', \'labels\', \'logits\', \'features\', \'mode\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "metrics"
+    argspec: "args=[\'self\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "predictions"
+    argspec: "args=[\'self\', \'logits\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "update_metrics"
+    argspec: "args=[\'self\', \'eval_metrics\', \'features\', \'logits\', \'labels\', \'regularization_losses\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.estimator.pbtxt

@@ -16,6 +16,10 @@ tf_module {
     name: "BestExporter"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "BinaryClassHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "BoostedTreesClassifier"
     mtype: "<type \'type\'>"
@@ -92,6 +96,10 @@ tf_module {
     name: "GlobalStepWaiterHook"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "Head"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "LatestExporter"
     mtype: "<type \'type\'>"
@@ -112,10 +120,26 @@ tf_module {
     name: "LoggingTensorHook"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "LogisticRegressionHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "ModeKeys"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "MultiClassHead"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "MultiHead"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "MultiLabelHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "NanLossDuringTrainingError"
     mtype: "<type \'type\'>"
@@ -124,10 +148,18 @@ tf_module {
     name: "NanTensorHook"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "PoissonRegressionHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "ProfilerHook"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "RegressionHead"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "RunConfig"
     mtype: "<type \'type\'>"

tensorflow/tools/compatibility/tf_upgrade_v2.py

@@ -686,6 +686,14 @@ class TFAPIChangeSpec(ast_edits.NoUpdateSpec):
         "Please check the new API and use that instead."
     )
 
+    contrib_estimator_head_comment = (
+        ast_edits.WARNING,
+        "(Manual edit required) `tf.contrib.estimator.*_head` has been "
+        "deprecated, and its implementation has been integrated with "
+        "`tf.estimator.*Head` in TensorFlow 2.0. "
+        "Please check the new API and use that instead."
+    )
+
     initializers_no_dtype_comment = (
         ast_edits.INFO, "Initializers no longer have the "
         "dtype argument in the constructor or partition_info argument in the "
@@ -892,6 +900,20 @@ class TFAPIChangeSpec(ast_edits.NoUpdateSpec):
             assert_rank_comment,
         "tf.contrib.layers.layer_norm":
             contrib_layers_layer_norm_comment,
+        "tf.contrib.estimator.binary_classification_head":
+            contrib_estimator_head_comment,
+        "tf.contrib.estimator.logistic_regression_head":
+            contrib_estimator_head_comment,
+        "tf.contrib.estimator.multi_class_head":
+            contrib_estimator_head_comment,
+        "tf.contrib.estimator.multi_head":
+            contrib_estimator_head_comment,
+        "tf.contrib.estimator.multi_label_head":
+            contrib_estimator_head_comment,
+        "tf.contrib.estimator.poisson_regression_head":
+            contrib_estimator_head_comment,
+        "tf.contrib.estimator.regression_head":
+            contrib_estimator_head_comment,
         "tf.contrib.summary.all_summary_ops":
             contrib_summary_comment,
         "tf.contrib.summary.audio":

tensorflow/tools/compatibility/tf_upgrade_v2_test.py

@@ -1646,6 +1646,15 @@ def _log_prob(self, x):
     _, _, errors, _ = self._upgrade("tf.flags.FLAGS")
     self.assertIn("tf.flags has been removed", errors[0])
 
+  def test_contrib_estimator_head_deprecation(self):
+    api_symbols = ["binary_classification_head", "logistic_regression_head",
+                   "multi_class_head", "multi_head", "multi_label_head",
+                   "poisson_regression_head", "regression_head"]
+    for symbol in api_symbols:
+      text = "tf.contrib.estimator." + symbol
+      _, report, _, _ = self._upgrade(text)
+      self.assertIn("`tf.contrib.estimator.*_head` has been deprecated", report)
+
   def test_contrib_layers_layer_norm_deprecation(self):
     _, report, _, _ = self._upgrade("tf.contrib.layers.layer_norm")
     self.assertIn("`tf.contrib.layers.layer_norm` has been deprecated", report)