SYMBOL INDEX (4635 symbols across 144 files)

FILE: tensorflow_estimator/python/estimator/canned/baseline.py
  function _get_weight_column_key (line 70) | def _get_weight_column_key(weight_column):
  function _get_weight_column_key_v2 (line 81) | def _get_weight_column_key_v2(weight_column):
  function _get_batch_size_and_size_checks (line 92) | def _get_batch_size_and_size_checks(features, weight_column_key):
  function _baseline_logit_fn_builder (line 114) | def _baseline_logit_fn_builder(num_outputs, weight_column=None):
  function _baseline_model_fn (line 156) | def _baseline_model_fn(features,
  function _baseline_model_fn_builder_v2 (line 204) | def _baseline_model_fn_builder_v2(features, num_outputs, weight_column=N...
  function _baseline_model_fn_v2 (line 229) | def _baseline_model_fn_v2(
  class BaselineClassifierV2 (line 293) | class BaselineClassifierV2(estimator.EstimatorV2):
    method __init__ (line 346) | def __init__(self,
  class BaselineClassifier (line 404) | class BaselineClassifier(estimator.Estimator):
    method __init__ (line 407) | def __init__(self,
  class BaselineEstimatorV2 (line 433) | class BaselineEstimatorV2(estimator.EstimatorV2):
    method __init__ (line 483) | def __init__(self, head, model_dir=None, optimizer='Ftrl', config=None):
  class BaselineEstimator (line 512) | class BaselineEstimator(estimator.Estimator):
    method __init__ (line 515) | def __init__(self, head, model_dir=None, optimizer='Ftrl', config=None):
  class BaselineRegressorV2 (line 531) | class BaselineRegressorV2(estimator.EstimatorV2):
    method __init__ (line 579) | def __init__(self,
  class BaselineRegressor (line 627) | class BaselineRegressor(estimator.Estimator):
    method __init__ (line 630) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/canned/baseline_estimator_test.py
  function assert_close (line 40) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 55) | def save_variables_to_ckpt(model_dir):
  function _baseline_estimator_fn (line 62) | def _baseline_estimator_fn(weight_column=None, label_dimension=1, **kwar...
  function mock_optimizer_v2 (line 71) | def mock_optimizer_v2(testcase, expected_loss=None):
  class BaselineEstimatorEvaluationTest (line 114) | class BaselineEstimatorEvaluationTest(tf.test.TestCase):
    method setUp (line 116) | def setUp(self):
    method tearDown (line 119) | def tearDown(self):
    method test_evaluation_batch (line 124) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 149) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 179) | def test_evaluation_for_multi_dimensions(self):
  class BaselineEstimatorPredictTest (line 208) | class BaselineEstimatorPredictTest(tf.test.TestCase):
    method setUp (line 210) | def setUp(self):
    method tearDown (line 213) | def tearDown(self):
    method test_1d (line 218) | def test_1d(self):
    method testMultiDim (line 238) | def testMultiDim(self):
  class BaselineEstimatorIntegrationTest (line 264) | class BaselineEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 266) | def setUp(self):
    method tearDown (line 269) | def tearDown(self):
    method _test_complete_flow (line 274) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 304) | def test_numpy_input_fn(self):
  class BaselineEstimatorTrainingTest (line 341) | class BaselineEstimatorTrainingTest(tf.test.TestCase):
    method setUp (line 343) | def setUp(self):
    method tearDown (line 346) | def tearDown(self):
    method _assert_checkpoint (line 351) | def _assert_checkpoint(self,
    method testFromScratch (line 371) | def testFromScratch(self):
    method testFromCheckpoint (line 390) | def testFromCheckpoint(self):

FILE: tensorflow_estimator/python/estimator/canned/baseline_test.py
  function assert_close (line 56) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 71) | def save_variables_to_ckpt(model_dir):
  function queue_parsed_features (line 78) | def queue_parsed_features(feature_map):
  function sorted_key_dict (line 93) | def sorted_key_dict(unsorted_dict):
  function sigmoid (line 97) | def sigmoid(x):
  function _baseline_regressor_fn (line 101) | def _baseline_regressor_fn(*args, **kwargs):
  function _baseline_classifier_fn (line 105) | def _baseline_classifier_fn(*args, **kwargs):
  function mock_optimizer_v2 (line 109) | def mock_optimizer_v2(testcase, expected_loss=None):
  class BaselineRegressorEvaluationTest (line 157) | class BaselineRegressorEvaluationTest(tf.test.TestCase):
    method setUp (line 159) | def setUp(self):
    method tearDown (line 162) | def tearDown(self):
    method test_evaluation_for_simple_data (line 167) | def test_evaluation_for_simple_data(self):
    method test_evaluation_batch (line 190) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 217) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 247) | def test_evaluation_for_multi_dimensions(self):
  class BaselineRegressorPredictTest (line 276) | class BaselineRegressorPredictTest(tf.test.TestCase):
    method setUp (line 278) | def setUp(self):
    method tearDown (line 281) | def tearDown(self):
    method test_1d (line 286) | def test_1d(self):
    method testMultiDim (line 306) | def testMultiDim(self):
  class BaselineRegressorIntegrationTest (line 332) | class BaselineRegressorIntegrationTest(tf.test.TestCase):
    method setUp (line 334) | def setUp(self):
    method tearDown (line 337) | def tearDown(self):
    method _test_complete_flow (line 342) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 372) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 408) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 437) | def test_input_fn_from_parse_example(self):
  class BaselineRegressorTrainingTest (line 498) | class BaselineRegressorTrainingTest(tf.test.TestCase):
    method setUp (line 500) | def setUp(self):
    method tearDown (line 503) | def tearDown(self):
    method _assert_checkpoint (line 508) | def _assert_checkpoint(self,
    method testFromScratchWithDefaultOptimizer (line 528) | def testFromScratchWithDefaultOptimizer(self):
    method testTrainWithOneDimLabel (line 542) | def testTrainWithOneDimLabel(self):
    method testTrainWithOneDimWeight (line 559) | def testTrainWithOneDimWeight(self):
    method testFromScratch (line 582) | def testFromScratch(self):
    method testFromCheckpoint (line 603) | def testFromCheckpoint(self):
    method testFromCheckpointMultiBatch (line 635) | def testFromCheckpointMultiBatch(self):
  class BaselineClassifierTrainingTest (line 675) | class BaselineClassifierTrainingTest(tf.test.TestCase):
    method setUp (line 677) | def setUp(self):
    method tearDown (line 680) | def tearDown(self):
    method _assert_checkpoint (line 684) | def _assert_checkpoint(self,
    method _testFromScratchWithDefaultOptimizer (line 706) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassesFromScratchWithDefaultOptimizer (line 720) | def testBinaryClassesFromScratchWithDefaultOptimizer(self):
    method testMultiClassesFromScratchWithDefaultOptimizer (line 723) | def testMultiClassesFromScratchWithDefaultOptimizer(self):
    method _testTrainWithTwoDimsLabel (line 726) | def _testTrainWithTwoDimsLabel(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsLabel (line 745) | def testBinaryClassesTrainWithTwoDimsLabel(self):
    method testMultiClassesTrainWithTwoDimsLabel (line 748) | def testMultiClassesTrainWithTwoDimsLabel(self):
    method _testTrainWithOneDimLabel (line 751) | def _testTrainWithOneDimLabel(self, n_classes):
    method testBinaryClassesTrainWithOneDimLabel (line 768) | def testBinaryClassesTrainWithOneDimLabel(self):
    method testMultiClassesTrainWithOneDimLabel (line 771) | def testMultiClassesTrainWithOneDimLabel(self):
    method _testTrainWithTwoDimsWeight (line 774) | def _testTrainWithTwoDimsWeight(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsWeight (line 796) | def testBinaryClassesTrainWithTwoDimsWeight(self):
    method testMultiClassesTrainWithTwoDimsWeight (line 799) | def testMultiClassesTrainWithTwoDimsWeight(self):
    method _testTrainWithOneDimWeight (line 802) | def _testTrainWithOneDimWeight(self, n_classes):
    method testBinaryClassesTrainWithOneDimWeight (line 822) | def testBinaryClassesTrainWithOneDimWeight(self):
    method testMultiClassesTrainWithOneDimWeight (line 825) | def testMultiClassesTrainWithOneDimWeight(self):
    method _testFromScratch (line 828) | def _testFromScratch(self, n_classes):
    method testBinaryClassesFromScratch (line 862) | def testBinaryClassesFromScratch(self):
    method testMultiClassesFromScratch (line 865) | def testMultiClassesFromScratch(self):
    method _testFromCheckpoint (line 868) | def _testFromCheckpoint(self, n_classes):
    method testBinaryClassesFromCheckpoint (line 918) | def testBinaryClassesFromCheckpoint(self):
    method testMultiClassesFromCheckpoint (line 921) | def testMultiClassesFromCheckpoint(self):
    method _testFromCheckpointFloatLabels (line 924) | def _testFromCheckpointFloatLabels(self, n_classes):
    method testBinaryClassesFromCheckpointFloatLabels (line 960) | def testBinaryClassesFromCheckpointFloatLabels(self):
    method testMultiClassesFromCheckpointFloatLabels (line 963) | def testMultiClassesFromCheckpointFloatLabels(self):
    method _testFromCheckpointMultiBatch (line 966) | def _testFromCheckpointMultiBatch(self, n_classes):
    method testBinaryClassesFromCheckpointMultiBatch (line 1024) | def testBinaryClassesFromCheckpointMultiBatch(self):
    method testMultiClassesFromCheckpointMultiBatch (line 1027) | def testMultiClassesFromCheckpointMultiBatch(self):
  class BaselineClassifierEvaluationTest (line 1031) | class BaselineClassifierEvaluationTest(tf.test.TestCase):
    method setUp (line 1033) | def setUp(self):
    method tearDown (line 1036) | def tearDown(self):
    method _test_evaluation_for_simple_data (line 1040) | def _test_evaluation_for_simple_data(self, n_classes):
    method test_binary_classes_evaluation_for_simple_data (line 1094) | def test_binary_classes_evaluation_for_simple_data(self):
    method test_multi_classes_evaluation_for_simple_data (line 1097) | def test_multi_classes_evaluation_for_simple_data(self):
    method _test_evaluation_batch (line 1100) | def _test_evaluation_batch(self, n_classes):
    method test_binary_classes_evaluation_batch (line 1164) | def test_binary_classes_evaluation_batch(self):
    method test_multi_classes_evaluation_batch (line 1167) | def test_multi_classes_evaluation_batch(self):
    method _test_evaluation_weights (line 1170) | def _test_evaluation_weights(self, n_classes):
    method test_binary_classes_evaluation_weights (line 1249) | def test_binary_classes_evaluation_weights(self):
    method test_multi_classes_evaluation_weights (line 1252) | def test_multi_classes_evaluation_weights(self):
  class BaselineClassifierPredictTest (line 1256) | class BaselineClassifierPredictTest(tf.test.TestCase):
    method setUp (line 1258) | def setUp(self):
    method tearDown (line 1261) | def tearDown(self):
    method _testPredictions (line 1265) | def _testPredictions(self, n_classes, label_vocabulary, label_output_fn):
    method testBinaryClassesWithoutLabelVocabulary (line 1332) | def testBinaryClassesWithoutLabelVocabulary(self):
    method testBinaryClassesWithLabelVocabulary (line 1339) | def testBinaryClassesWithLabelVocabulary(self):
    method testMultiClassesWithoutLabelVocabulary (line 1346) | def testMultiClassesWithoutLabelVocabulary(self):
    method testMultiClassesWithLabelVocabulary (line 1353) | def testMultiClassesWithLabelVocabulary(self):
  class BaselineClassifierIntegrationTest (line 1361) | class BaselineClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 1363) | def setUp(self):
    method tearDown (line 1366) | def tearDown(self):
    method _test_complete_flow (line 1370) | def _test_complete_flow(self, n_classes, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn (line 1400) | def _test_numpy_input_fn(self, n_classes):
    method test_binary_classes_numpy_input_fn (line 1436) | def test_binary_classes_numpy_input_fn(self):
    method test_multi_classes_numpy_input_fn (line 1439) | def test_multi_classes_numpy_input_fn(self):
    method _test_pandas_input_fn (line 1442) | def _test_pandas_input_fn(self, n_classes):
    method test_binary_classes_pandas_input_fn (line 1471) | def test_binary_classes_pandas_input_fn(self):
    method test_multi_classes_pandas_input_fn (line 1474) | def test_multi_classes_pandas_input_fn(self):
    method _test_input_fn_from_parse_example (line 1477) | def _test_input_fn_from_parse_example(self, n_classes):
    method test_binary_classes_input_fn_from_parse_example (line 1536) | def test_binary_classes_input_fn_from_parse_example(self):
    method test_multi_classes_input_fn_from_parse_example (line 1539) | def test_multi_classes_input_fn_from_parse_example(self):
  class BaselineLogitFnTest (line 1546) | class BaselineLogitFnTest(tf.test.TestCase):
    method test_basic_logit_correctness (line 1548) | def test_basic_logit_correctness(self):

FILE: tensorflow_estimator/python/estimator/canned/canned_estimator_ds_integration_test.py
  class CannedEstimatorDistributionStrategyTest (line 35) | class CannedEstimatorDistributionStrategyTest(tf.test.TestCase,
    method setUp (line 38) | def setUp(self):
    method dataset_input_fn (line 45) | def dataset_input_fn(self, x, y, batch_size, shuffle):
    method test_canned_estimator (line 69) | def test_canned_estimator(self, distribution, estimator_cls):

FILE: tensorflow_estimator/python/estimator/canned/dnn.py
  function _add_hidden_layer_summary (line 41) | def _add_hidden_layer_summary(value, tag):
  function dnn_logit_fn_builder (line 48) | def dnn_logit_fn_builder(units, hidden_units, feature_columns, activatio...
  function dnn_logit_fn_builder_v2 (line 101) | def dnn_logit_fn_builder_v2(units, hidden_units, feature_columns, activa...
  function _get_previous_name_scope (line 152) | def _get_previous_name_scope():
  class _DNNModel (line 157) | class _DNNModel(tf_keras.Model):
    method __init__ (line 160) | def __init__(self,
    method call (line 232) | def call(self, features, mode):
    method _add_layer (line 259) | def _add_layer(self, layer, layer_name):
  function _name_from_scope_name (line 266) | def _name_from_scope_name(name):
  class _DNNModelV2 (line 278) | class _DNNModelV2(tf_keras.Model):
    method __init__ (line 281) | def __init__(self,
    method call (line 349) | def call(self, features, mode):
  function _validate_features (line 368) | def _validate_features(features):
  function _get_dnn_estimator_spec (line 374) | def _get_dnn_estimator_spec(use_tpu, head, features, labels, mode, logits,
  function _dnn_model_fn (line 393) | def _dnn_model_fn(features,
  function _dnn_model_fn_builder_v2 (line 466) | def _dnn_model_fn_builder_v2(units, hidden_units, feature_columns,
  function dnn_model_fn_v2 (line 513) | def dnn_model_fn_v2(features,
  class DNNClassifierV2 (line 596) | class DNNClassifierV2(estimator.EstimatorV2):
    method __init__ (line 682) | def __init__(
  class DNNClassifier (line 773) | class DNNClassifier(estimator.Estimator):
    method __init__ (line 776) | def __init__(
  class DNNEstimatorV2 (line 821) | class DNNEstimatorV2(estimator.EstimatorV2):
    method __init__ (line 911) | def __init__(self,
  class DNNEstimator (line 976) | class DNNEstimator(estimator.Estimator):
    method __init__ (line 979) | def __init__(self,
  class DNNRegressorV2 (line 1017) | class DNNRegressorV2(estimator.EstimatorV2):
    method __init__ (line 1103) | def __init__(
  class DNNRegressor (line 1186) | class DNNRegressor(estimator.Estimator):
    method __init__ (line 1189) | def __init__(

FILE: tensorflow_estimator/python/estimator/canned/dnn_estimator_test.py
  function _dnn_estimator_fn (line 37) | def _dnn_estimator_fn(weight_column=None, label_dimension=1, **kwargs):
  function _dnn_estimator_classifier_fn (line 48) | def _dnn_estimator_classifier_fn(n_classes=3, **kwargs):
  class DNNLogitFnBuilderTest (line 53) | class DNNLogitFnBuilderTest(tf.test.TestCase):
    method testLongInPy2 (line 55) | def testLongInPy2(self):
  class DNNEstimatorEvaluateTest (line 62) | class DNNEstimatorEvaluateTest(dnn_testing_utils.BaseDNNRegressorEvaluat...
    method __init__ (line 65) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorPredictTest (line 71) | class DNNEstimatorPredictTest(dnn_testing_utils.BaseDNNRegressorPredictT...
    method __init__ (line 74) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorTrainTest (line 80) | class DNNEstimatorTrainTest(dnn_testing_utils.BaseDNNRegressorTrainTest,
    method __init__ (line 83) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorWarmStartingTest (line 89) | class DNNEstimatorWarmStartingTest(dnn_testing_utils.BaseDNNWarmStarting...
    method __init__ (line 92) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorIntegrationTest (line 98) | class DNNEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 100) | def setUp(self):
    method tearDown (line 103) | def tearDown(self):
    method _test_complete_flow (line 108) | def _test_complete_flow(self,
    method _create_input_fn (line 150) | def _create_input_fn(self, label_dimension, batch_size):
    method test_numpy_input_fn (line 168) | def test_numpy_input_fn(self):
    method test_numpy_input_fn_with_optimizer_instance (line 183) | def test_numpy_input_fn_with_optimizer_instance(self):

FILE: tensorflow_estimator/python/estimator/canned/dnn_linear_combined.py
  function _check_no_sync_replicas_optimizer (line 41) | def _check_no_sync_replicas_optimizer(optimizer):
  function _linear_learning_rate (line 50) | def _linear_learning_rate(num_linear_feature_columns):
  function _add_layer_summary (line 67) | def _add_layer_summary(value, tag):
  function _validate_feature_columns (line 73) | def _validate_feature_columns(linear_feature_columns, dnn_feature_columns):
  function _dnn_linear_combined_model_fn_v2 (line 84) | def _dnn_linear_combined_model_fn_v2(
  function _dnn_linear_combined_model_fn (line 235) | def _dnn_linear_combined_model_fn(features,
  class DNNLinearCombinedClassifierV2 (line 393) | class DNNLinearCombinedClassifierV2(estimator.EstimatorV2):
    method __init__ (line 477) | def __init__(self,
  class DNNLinearCombinedClassifier (line 594) | class DNNLinearCombinedClassifier(estimator.Estimator):
    method __init__ (line 598) | def __init__(self,
  function _init_dnn_linear_combined_estimator (line 651) | def _init_dnn_linear_combined_estimator(head, linear_feature_columns,
  class DNNLinearCombinedEstimatorV2 (line 687) | class DNNLinearCombinedEstimatorV2(estimator.EstimatorV2):
    method __init__ (line 770) | def __init__(self,
  class DNNLinearCombinedEstimator (line 852) | class DNNLinearCombinedEstimator(estimator.Estimator):
    method __init__ (line 855) | def __init__(self,
  class DNNLinearCombinedRegressorV2 (line 899) | class DNNLinearCombinedRegressorV2(estimator.EstimatorV2):
    method __init__ (line 983) | def __init__(self,
  class DNNLinearCombinedRegressor (line 1091) | class DNNLinearCombinedRegressor(estimator.Estimator):
    method __init__ (line 1095) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/canned/dnn_linear_combined_estimator_test.py
  function _dnn_only_estimator_fn (line 37) | def _dnn_only_estimator_fn(hidden_units,
  class DNNOnlyEstimatorEvaluateTest (line 61) | class DNNOnlyEstimatorEvaluateTest(
    method __init__ (line 64) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyEstimatorPredictTest (line 70) | class DNNOnlyEstimatorPredictTest(dnn_testing_utils.BaseDNNRegressorPred...
    method __init__ (line 73) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyEstimatorTrainTest (line 79) | class DNNOnlyEstimatorTrainTest(dnn_testing_utils.BaseDNNRegressorTrainT...
    method __init__ (line 82) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _linear_only_estimator_fn (line 88) | def _linear_only_estimator_fn(feature_columns,
  class LinearOnlyEstimatorEvaluateTest (line 108) | class LinearOnlyEstimatorEvaluateTest(
    method __init__ (line 111) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyEstimatorPredictTest (line 117) | class LinearOnlyEstimatorPredictTest(
    method __init__ (line 120) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyEstimatorTrainTest (line 126) | class LinearOnlyEstimatorTrainTest(
    method __init__ (line 129) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedEstimatorIntegrationTest (line 135) | class DNNLinearCombinedEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 137) | def setUp(self):
    method tearDown (line 140) | def tearDown(self):
    method _test_complete_flow (line 145) | def _test_complete_flow(self,
    method _create_input_fn (line 194) | def _create_input_fn(self, label_dimension, batch_size):
    method test_numpy_input_fn (line 212) | def test_numpy_input_fn(self):
    method test_numpy_input_fn_with_optimizer_instance (line 227) | def test_numpy_input_fn_with_optimizer_instance(self):

FILE: tensorflow_estimator/python/estimator/canned/dnn_linear_combined_test.py
  class DNNOnlyModelFnTest (line 53) | class DNNOnlyModelFnTest(dnn_testing_utils.BaseDNNModelFnTest,
    method __init__ (line 56) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method _dnn_only_model_fn (line 60) | def _dnn_only_model_fn(self,
  function _linear_regressor_fn (line 86) | def _linear_regressor_fn(feature_columns,
  class LinearOnlyRegressorEvaluationV2Test (line 103) | class LinearOnlyRegressorEvaluationV2Test(
    method __init__ (line 106) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorPredictV2Test (line 112) | class LinearOnlyRegressorPredictV2Test(
    method __init__ (line 115) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorIntegrationV2Test (line 121) | class LinearOnlyRegressorIntegrationV2Test(
    method __init__ (line 124) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorTrainingV2Test (line 130) | class LinearOnlyRegressorTrainingV2Test(
    method __init__ (line 133) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _linear_classifier_fn (line 139) | def _linear_classifier_fn(feature_columns,
  class LinearOnlyClassifierTrainingV2Test (line 158) | class LinearOnlyClassifierTrainingV2Test(
    method __init__ (line 161) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierClassesEvaluationV2Test (line 169) | class LinearOnlyClassifierClassesEvaluationV2Test(
    method __init__ (line 172) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierPredictV2Test (line 180) | class LinearOnlyClassifierPredictV2Test(
    method __init__ (line 183) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierIntegrationV2Test (line 191) | class LinearOnlyClassifierIntegrationV2Test(
    method __init__ (line 194) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedRegressorIntegrationTest (line 203) | class DNNLinearCombinedRegressorIntegrationTest(tf.test.TestCase):
    method setUp (line 205) | def setUp(self):
    method tearDown (line 208) | def tearDown(self):
    method _test_complete_flow_helper (line 213) | def _test_complete_flow_helper(self, linear_feature_columns,
    method _test_complete_flow (line 248) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method _test_complete_flow_dnn_fc_v1 (line 264) | def _test_complete_flow_dnn_fc_v1(self, train_input_fn, eval_input_fn,
    method _test_complete_flow_linear_fc_v1 (line 282) | def _test_complete_flow_linear_fc_v1(self, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn_helper (line 300) | def _test_numpy_input_fn_helper(self, fc_impl, fn_to_run):
    method test_numpy_input_fn_basic (line 327) | def test_numpy_input_fn_basic(self, fc_impl):
    method test_numpy_input_fn_dnn_fc_v1 (line 330) | def test_numpy_input_fn_dnn_fc_v1(self, fc_impl):
    method test_numpy_input_fn_linear_fc_v1 (line 336) | def test_numpy_input_fn_linear_fc_v1(self, fc_impl):
    method _test_pandas_input_fn_helper (line 342) | def _test_pandas_input_fn_helper(self, fc_impl, fn_to_run):
    method test_pandas_input_fn_basic (line 367) | def test_pandas_input_fn_basic(self, fc_impl):
    method test_pandas_input_fn_dnn_fc_v1 (line 370) | def test_pandas_input_fn_dnn_fc_v1(self, fc_impl):
    method test_pandas_input_fn_linear_fc_v1 (line 376) | def test_pandas_input_fn_linear_fc_v1(self, fc_impl):
    method _test_input_fn_from_parse_example_helper (line 382) | def _test_input_fn_from_parse_example_helper(self, fc_impl, fn_to_run):
    method test_input_fn_from_parse_example_basic (line 440) | def test_input_fn_from_parse_example_basic(self, fc_impl):
    method test_input_fn_from_parse_example_dnn_fc_v1 (line 444) | def test_input_fn_from_parse_example_dnn_fc_v1(self, fc_impl):
    method test_input_fn_from_parse_example_linear_fc_v1 (line 450) | def test_input_fn_from_parse_example_linear_fc_v1(self, fc_impl):
  function _dnn_classifier_fn (line 458) | def _dnn_classifier_fn(hidden_units,
  class DNNOnlyClassifierEvaluateV2Test (line 477) | class DNNOnlyClassifierEvaluateV2Test(
    method __init__ (line 480) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierPredictV2Test (line 486) | class DNNOnlyClassifierPredictV2Test(
    method __init__ (line 489) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierTrainV2Test (line 495) | class DNNOnlyClassifierTrainV2Test(dnn_testing_utils.BaseDNNClassifierTr...
    method __init__ (line 498) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 505) | def _dnn_regressor_fn(hidden_units,
  class DNNOnlyRegressorEvaluateV2Test (line 522) | class DNNOnlyRegressorEvaluateV2Test(
    method __init__ (line 525) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorPredictV2Test (line 531) | class DNNOnlyRegressorPredictV2Test(
    method __init__ (line 534) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorTrainV2Test (line 540) | class DNNOnlyRegressorTrainV2Test(dnn_testing_utils.BaseDNNRegressorTrai...
    method __init__ (line 543) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedClassifierIntegrationTest (line 550) | class DNNLinearCombinedClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 552) | def setUp(self):
    method tearDown (line 555) | def tearDown(self):
    method _as_label (line 560) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 563) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 603) | def test_numpy_input_fn(self, fc_impl):
    method test_pandas_input_fn (line 633) | def test_pandas_input_fn(self, fc_impl):
    method test_input_fn_from_parse_example (line 659) | def test_input_fn_from_parse_example(self, fc_impl):
  class DNNLinearCombinedTests (line 722) | class DNNLinearCombinedTests(tf.test.TestCase):
    method setUp (line 724) | def setUp(self):
    method tearDown (line 727) | def tearDown(self):
    method test_train_op_calls_both_dnn_and_linear (line 731) | def test_train_op_calls_both_dnn_and_linear(self, fc_impl):
    method test_dnn_and_linear_logits_are_added (line 757) | def test_dnn_and_linear_logits_are_added(self, fc_impl):
  class DNNLinearCombinedWarmStartingTest (line 785) | class DNNLinearCombinedWarmStartingTest(tf.test.TestCase):
    method setUp (line 787) | def setUp(self):
    method tearDown (line 801) | def tearDown(self):
    method test_classifier_basic_warm_starting (line 806) | def test_classifier_basic_warm_starting(self, fc_impl):
    method test_regressor_basic_warm_starting (line 856) | def test_regressor_basic_warm_starting(self, fc_impl):
    method test_warm_starting_selective_variables (line 904) | def test_warm_starting_selective_variables(self, fc_impl):

FILE: tensorflow_estimator/python/estimator/canned/dnn_test_fc_v2.py
  function _dnn_classifier_fn (line 52) | def _dnn_classifier_fn(*args, **kwargs):
  class DNNModelFnV2Test (line 56) | class DNNModelFnV2Test(dnn_testing_utils.BaseDNNModelFnTest, tf.test.Tes...
    method __init__ (line 58) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLogitFnV2Test (line 64) | class DNNLogitFnV2Test(dnn_testing_utils.BaseDNNLogitFnTest, tf.test.Tes...
    method __init__ (line 66) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNWarmStartingV2Test (line 72) | class DNNWarmStartingV2Test(dnn_testing_utils.BaseDNNWarmStartingTest,
    method __init__ (line 75) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierEvaluateV2Test (line 81) | class DNNClassifierEvaluateV2Test(
    method __init__ (line 84) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierPredictV2Test (line 90) | class DNNClassifierPredictV2Test(dnn_testing_utils.BaseDNNClassifierPred...
    method __init__ (line 93) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierTrainV2Test (line 99) | class DNNClassifierTrainV2Test(dnn_testing_utils.BaseDNNClassifierTrainT...
    method __init__ (line 102) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 108) | def _dnn_regressor_fn(*args, **kwargs):
  class DNNRegressorEvaluateV2Test (line 112) | class DNNRegressorEvaluateV2Test(dnn_testing_utils.BaseDNNRegressorEvalu...
    method __init__ (line 115) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorPredictV2Test (line 121) | class DNNRegressorPredictV2Test(dnn_testing_utils.BaseDNNRegressorPredic...
    method __init__ (line 124) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorTrainV2Test (line 130) | class DNNRegressorTrainV2Test(dnn_testing_utils.BaseDNNRegressorTrainTest,
    method __init__ (line 133) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _queue_parsed_features (line 139) | def _queue_parsed_features(feature_map):
  class DNNRegressorIntegrationTest (line 154) | class DNNRegressorIntegrationTest(tf.test.TestCase, parameterized.TestCa...
    method setUp (line 156) | def setUp(self):
    method tearDown (line 159) | def tearDown(self):
    method _test_complete_flow (line 164) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 200) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 226) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 250) | def test_input_fn_from_parse_example(self):
  class DNNClassifierIntegrationTest (line 308) | class DNNClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 310) | def setUp(self):
    method tearDown (line 313) | def tearDown(self):
    method _as_label (line 318) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 321) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 357) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 386) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 411) | def test_input_fn_from_parse_example(self):
  class DNNTrainingMode (line 472) | class DNNTrainingMode(tf.test.TestCase):
    method setUp (line 475) | def setUp(self):
    method tearDown (line 480) | def tearDown(self):
    method _create_data (line 485) | def _create_data(self):
    method _get_estimator (line 490) | def _get_estimator(self):
    method test_train_vs_eval_mode (line 500) | def test_train_vs_eval_mode(self):

FILE: tensorflow_estimator/python/estimator/canned/dnn_testing_utils.py
  function assert_close (line 59) | def assert_close(expected, actual, rtol=1e-04, message='', name='assert_...
  function create_checkpoint (line 75) | def create_checkpoint(weights_and_biases,
  function mock_head (line 123) | def mock_head(testcase, hidden_units, logits_dimension, expected_logits):
  function mock_optimizer (line 191) | def mock_optimizer(testcase, hidden_units, expected_loss=None):
  class BaseDNNModelFnTest (line 242) | class BaseDNNModelFnTest(object):
    method __init__ (line 245) | def __init__(self, dnn_model_fn, fc_impl=feature_column_v2):
    method setUp (line 249) | def setUp(self):
    method tearDown (line 252) | def tearDown(self):
    method _test_logits (line 257) | def _test_logits(self, mode, hidden_units, logits_dimension, inputs,
    method test_one_dim_logits (line 289) | def test_one_dim_logits(self):
    method test_multi_dim_logits (line 313) | def test_multi_dim_logits(self):
    method test_multi_example_multi_dim_logits (line 338) | def test_multi_example_multi_dim_logits(self):
    method test_multi_dim_input_one_dim_logits (line 367) | def test_multi_dim_input_one_dim_logits(self):
    method test_multi_dim_input_multi_dim_logits (line 392) | def test_multi_dim_input_multi_dim_logits(self):
    method test_multi_feature_column_multi_dim_logits (line 417) | def test_multi_feature_column_multi_dim_logits(self):
    method test_multi_feature_column_mix_multi_dim_logits (line 468) | def test_multi_feature_column_mix_multi_dim_logits(self):
    method test_features_tensor_raises_value_error (line 519) | def test_features_tensor_raises_value_error(self):
  class BaseDNNLogitFnTest (line 547) | class BaseDNNLogitFnTest(object):
    method __init__ (line 550) | def __init__(self, dnn_logit_fn_builder, fc_impl=feature_column_v2):
    method setUp (line 554) | def setUp(self):
    method tearDown (line 557) | def tearDown(self):
    method _test_logits (line 562) | def _test_logits(self,
    method test_one_dim_logits (line 589) | def test_one_dim_logits(self):
    method test_one_dim_logits_with_batch_norm (line 612) | def test_one_dim_logits_with_batch_norm(self):
    method test_multi_dim_logits (line 677) | def test_multi_dim_logits(self):
    method test_multi_example_multi_dim_logits (line 701) | def test_multi_example_multi_dim_logits(self):
    method test_multi_dim_input_one_dim_logits (line 729) | def test_multi_dim_input_one_dim_logits(self):
    method test_multi_dim_input_multi_dim_logits (line 754) | def test_multi_dim_input_multi_dim_logits(self):
    method test_multi_feature_column_multi_dim_logits (line 778) | def test_multi_feature_column_multi_dim_logits(self):
    method test_multi_feature_column_mix_multi_dim_logits (line 822) | def test_multi_feature_column_mix_multi_dim_logits(self):
  class BaseDNNWarmStartingTest (line 867) | class BaseDNNWarmStartingTest(object):
    method __init__ (line 869) | def __init__(self,
    method setUp (line 877) | def setUp(self):
    method tearDown (line 895) | def tearDown(self):
    method assertAllNotClose (line 900) | def assertAllNotClose(self, t1, t2):
    method test_classifier_basic_warm_starting (line 911) | def test_classifier_basic_warm_starting(self):
    method test_regressor_basic_warm_starting (line 949) | def test_regressor_basic_warm_starting(self):
    method test_warm_starting_selective_variables (line 985) | def test_warm_starting_selective_variables(self):
    method test_warm_starting_with_vocab_remapping (line 1034) | def test_warm_starting_with_vocab_remapping(self):
    method test_warm_starting_with_naming_change (line 1128) | def test_warm_starting_with_naming_change(self):
  class BaseDNNClassifierEvaluateTest (line 1183) | class BaseDNNClassifierEvaluateTest(object):
    method __init__ (line 1185) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column_v2):
    method setUp (line 1189) | def setUp(self):
    method tearDown (line 1192) | def tearDown(self):
    method test_one_dim (line 1197) | def test_one_dim(self):
    method test_multi_dim (line 1250) | def test_multi_dim(self):
    method test_float_labels (line 1287) | def test_float_labels(self):
    method test_multi_dim_weights (line 1315) | def test_multi_dim_weights(self):
  class BaseDNNRegressorEvaluateTest (line 1346) | class BaseDNNRegressorEvaluateTest(object):
    method __init__ (line 1348) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column_v2):
    method setUp (line 1352) | def setUp(self):
    method tearDown (line 1355) | def tearDown(self):
    method test_one_dim (line 1360) | def test_one_dim(self):
    method test_multi_dim (line 1392) | def test_multi_dim(self):
    method test_multi_dim_weights (line 1428) | def test_multi_dim_weights(self):
  class BaseDNNClassifierPredictTest (line 1458) | class BaseDNNClassifierPredictTest(object):
    method __init__ (line 1460) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column_v2):
    method setUp (line 1464) | def setUp(self):
    method tearDown (line 1467) | def tearDown(self):
    method _test_one_dim (line 1472) | def _test_one_dim(self, label_vocabulary, label_output_fn):
    method test_one_dim_without_label_vocabulary (line 1513) | def test_one_dim_without_label_vocabulary(self):
    method test_one_dim_with_label_vocabulary (line 1517) | def test_one_dim_with_label_vocabulary(self):
    method _test_multi_dim_with_3_classes (line 1523) | def _test_multi_dim_with_3_classes(self, label_vocabulary, label_outpu...
    method test_multi_dim_with_3_classes_but_no_label_vocab (line 1577) | def test_multi_dim_with_3_classes_but_no_label_vocab(self):
    method test_multi_dim_with_3_classes_and_label_vocab (line 1581) | def test_multi_dim_with_3_classes_and_label_vocab(self):
  class BaseDNNRegressorPredictTest (line 1588) | class BaseDNNRegressorPredictTest(object):
    method __init__ (line 1590) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column_v2):
    method setUp (line 1594) | def setUp(self):
    method tearDown (line 1597) | def tearDown(self):
    method test_one_dim (line 1602) | def test_one_dim(self):
    method test_multi_dim (line 1626) | def test_multi_dim(self):
  class _SummaryHook (line 1655) | class _SummaryHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 1658) | def __init__(self):
    method begin (line 1661) | def begin(self):
    method before_run (line 1664) | def before_run(self, run_context):
    method after_run (line 1667) | def after_run(self, run_context, run_values):
    method summaries (line 1672) | def summaries(self):
  function _assert_checkpoint (line 1676) | def _assert_checkpoint(testcase, global_step, input_units, hidden_units,
  function _assert_simple_summary (line 1712) | def _assert_simple_summary(testcase, expected_values, actual_summary):
  class BaseDNNClassifierTrainTest (line 1728) | class BaseDNNClassifierTrainTest(object):
    method __init__ (line 1730) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column_v2):
    method setUp (line 1734) | def setUp(self):
    method tearDown (line 1737) | def tearDown(self):
    method test_from_scratch_with_default_optimizer_binary (line 1742) | def test_from_scratch_with_default_optimizer_binary(self):
    method test_from_scratch_with_default_optimizer_multi_class (line 1763) | def test_from_scratch_with_default_optimizer_multi_class(self):
    method test_from_scratch_validate_summary (line 1786) | def test_from_scratch_validate_summary(self):
    method test_binary_classification (line 1820) | def test_binary_classification(self):
    method test_binary_classification_float_labels (line 1872) | def test_binary_classification_float_labels(self):
    method test_multi_class (line 1904) | def test_multi_class(self):
  class BaseDNNRegressorTrainTest (line 1959) | class BaseDNNRegressorTrainTest(object):
    method __init__ (line 1961) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column_v2):
    method setUp (line 1965) | def setUp(self):
    method tearDown (line 1968) | def tearDown(self):
    method test_from_scratch_with_default_optimizer (line 1973) | def test_from_scratch_with_default_optimizer(self):
    method test_from_scratch (line 1994) | def test_from_scratch(self):
    method test_one_dim (line 2028) | def test_one_dim(self):
    method test_multi_dim (line 2081) | def test_multi_dim(self):

FILE: tensorflow_estimator/python/estimator/canned/head.py
  function _summary_key (line 55) | def _summary_key(head_name, val):
  function _create_eval_metrics_tuple (line 59) | def _create_eval_metrics_tuple(fn, kwargs):
  class _Head (line 90) | class _Head(object):
    method name (line 147) | def name(self):
    method logits_dimension (line 156) | def logits_dimension(self):
    method create_loss (line 167) | def create_loss(self, features, mode, logits, labels):
    method create_estimator_spec (line 195) | def create_estimator_spec(self,
    method _create_tpu_estimator_spec (line 242) | def _create_tpu_estimator_spec(self,
  function _check_dense_labels_match_logits_and_reshape (line 276) | def _check_dense_labels_match_logits_and_reshape(labels, logits,
  function _get_weights_and_check_match_logits (line 355) | def _get_weights_and_check_match_logits(features,
  function _check_logits_final_dim (line 444) | def _check_logits_final_dim(logits, expected_logits_dimension):
  function _validate_loss_fn_args (line 473) | def _validate_loss_fn_args(loss_fn):
  function _validate_n_classes (line 495) | def _validate_n_classes(n_classes):
  function _call_loss_fn (line 521) | def _call_loss_fn(loss_fn, labels, logits, features, expected_loss_dim=1):
  function _indicator_labels_mean (line 560) | def _indicator_labels_mean(labels, weights=None, name=None):
  function _all_class_ids (line 568) | def _all_class_ids(logits, n_classes):
  function _all_classes (line 576) | def _all_classes(logits, n_classes, label_vocabulary=None):
  function _classification_output (line 587) | def _classification_output(scores, n_classes, label_vocabulary=None):
  function _accuracy_baseline (line 602) | def _accuracy_baseline(labels_mean):
  function _predictions_mean (line 619) | def _predictions_mean(predictions, weights=None, name=None):
  function _auc (line 629) | def _auc(labels, predictions, weights=None, curve='ROC', name=None):
  function _accuracy_at_threshold (line 643) | def _accuracy_at_threshold(labels, predictions, weights, threshold, name...
  function _precision_at_threshold (line 655) | def _precision_at_threshold(labels, predictions, weights, threshold, nam...
  function _recall_at_threshold (line 668) | def _recall_at_threshold(labels, predictions, weights, threshold, name=N...
  function _multi_class_head_with_softmax_cross_entropy_loss (line 681) | def _multi_class_head_with_softmax_cross_entropy_loss(
  class _MultiClassHeadWithSoftmaxCrossEntropyLoss (line 755) | class _MultiClassHeadWithSoftmaxCrossEntropyLoss(_Head):
    method __init__ (line 758) | def __init__(self,
    method name (line 775) | def name(self):
    method logits_dimension (line 779) | def logits_dimension(self):
    method _eval_metric_ops (line 782) | def _eval_metric_ops(self, labels, class_ids, weights, unreduced_loss,
    method _label_ids (line 808) | def _label_ids(self, labels):
    method create_loss (line 825) | def create_loss(self, features, mode, logits, labels):
    method _create_tpu_estimator_spec (line 856) | def _create_tpu_estimator_spec(self,
  function _binary_logistic_head_with_sigmoid_cross_entropy_loss (line 999) | def _binary_logistic_head_with_sigmoid_cross_entropy_loss(
  class _BinaryLogisticHeadWithSigmoidCrossEntropyLoss (line 1084) | class _BinaryLogisticHeadWithSigmoidCrossEntropyLoss(_Head):
    method __init__ (line 1087) | def __init__(self,
    method name (line 1102) | def name(self):
    method logits_dimension (line 1106) | def logits_dimension(self):
    method _eval_metric_ops (line 1109) | def _eval_metric_ops(self, labels, logits, logistic, class_ids, weights,
    method create_loss (line 1195) | def create_loss(self, features, mode, logits, labels):
    method _create_tpu_estimator_spec (line 1227) | def _create_tpu_estimator_spec(self,
  function _regression_head (line 1378) | def _regression_head(weight_column=None,
  class _RegressionHeadWithMeanSquaredErrorLoss (line 1447) | class _RegressionHeadWithMeanSquaredErrorLoss(_Head):
    method __init__ (line 1450) | def __init__(self,
    method name (line 1468) | def name(self):
    method logits_dimension (line 1472) | def logits_dimension(self):
    method create_loss (line 1475) | def create_loss(self, features, mode, logits, labels):
    method _eval_metric_ops (line 1509) | def _eval_metric_ops(self, predicted_value, labels, weights, unreduced...
    method _create_tpu_estimator_spec (line 1532) | def _create_tpu_estimator_spec(self,
  function _append_update_ops (line 1657) | def _append_update_ops(train_op):
  function _assert_range (line 1665) | def _assert_range(labels, n_classes, message=None):
  function _binary_logistic_or_multi_class_head (line 1677) | def _binary_logistic_or_multi_class_head(n_classes, weight_column,

FILE: tensorflow_estimator/python/estimator/canned/head_test.py
  function _initialize_variables (line 36) | def _initialize_variables(test_case, scaffold):
  function _assert_simple_summaries (line 47) | def _assert_simple_summaries(test_case,
  function _assert_no_hooks (line 67) | def _assert_no_hooks(test_case, spec):
  function _sigmoid (line 72) | def _sigmoid(logits):
  class CreateEstimatorSpecTest (line 77) | class CreateEstimatorSpecTest(tf.test.TestCase):
    class _HeadWithTPUSupport (line 79) | class _HeadWithTPUSupport(head_lib._Head):
      method name (line 82) | def name(self):
      method logits_dimension (line 85) | def logits_dimension(self):
      method create_loss (line 88) | def create_loss(self, features, mode, logits, labels):
      method _create_tpu_estimator_spec (line 91) | def _create_tpu_estimator_spec(self,
    class _HeadWithOutTPUSupport (line 102) | class _HeadWithOutTPUSupport(head_lib._Head):
      method name (line 105) | def name(self):
      method logits_dimension (line 108) | def logits_dimension(self):
      method create_loss (line 111) | def create_loss(self, features, mode, logits, labels):
      method create_estimator_spec (line 114) | def create_estimator_spec(self,
    class _InvalidHead (line 125) | class _InvalidHead(head_lib._Head):
      method name (line 128) | def name(self):
      method logits_dimension (line 131) | def logits_dimension(self):
      method create_loss (line 134) | def create_loss(self, features, mode, logits, labels):
    method test_head_override_tpu_estimator_spec (line 137) | def test_head_override_tpu_estimator_spec(self):
    method test_head_override_estimator_spec (line 147) | def test_head_override_estimator_spec(self):
    method test_invalid_head_class (line 158) | def test_invalid_head_class(self):
  class MultiClassHeadWithSoftmaxCrossEntropyLoss (line 173) | class MultiClassHeadWithSoftmaxCrossEntropyLoss(tf.test.TestCase):
    method setUp (line 175) | def setUp(self):
    method test_n_classes_is_none (line 178) | def test_n_classes_is_none(self):
    method test_n_classes_is_2 (line 182) | def test_n_classes_is_2(self):
    method test_invalid_loss_reduction (line 186) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 195) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 206) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 217) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 225) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits_shape (line 235) | def test_invalid_logits_shape(self):
    method test_invalid_labels_shape (line 261) | def test_invalid_labels_shape(self):
    method test_invalid_labels_type (line 297) | def test_invalid_labels_type(self):
    method test_invalid_labels_values (line 326) | def test_invalid_labels_values(self):
    method test_invalid_labels_sparse_tensor (line 356) | def test_invalid_labels_sparse_tensor(self):
    method test_incompatible_labels_shape (line 375) | def test_incompatible_labels_shape(self):
    method test_name (line 416) | def test_name(self):
    method test_predict (line 421) | def test_predict(self):
    method test_predict_with_tensor_n_classes (line 471) | def test_predict_with_tensor_n_classes(self):
    method test_predict_with_vocabulary_list (line 521) | def test_predict_with_vocabulary_list(self):
    method test_weight_should_not_impact_prediction (line 544) | def test_weight_should_not_impact_prediction(self):
    method test_eval_create_loss (line 570) | def test_eval_create_loss(self):
    method test_eval_create_loss_loss_fn (line 587) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 613) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_eval_labels_none (line 639) | def test_eval_labels_none(self):
    method test_eval (line 652) | def test_eval(self):
    method test_eval_metric_ops_with_head_name (line 693) | def test_eval_metric_ops_with_head_name(self):
    method test_eval_with_regularization_losses (line 710) | def test_eval_with_regularization_losses(self):
    method test_eval_with_label_vocabulary_create_loss (line 756) | def test_eval_with_label_vocabulary_create_loss(self):
    method test_eval_with_label_vocabulary (line 772) | def test_eval_with_label_vocabulary(self):
    method test_weighted_multi_example_eval (line 805) | def test_weighted_multi_example_eval(self):
    method test_train_create_loss (line 856) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 881) | def test_train_create_loss_loss_reduction(self):
    method test_train_labels_none (line 908) | def test_train_labels_none(self):
    method test_train (line 926) | def test_train(self):
    method test_train_with_optimizer (line 973) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 1009) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 1040) | def test_train_summaries_with_head_name(self):
    method test_train_with_regularization_losses (line 1077) | def test_train_with_regularization_losses(self):
    method test_train_one_dim_create_loss (line 1127) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 1160) | def test_train_one_dim(self):
    method test_train_with_vocabulary_create_loss (line 1219) | def test_train_with_vocabulary_create_loss(self):
    method test_train_with_vocabulary (line 1236) | def test_train_with_vocabulary(self):
    method test_weighted_multi_example_train (line 1264) | def test_weighted_multi_example_train(self):
    method test_multi_dim_weighted_train_create_loss (line 1324) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 1354) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 1391) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 1417) | def test_multi_dim_train_weights_wrong_outer_dim(self):
    method test_multi_dim_weighted_eval (line 1445) | def test_multi_dim_weighted_eval(self):
  class BinaryLogisticHeadWithSigmoidCrossEntropyLossTest (line 1488) | class BinaryLogisticHeadWithSigmoidCrossEntropyLossTest(tf.test.TestCase):
    method setUp (line 1490) | def setUp(self):
    method test_threshold_too_small (line 1493) | def test_threshold_too_small(self):
    method test_threshold_too_large (line 1498) | def test_threshold_too_large(self):
    method test_invalid_loss_reduction (line 1503) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 1512) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 1523) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 1534) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 1541) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits_shape (line 1551) | def test_invalid_logits_shape(self):
    method test_invalid_labels_shape (line 1576) | def test_invalid_labels_shape(self):
    method test_incompatible_labels_shape (line 1609) | def test_incompatible_labels_shape(self):
    method test_name (line 1658) | def test_name(self):
    method test_predict (line 1663) | def test_predict(self):
    method test_predict_with_vocabulary_list (line 1721) | def test_predict_with_vocabulary_list(self):
    method test_eval_create_loss (line 1739) | def test_eval_create_loss(self):
    method test_eval_labels_none (line 1755) | def test_eval_labels_none(self):
    method test_eval (line 1767) | def test_eval(self):
    method test_eval_metric_ops_with_head_name (line 1812) | def test_eval_metric_ops_with_head_name(self):
    method test_eval_with_regularization_losses (line 1835) | def test_eval_with_regularization_losses(self):
    method test_eval_with_vocabulary_list_create_loss (line 1884) | def test_eval_with_vocabulary_list_create_loss(self):
    method test_eval_with_vocabulary_list (line 1897) | def test_eval_with_vocabulary_list(self):
    method test_eval_with_thresholds_create_loss (line 1916) | def test_eval_with_thresholds_create_loss(self):
    method test_eval_with_thresholds (line 1937) | def test_eval_with_thresholds(self):
    method test_train_create_loss (line 1990) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 2011) | def test_train_create_loss_loss_reduction(self):
    method test_eval_create_loss_loss_fn (line 2034) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 2060) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_train_labels_none (line 2086) | def test_train_labels_none(self):
    method test_train (line 2103) | def test_train(self):
    method test_train_with_optimizer (line 2152) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 2186) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 2216) | def test_train_summaries_with_head_name(self):
    method test_train_with_regularization_losses (line 2253) | def test_train_with_regularization_losses(self):
    method test_float_labels_invalid_values (line 2300) | def test_float_labels_invalid_values(self):
    method test_float_labels_train_create_loss (line 2312) | def test_float_labels_train_create_loss(self):
    method test_float_labels_train (line 2333) | def test_float_labels_train(self):
    method test_float_labels_eval_create_loss (line 2368) | def test_float_labels_eval_create_loss(self):
    method test_float_labels_eval (line 2389) | def test_float_labels_eval(self):
    method test_weighted_multi_example_predict (line 2416) | def test_weighted_multi_example_predict(self):
    method test_weighted_multi_example_eval (line 2449) | def test_weighted_multi_example_eval(self):
    method test_train_one_dim_create_loss (line 2505) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 2538) | def test_train_one_dim(self):
    method test_weighted_multi_example_train (line 2595) | def test_weighted_multi_example_train(self):
    method test_multi_dim_weighted_train_create_loss (line 2649) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 2678) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 2715) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 2741) | def test_multi_dim_train_weights_wrong_outer_dim(self):
    method test_multi_dim_weighted_eval (line 2770) | def test_multi_dim_weighted_eval(self):
  class RegressionHead (line 2824) | class RegressionHead(tf.test.TestCase):
    method setUp (line 2826) | def setUp(self):
    method test_invalid_label_dimension (line 2829) | def test_invalid_label_dimension(self):
    method test_invalid_loss_reduction (line 2835) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 2843) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 2853) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 2863) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 2869) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits (line 2878) | def test_invalid_logits(self):
    method test_incompatible_labels_eval (line 2901) | def test_incompatible_labels_eval(self):
    method test_incompatible_labels_train (line 2950) | def test_incompatible_labels_train(self):
    method test_name (line 3001) | def test_name(self):
    method test_predict (line 3005) | def test_predict(self):
    method test_predict_with_inverse_link_fn (line 3039) | def test_predict_with_inverse_link_fn(self):
    method test_eval_create_loss (line 3081) | def test_eval_create_loss(self):
    method test_eval_create_loss_loss_fn (line 3094) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 3119) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_eval_labels_none (line 3144) | def test_eval_labels_none(self):
    method test_eval (line 3156) | def test_eval(self):
    method test_eval_metric_ops_with_head_name_for_regression (line 3197) | def test_eval_metric_ops_with_head_name_for_regression(self):
    method test_eval_with_regularization_losses (line 3214) | def test_eval_with_regularization_losses(self):
    method test_train_create_loss (line 3261) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 3281) | def test_train_create_loss_loss_reduction(self):
    method test_train_labels_none (line 3303) | def test_train_labels_none(self):
    method test_train (line 3320) | def test_train(self):
    method test_train_with_optimizer (line 3375) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 3410) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 3440) | def test_train_summaries_with_head_name(self):
    method test_train_with_regularization_losses (line 3479) | def test_train_with_regularization_losses(self):
    method test_weighted_multi_example_eval (line 3529) | def test_weighted_multi_example_eval(self):
    method test_weight_with_numeric_column (line 3575) | def test_weight_with_numeric_column(self):
    method test_weighted_multi_example_train (line 3601) | def test_weighted_multi_example_train(self):
    method test_train_one_dim_create_loss (line 3658) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 3684) | def test_train_one_dim(self):
    method test_weighted_multi_value_eval_create_loss (line 3746) | def test_weighted_multi_value_eval_create_loss(self):
    method test_weighted_multi_value_eval (line 3765) | def test_weighted_multi_value_eval(self):
    method test_weighted_multi_value_train_create_loss (line 3811) | def test_weighted_multi_value_train_create_loss(self):
    method test_weighted_multi_value_train (line 3830) | def test_weighted_multi_value_train(self):
    method test_weighted_multi_batch_eval (line 3890) | def test_weighted_multi_batch_eval(self):
    method test_weighted_multi_batch_train (line 3962) | def test_weighted_multi_batch_train(self):
    method test_multi_dim_weighted_train_create_loss (line 4009) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 4040) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 4073) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 4102) | def test_multi_dim_train_weights_wrong_outer_dim(self):

FILE: tensorflow_estimator/python/estimator/canned/kmeans.py
  class _LossRelativeChangeHook (line 36) | class _LossRelativeChangeHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 39) | def __init__(self, loss_tensor, tolerance):
    method before_run (line 50) | def before_run(self, run_context):
    method after_run (line 54) | def after_run(self, run_context, run_values):
  class _InitializeClustersHook (line 65) | class _InitializeClustersHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 72) | def __init__(self, init_op, is_initialized_var, is_chief):
    method after_create_session (line 86) | def after_create_session(self, session, coord):
  function _parse_features_if_necessary (line 102) | def _parse_features_if_necessary(features, feature_columns):
  class _ModelFn (line 130) | class _ModelFn(object):
    method __init__ (line 133) | def __init__(self, num_clusters, initial_clusters, distance_metric, seed,
    method model_fn (line 147) | def model_fn(self, features, mode, config):
  class KMeansClustering (line 241) | class KMeansClustering(estimator.Estimator):
    method __init__ (line 315) | def __init__(self,
    method _predict_one_key (line 423) | def _predict_one_key(self, input_fn, predict_key):
    method predict_cluster_index (line 427) | def predict_cluster_index(self, input_fn):
    method score (line 440) | def score(self, input_fn):
    method transform (line 456) | def transform(self, input_fn):
    method cluster_centers (line 477) | def cluster_centers(self):

FILE: tensorflow_estimator/python/estimator/canned/kmeans_test.py
  function normalize (line 36) | def normalize(x):
  function cosine_similarity (line 40) | def cosine_similarity(x, y):
  function make_random_centers (line 44) | def make_random_centers(num_centers, num_dims, center_norm=500):
  function make_random_points (line 49) | def make_random_points(centers, num_points, max_offset=20):
  class KMeansTestBase (line 58) | class KMeansTestBase(tf.test.TestCase):
    method input_fn (line 60) | def input_fn(self,
    method config (line 106) | def config(tf_random_seed):
    method initial_clusters (line 110) | def initial_clusters(self):
    method batch_size (line 114) | def batch_size(self):
    method use_mini_batch (line 118) | def use_mini_batch(self):
    method mini_batch_steps_per_iteration (line 122) | def mini_batch_steps_per_iteration(self):
  class KMeansTest (line 127) | class KMeansTest(KMeansTestBase):
    method setUp (line 129) | def setUp(self):
    method _kmeans (line 139) | def _kmeans(self, relative_tolerance=None):
    method test_clusters (line 149) | def test_clusters(self):
    method test_fit (line 155) | def test_fit(self):
    method test_monitor (line 165) | def test_monitor(self):
    method _infer_helper (line 186) | def _infer_helper(self, kmeans, clusters, num_points):
    method test_infer (line 208) | def test_infer(self):
    method _parse_feature_dict_helper (line 219) | def _parse_feature_dict_helper(self, features, parsed_feature_dict):
    method test_parse_features (line 228) | def test_parse_features(self):
  class KMeansTestMultiStageInit (line 262) | class KMeansTestMultiStageInit(KMeansTestBase):
    method test_random (line 264) | def test_random(self):
    method test_kmeans_plus_plus_batch_just_right (line 281) | def test_kmeans_plus_plus_batch_just_right(self):
    method test_kmeans_plus_plus_batch_too_small (line 297) | def test_kmeans_plus_plus_batch_too_small(self):
  class MiniBatchKMeansTest (line 315) | class MiniBatchKMeansTest(KMeansTest):
    method batch_size (line 318) | def batch_size(self):
    method use_mini_batch (line 322) | def use_mini_batch(self):
  class FullBatchAsyncKMeansTest (line 327) | class FullBatchAsyncKMeansTest(KMeansTest):
    method batch_size (line 330) | def batch_size(self):
    method use_mini_batch (line 334) | def use_mini_batch(self):
    method mini_batch_steps_per_iteration (line 338) | def mini_batch_steps_per_iteration(self):
  class KMeansCosineDistanceTest (line 343) | class KMeansCosineDistanceTest(KMeansTestBase):
    method setUp (line 345) | def setUp(self):
    method test_fit (line 370) | def test_fit(self):
    method test_transform (line 378) | def test_transform(self):
    method test_predict (line 387) | def test_predict(self):
    method test_predict_kmeans_plus_plus (line 407) | def test_predict_kmeans_plus_plus(self):
  class MiniBatchKMeansCosineTest (line 449) | class MiniBatchKMeansCosineTest(KMeansCosineDistanceTest):
    method batch_size (line 452) | def batch_size(self):
    method use_mini_batch (line 456) | def use_mini_batch(self):
  class FullBatchAsyncKMeansCosineTest (line 461) | class FullBatchAsyncKMeansCosineTest(KMeansCosineDistanceTest):
    method batch_size (line 464) | def batch_size(self):
    method use_mini_batch (line 468) | def use_mini_batch(self):
    method mini_batch_steps_per_iteration (line 472) | def mini_batch_steps_per_iteration(self):
  class KMeansBenchmark (line 476) | class KMeansBenchmark(benchmark.Benchmark):
    method SetUp (line 479) | def SetUp(self,
    method _report (line 494) | def _report(self, num_iters, start, end, scores):
    method _fit (line 504) | def _fit(self, num_iters=10):
    method benchmark_01_2dim_5center_500point (line 507) | def benchmark_01_2dim_5center_500point(self):
    method benchmark_02_20dim_20center_10kpoint (line 511) | def benchmark_02_20dim_20center_10kpoint(self):
    method benchmark_03_100dim_50center_50kpoint (line 515) | def benchmark_03_100dim_50center_50kpoint(self):
    method benchmark_03_100dim_50center_50kpoint_unseparated (line 519) | def benchmark_03_100dim_50center_50kpoint_unseparated(self):
    method benchmark_04_100dim_500center_500kpoint (line 527) | def benchmark_04_100dim_500center_500kpoint(self):
    method benchmark_05_100dim_500center_500kpoint_unseparated (line 531) | def benchmark_05_100dim_500center_500kpoint_unseparated(self):
  class TensorflowKMeansBenchmark (line 540) | class TensorflowKMeansBenchmark(KMeansBenchmark):
    method _fit (line 542) | def _fit(self, num_iters=10):
  class SklearnKMeansBenchmark (line 562) | class SklearnKMeansBenchmark(KMeansBenchmark):
    method _fit (line 564) | def _fit(self, num_iters=10):
  class KMeansTestQueues (line 582) | class KMeansTestQueues(tf.test.TestCase):
    method input_fn (line 584) | def input_fn(self):
    method test_queues (line 600) | def test_queues(self):

FILE: tensorflow_estimator/python/estimator/canned/linear.py
  class LinearSDCA (line 47) | class LinearSDCA(object):
    method __init__ (line 107) | def __init__(self,
    method _prune_and_unique_sparse_ids (line 135) | def _prune_and_unique_sparse_ids(self, id_weight_pair):
    method get_train_step (line 177) | def get_train_step(self, state_manager, weight_column_name, loss_type,
  function _get_default_optimizer_v2 (line 242) | def _get_default_optimizer_v2(feature_columns):
  function _get_default_optimizer (line 247) | def _get_default_optimizer(feature_columns):
  function _get_expanded_variable_list (line 252) | def _get_expanded_variable_list(var_list):
  function _compute_fraction_of_zero (line 274) | def _compute_fraction_of_zero(variables):
  function linear_logit_fn_builder_v2 (line 311) | def linear_logit_fn_builder_v2(units, feature_columns, sparse_combiner='...
  function linear_logit_fn_builder (line 379) | def linear_logit_fn_builder(units, feature_columns, sparse_combiner='sum'):
  function _sdca_model_fn (line 451) | def _sdca_model_fn(features, labels, mode, head, feature_columns, optimi...
  class _SDCAUpdateWeightsHook (line 543) | class _SDCAUpdateWeightsHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 546) | def __init__(self, sdca_model, train_op):
    method begin (line 550) | def begin(self):
    method before_run (line 557) | def before_run(self, run_context):
  function _linear_model_fn_builder_v2 (line 562) | def _linear_model_fn_builder_v2(units,
  function _linear_model_fn_v2 (line 622) | def _linear_model_fn_v2(features,
  function _linear_model_fn (line 687) | def _linear_model_fn(features,
  function _validate_linear_sdca_optimizer_for_linear_classifier (line 756) | def _validate_linear_sdca_optimizer_for_linear_classifier(
  class LinearClassifierV2 (line 771) | class LinearClassifierV2(estimator.EstimatorV2):
    method __init__ (line 858) | def __init__(self,
  class LinearClassifier (line 953) | class LinearClassifier(estimator.Estimator):
    method __init__ (line 956) | def __init__(self,
  class LinearEstimatorV2 (line 999) | class LinearEstimatorV2(estimator.EstimatorV2):
    method __init__ (line 1078) | def __init__(self,
  class LinearEstimator (line 1131) | class LinearEstimator(estimator.Estimator):
    method __init__ (line 1134) | def __init__(self,
  function _validate_linear_sdca_optimizer_for_linear_regressor (line 1189) | def _validate_linear_sdca_optimizer_for_linear_regressor(
  class LinearRegressorV2 (line 1205) | class LinearRegressorV2(estimator.EstimatorV2):
    method __init__ (line 1292) | def __init__(self,
  class LinearRegressor (line 1371) | class LinearRegressor(estimator.Estimator):
    method __init__ (line 1374) | def __init__(self,
  class _LinearModelLayer (line 1417) | class _LinearModelLayer(tf_keras.layers.Layer):
    method __init__ (line 1420) | def __init__(self,
    method build (line 1443) | def build(self, _):
    method call (line 1483) | def call(self, features):
    method get_config (line 1513) | def get_config(self):
    method from_config (line 1529) | def from_config(cls, config, custom_objects=None):
  class LinearModel (line 1540) | class LinearModel(tf_keras.Model):
    method __init__ (line 1579) | def __init__(self,
    method call (line 1653) | def call(self, features):
    method bias (line 1672) | def bias(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_estimator_test.py
  function _linear_estimator_fn (line 37) | def _linear_estimator_fn(weight_column=None, label_dimension=1, **kwargs):
  function _linear_estimator_classifier_fn (line 48) | def _linear_estimator_classifier_fn(n_classes=3, **kwargs):
  class LinearEstimatorEvaluateTest (line 53) | class LinearEstimatorEvaluateTest(
    method __init__ (line 56) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorPredictTest (line 62) | class LinearEstimatorPredictTest(
    method __init__ (line 65) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorTrainTest (line 71) | class LinearEstimatorTrainTest(
    method __init__ (line 74) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorWarmStartingTest (line 80) | class LinearEstimatorWarmStartingTest(
    method __init__ (line 83) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorIntegrationTest (line 91) | class LinearEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 93) | def setUp(self):
    method tearDown (line 96) | def tearDown(self):
    method _test_complete_flow (line 101) | def _test_complete_flow(self,
    method _create_input_fn (line 143) | def _create_input_fn(self, label_dimension, batch_size):
    method test_numpy_input_fn (line 161) | def test_numpy_input_fn(self):
    method test_numpy_input_fn_with_optimizer_instance (line 175) | def test_numpy_input_fn_with_optimizer_instance(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_model_test.py
  function _initialized_session (line 33) | def _initialized_session(config=None):
  function get_linear_model_bias (line 40) | def get_linear_model_bias(name='linear_model'):
  function get_linear_model_column_var (line 45) | def get_linear_model_column_var(column, name='linear_model'):
  class BaseFeatureColumnForTests (line 50) | class BaseFeatureColumnForTests(tf.compat.v2.__internal__.feature_column...
    method parents (line 58) | def parents(self):
    method from_config (line 62) | def from_config(cls, config, custom_objects=None, columns_by_name=None):
    method get_config (line 65) | def get_config(self):
  class SortableFeatureColumnTest (line 69) | class SortableFeatureColumnTest(tf.test.TestCase):
    method test_linear_model (line 72) | def test_linear_model(self):
    method test_linear_model_sanitizes_scope_names (line 87) | def test_linear_model_sanitizes_scope_names(self):
  class BucketizedColumnTest (line 102) | class BucketizedColumnTest(tf.test.TestCase):
    method test_linear_model_one_input_value (line 104) | def test_linear_model_one_input_value(self):
    method test_linear_model_two_input_values (line 132) | def test_linear_model_two_input_values(self):
  class HashedCategoricalColumnTest (line 162) | class HashedCategoricalColumnTest(tf.test.TestCase):
    method test_linear_model (line 165) | def test_linear_model(self):
  class CrossedColumnTest (line 191) | class CrossedColumnTest(tf.test.TestCase):
    method test_linear_model (line 194) | def test_linear_model(self):
    method test_linear_model_with_weights (line 225) | def test_linear_model_with_weights(self):
  class LinearModelTest (line 289) | class LinearModelTest(tf.test.TestCase):
    method test_raises_if_empty_feature_columns (line 291) | def test_raises_if_empty_feature_columns(self):
    method test_should_be_feature_column (line 296) | def test_should_be_feature_column(self):
    method test_should_be_dense_or_categorical_column (line 300) | def test_should_be_dense_or_categorical_column(self):
    method test_does_not_support_dict_columns (line 323) | def test_does_not_support_dict_columns(self):
    method test_raises_if_duplicate_name (line 328) | def test_raises_if_duplicate_name(self):
    method test_not_dict_input_features (line 335) | def test_not_dict_input_features(self):
    method test_dense_bias (line 343) | def test_dense_bias(self):
    method test_sparse_bias (line 356) | def test_sparse_bias(self):
    method test_dense_and_sparse_bias (line 375) | def test_dense_and_sparse_bias(self):
    method test_dense_and_sparse_column (line 393) | def test_dense_and_sparse_column(self):
    method test_dense_multi_output (line 449) | def test_dense_multi_output(self):
    method test_sparse_multi_output (line 464) | def test_sparse_multi_output(self):
    method test_dense_multi_dimension (line 486) | def test_dense_multi_dimension(self):
    method test_sparse_multi_rank (line 498) | def test_sparse_multi_rank(self):
    method test_sparse_combiner (line 520) | def test_sparse_combiner(self):
    method test_sparse_combiner_sqrtn (line 536) | def test_sparse_combiner_sqrtn(self):
    method test_sparse_combiner_with_negative_weights (line 552) | def test_sparse_combiner_with_negative_weights(self):
    method test_dense_multi_dimension_multi_output (line 573) | def test_dense_multi_dimension_multi_output(self):
    method test_raises_if_shape_mismatch (line 588) | def test_raises_if_shape_mismatch(self):
    method test_dense_reshaping (line 598) | def test_dense_reshaping(self):
    method test_dense_multi_column (line 612) | def test_dense_multi_column(self):
    method test_dense_trainable_default (line 630) | def test_dense_trainable_default(self):
    method test_sparse_trainable_default (line 641) | def test_sparse_trainable_default(self):
    method test_dense_trainable_false (line 654) | def test_dense_trainable_false(self):
    method test_sparse_trainable_false (line 663) | def test_sparse_trainable_false(self):
    method test_column_order (line 674) | def test_column_order(self):
    method test_variable_names (line 710) | def test_variable_names(self):
    method test_fit_and_predict (line 740) | def test_fit_and_predict(self):
    method test_static_batch_size_mismatch (line 756) | def test_static_batch_size_mismatch(self):
    method test_subset_of_static_batch_size_mismatch (line 770) | def test_subset_of_static_batch_size_mismatch(self):
    method test_runtime_batch_size_mismatch (line 786) | def test_runtime_batch_size_mismatch(self):
    method test_runtime_batch_size_matches (line 802) | def test_runtime_batch_size_matches(self):
    method test_with_1d_sparse_tensor (line 821) | def test_with_1d_sparse_tensor(self):
    method test_with_1d_unknown_shape_sparse_tensor (line 861) | def test_with_1d_unknown_shape_sparse_tensor(self):
    method test_with_rank_0_feature (line 906) | def test_with_rank_0_feature(self):
    method test_multiple_linear_models (line 929) | def test_multiple_linear_models(self):
  class VocabularyFileCategoricalColumnTest (line 951) | class VocabularyFileCategoricalColumnTest(tf.test.TestCase):
    method setUp (line 953) | def setUp(self):
    method DISABLED_test_linear_model (line 968) | def DISABLED_test_linear_model(self):
  class VocabularyListCategoricalColumnTest (line 998) | class VocabularyListCategoricalColumnTest(tf.test.TestCase):
    method test_linear_model (line 1001) | def test_linear_model(self):
  class IdentityCategoricalColumnTest (line 1030) | class IdentityCategoricalColumnTest(tf.test.TestCase):
    method test_linear_model (line 1033) | def test_linear_model(self):
  class IndicatorColumnTest (line 1059) | class IndicatorColumnTest(tf.test.TestCase):
    method test_linear_model (line 1062) | def test_linear_model(self):
  class EmbeddingColumnTest (line 1086) | class EmbeddingColumnTest(tf.test.TestCase, parameterized.TestCase):
    method test_linear_model (line 1089) | def test_linear_model(self):
  class SharedEmbeddingColumnTest (line 1171) | class SharedEmbeddingColumnTest(tf.test.TestCase, parameterized.TestCase):
    method test_linear_model (line 1174) | def test_linear_model(self):
  class WeightedCategoricalColumnTest (line 1272) | class WeightedCategoricalColumnTest(tf.test.TestCase):
    method test_linear_model (line 1275) | def test_linear_model(self):
    method test_linear_model_mismatched_shape (line 1308) | def test_linear_model_mismatched_shape(self):
    method test_linear_model_mismatched_dense_values (line 1330) | def test_linear_model_mismatched_dense_values(self):
    method test_linear_model_mismatched_dense_shape (line 1354) | def test_linear_model_mismatched_dense_shape(self):
  class LinearModelLayerSerializationTest (line 1385) | class LinearModelLayerSerializationTest(tf.test.TestCase, parameterized....
    method test_get_config (line 1390) | def test_get_config(self, units, sparse_combiner, trainable, name):
    method test_from_config (line 1411) | def test_from_config(self, units, sparse_combiner, trainable, name):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/sdca_test.py
  class SDCAClassifierTest (line 26) | class SDCAClassifierTest(tf.test.TestCase):
    method testRealValuedFeatures (line 28) | def testRealValuedFeatures(self):
    method testRealValuedFeatureWithHigherDimension (line 50) | def testRealValuedFeatureWithHigherDimension(self):
    method testBucketizedFeatures (line 70) | def testBucketizedFeatures(self):
    method testSparseFeatures (line 95) | def testSparseFeatures(self):
    method testWeightedSparseFeatures (line 121) | def testWeightedSparseFeatures(self):
    method testWeightedSparseFeaturesOOVWithNoOOVBuckets (line 152) | def testWeightedSparseFeaturesOOVWithNoOOVBuckets(self):
    method testCrossedFeatures (line 184) | def testCrossedFeatures(self):
    method testMixedFeatures (line 213) | def testMixedFeatures(self):
    method testPartitionedVariables (line 251) | def testPartitionedVariables(self):
  class SDCARegressorTest (line 293) | class SDCARegressorTest(tf.test.TestCase):
    method testRealValuedLinearFeatures (line 295) | def testRealValuedLinearFeatures(self):
    method testMixedFeaturesArbitraryWeights (line 326) | def testMixedFeaturesArbitraryWeights(self):
    method testPartitionedVariables (line 364) | def testPartitionedVariables(self):
    method testSparseFeaturesWithL1Reg (line 404) | def testSparseFeaturesWithL1Reg(self):
    method testBiasOnly (line 478) | def testBiasOnly(self):
    method testBiasAndOtherColumns (line 509) | def testBiasAndOtherColumns(self):
    method testBiasAndOtherColumnsFabricatedCentered (line 571) | def testBiasAndOtherColumnsFabricatedCentered(self):
    method testUnknownBatchSize (line 625) | def testUnknownBatchSize(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sdca_ops.py
  class _SparseFeatureColumn (line 34) | class _SparseFeatureColumn(object):
    method __init__ (line 81) | def __init__(self, example_indices, feature_indices, feature_values):
    method example_indices (line 108) | def example_indices(self):
    method feature_indices (line 117) | def feature_indices(self):
    method feature_values (line 126) | def feature_values(self):
  class _SDCAModel (line 135) | class _SDCAModel(object):
    method __init__ (line 197) | def __init__(self, examples, variables, options):
    method _symmetric_l1_regularization (line 253) | def _symmetric_l1_regularization(self):
    method _symmetric_l2_regularization (line 256) | def _symmetric_l2_regularization(self):
    method _num_loss_partitions (line 259) | def _num_loss_partitions(self):
    method _adaptive (line 263) | def _adaptive(self):
    method _num_table_shards (line 267) | def _num_table_shards(self):
    method _create_slots (line 273) | def _create_slots(self):
    method _assert_specified (line 312) | def _assert_specified(self, items, check_in):
    method _assert_list (line 317) | def _assert_list(self, items, check_in):
    method _var_to_list (line 322) | def _var_to_list(self, var):
    method _l1_loss (line 328) | def _l1_loss(self):
    method _l2_loss (line 343) | def _l2_loss(self):
    method _convert_n_to_tensor (line 358) | def _convert_n_to_tensor(self, input_list, as_ref=False):
    method _get_first_dimension_size_statically (line 374) | def _get_first_dimension_size_statically(self, w, num_partitions):
    method _linear_predictions (line 381) | def _linear_predictions(self, examples):
    method predictions (line 412) | def predictions(self, examples):
    method _get_partitioned_update_ops (line 443) | def _get_partitioned_update_ops(self, v_num, num_partitions_by_var,
    method minimize (line 459) | def minimize(self, global_step=None, name=None):
    method update_weights (line 639) | def update_weights(self, train_op):
    method approximate_duality_gap (line 680) | def approximate_duality_gap(self):
    method unregularized_loss (line 710) | def unregularized_loss(self, examples):
    method regularized_loss (line 771) | def regularized_loss(self, examples):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sdca_ops_test.py
  function make_example_proto (line 38) | def make_example_proto(feature_dict, target, value=1.0):
  function make_example_dict (line 52) | def make_example_dict(example_protos, example_weights):
  function make_random_examples_and_variables_dicts (line 98) | def make_random_examples_and_variables_dicts(num_examples, dim, num_non_...
  function make_variable_dict (line 124) | def make_variable_dict(max_age, max_gender, num_shards=None, partitioned...
  function make_dense_examples_and_variables_dicts (line 145) | def make_dense_examples_and_variables_dicts(dense_features_values, weights,
  function get_binary_predictions_for_logistic (line 190) | def get_binary_predictions_for_logistic(predictions, cutoff=0.5):
  function get_binary_predictions_for_hinge (line 197) | def get_binary_predictions_for_hinge(predictions):
  class _SDCAModelTest (line 206) | class _SDCAModelTest(TensorFlowTestCase):
    method _single_threaded_test_session (line 209) | def _single_threaded_test_session(self):
  class SdcaWithLogisticLossTest (line 217) | class SdcaWithLogisticLossTest(_SDCAModelTest):
    method testSimple (line 220) | def testSimple(self):
    method testPartitionedPrimals (line 268) | def testPartitionedPrimals(self):
    method testSomePartitionedPrimals (line 316) | def testSomePartitionedPrimals(self):
    method testSparseRandom (line 377) | def testSparseRandom(self):
    method testSparseDuplicate (line 401) | def testSparseDuplicate(self):
    method testDistributedSimple (line 428) | def testDistributedSimple(self):
    method testSimpleNoL2 (line 504) | def testSimpleNoL2(self):
    method testSomeUnweightedExamples (line 529) | def testSomeUnweightedExamples(self):
    method testFractionalExampleLabel (line 583) | def testFractionalExampleLabel(self):
    method testImbalanced (line 612) | def testImbalanced(self):
    method testImbalancedWithExampleWeights (line 661) | def testImbalancedWithExampleWeights(self):
    method testInstancesOfOneClassOnly (line 701) | def testInstancesOfOneClassOnly(self):
    method testOutOfRangeSparseFeatures (line 740) | def testOutOfRangeSparseFeatures(self):
    method testOutOfRangeDenseFeatures (line 767) | def testOutOfRangeDenseFeatures(self):
    method testMissingFeature (line 789) | def testMissingFeature(self):
  class SdcaWithLinearLossTest (line 832) | class SdcaWithLinearLossTest(_SDCAModelTest):
    method testSimple (line 835) | def testSimple(self):
    method testL2Regularization (line 874) | def testL2Regularization(self):
    method testL1Regularization (line 922) | def testL1Regularization(self):
    method testFeatureValues (line 960) | def testFeatureValues(self):
    method testDenseFeaturesWithDefaultWeights (line 1007) | def testDenseFeaturesWithDefaultWeights(self):
    method testDenseFeaturesWithArbitraryWeights (line 1037) | def testDenseFeaturesWithArbitraryWeights(self):
  class SdcaWithHingeLossTest (line 1072) | class SdcaWithHingeLossTest(_SDCAModelTest):
    method testSimple (line 1075) | def testSimple(self):
    method testDenseFeaturesPerfectlySeparable (line 1123) | def testDenseFeaturesPerfectlySeparable(self):
    method testDenseFeaturesSeparableWithinMargins (line 1155) | def testDenseFeaturesSeparableWithinMargins(self):
    method testDenseFeaturesWeightedExamples (line 1186) | def testDenseFeaturesWeightedExamples(self):
  class SdcaWithSmoothHingeLossTest (line 1222) | class SdcaWithSmoothHingeLossTest(_SDCAModelTest):
    method testSimple (line 1225) | def testSimple(self):
  class SdcaWithPoissonLossTest (line 1275) | class SdcaWithPoissonLossTest(_SDCAModelTest):
    method testSimple (line 1278) | def testSimple(self):
  class SdcaFprintTest (line 1327) | class SdcaFprintTest(_SDCAModelTest):
    method testFprint (line 1337) | def testFprint(self):
  class _SparseFeatureColumnTest (line 1347) | class _SparseFeatureColumnTest(TensorFlowTestCase):
    method testBasic (line 1350) | def testBasic(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sharded_mutable_dense_hashtable.py
  class _MutableDenseHashTable (line 33) | class _MutableDenseHashTable(lookup_ops.LookupInterface):
    method __init__ (line 37) | def __init__(self,
    method _create_resource (line 104) | def _create_resource(self):
    method name (line 125) | def name(self):
    method size (line 128) | def size(self, name=None):
    method lookup (line 143) | def lookup(self, keys, name=None):
    method insert (line 169) | def insert(self, keys, values, name=None):
    method export (line 196) | def export(self, name=None):
    method _serialize_to_tensors (line 214) | def _serialize_to_tensors(self):
    method _restore_from_tensors (line 218) | def _restore_from_tensors(self, restored_tensors):
    class _Saveable (line 224) | class _Saveable(BaseSaverBuilder.SaveableObject):
      method __init__ (line 227) | def __init__(self, table, name):
      method restore (line 236) | def restore(self, restored_tensors, restored_shapes):
  class _ShardedMutableDenseHashTable (line 247) | class _ShardedMutableDenseHashTable(object):
    method __init__ (line 258) | def __init__(self,
    method name (line 289) | def name(self):
    method _num_shards (line 293) | def _num_shards(self):
    method table_shards (line 297) | def table_shards(self):
    method size (line 300) | def size(self, name=None):
    method _shard_indices (line 305) | def _shard_indices(self, keys):
    method _check_keys (line 314) | def _check_keys(self, keys):
    method lookup (line 319) | def lookup(self, keys, name=None):
    method insert (line 342) | def insert(self, keys, values, name=None):
    method export_sharded (line 359) | def export_sharded(self, name=None):

FILE: tensorflow_estimator/python/estimator/canned/linear_optimizer/python/utils/sharded_mutable_dense_hashtable_test.py
  class _ShardedMutableDenseHashTableTest (line 26) | class _ShardedMutableDenseHashTableTest(tf.test.TestCase):
    method testShardedMutableHashTable (line 29) | def testShardedMutableHashTable(self):
    method testShardedMutableHashTableVectors (line 54) | def testShardedMutableHashTableVectors(self):
    method testExportSharded (line 82) | def testExportSharded(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_test.py
  function _linear_regressor_fn (line 29) | def _linear_regressor_fn(*args, **kwargs):
  function _linear_classifier_fn (line 33) | def _linear_classifier_fn(*args, **kwargs):
  class LinearRegressorEvaluationV2Test (line 40) | class LinearRegressorEvaluationV2Test(
    method __init__ (line 43) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorPredictV2Test (line 49) | class LinearRegressorPredictV2Test(
    method __init__ (line 52) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorIntegrationV2Test (line 58) | class LinearRegressorIntegrationV2Test(
    method __init__ (line 61) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorTrainingV2Test (line 67) | class LinearRegressorTrainingV2Test(
    method __init__ (line 70) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierTrainingV2Test (line 79) | class LinearClassifierTrainingV2Test(
    method __init__ (line 82) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierEvaluationV2Test (line 90) | class LinearClassifierEvaluationV2Test(
    method __init__ (line 93) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierPredictV2Test (line 101) | class LinearClassifierPredictV2Test(
    method __init__ (line 104) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierIntegrationV2Test (line 112) | class LinearClassifierIntegrationV2Test(
    method __init__ (line 115) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearLogitFnV2Test (line 126) | class LinearLogitFnV2Test(linear_testing_utils.BaseLinearLogitFnTest,
    method __init__ (line 129) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearWarmStartingV2Test (line 138) | class LinearWarmStartingV2Test(linear_testing_utils.BaseLinearWarmStarti...
    method __init__ (line 141) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class ComputeFractionOfZeroTest (line 150) | class ComputeFractionOfZeroTest(tf.test.TestCase):
    method _assertSparsity (line 152) | def _assertSparsity(self, expected_sparsity, tensor):
    method test_small_float32 (line 156) | def test_small_float32(self):
    method test_small_int32 (line 162) | def test_small_int32(self):
    method test_small_float64 (line 166) | def test_small_float64(self):
    method test_small_int64 (line 170) | def test_small_int64(self):
    method test_nested (line 174) | def test_nested(self):
    method test_none (line 179) | def test_none(self):
    method test_empty (line 183) | def test_empty(self):
    method test_multiple_empty (line 189) | def test_multiple_empty(self):
    method test_some_empty (line 198) | def test_some_empty(self):
    method test_mixed_types (line 206) | def test_mixed_types(self):
    method test_2_27_zeros__using_512_MiB_of_ram (line 213) | def test_2_27_zeros__using_512_MiB_of_ram(self):
    method test_2_27_ones__using_512_MiB_of_ram (line 217) | def test_2_27_ones__using_512_MiB_of_ram(self):

FILE: tensorflow_estimator/python/estimator/canned/linear_testing_utils.py
  function assert_close (line 62) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 77) | def save_variables_to_ckpt(model_dir):
  function queue_parsed_features (line 84) | def queue_parsed_features(feature_map):
  function sorted_key_dict (line 99) | def sorted_key_dict(unsorted_dict):
  function sigmoid (line 103) | def sigmoid(x):
  function mock_optimizer (line 107) | def mock_optimizer(testcase, expected_loss=None):
  class BaseLinearRegressorEvaluationTest (line 142) | class BaseLinearRegressorEvaluationTest(object):
    method __init__ (line 144) | def __init__(self, linear_regressor_fn, fc_lib=feature_column_v2):
    method setUp (line 148) | def setUp(self):
    method tearDown (line 151) | def tearDown(self):
    method test_evaluation_for_simple_data (line 156) | def test_evaluation_for_simple_data(self):
    method test_evaluation_batch (line 182) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 212) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 246) | def test_evaluation_for_multi_dimensions(self):
    method test_evaluation_for_multiple_feature_columns (line 282) | def test_evaluation_for_multiple_feature_columns(self):
    method test_evaluation_for_multiple_feature_columns_mix (line 320) | def test_evaluation_for_multiple_feature_columns_mix(self):
  class BaseLinearRegressorPredictTest (line 360) | class BaseLinearRegressorPredictTest(object):
    method __init__ (line 362) | def __init__(self, linear_regressor_fn, fc_lib=feature_column_v2):
    method setUp (line 366) | def setUp(self):
    method tearDown (line 369) | def tearDown(self):
    method test_1d (line 374) | def test_1d(self):
    method testMultiDim (line 397) | def testMultiDim(self):
    method testTwoFeatureColumns (line 430) | def testTwoFeatureColumns(self):
    method testTwoFeatureColumnsMix (line 458) | def testTwoFeatureColumnsMix(self):
    method testSparseCombiner (line 483) | def testSparseCombiner(self):
  class BaseLinearRegressorIntegrationTest (line 546) | class BaseLinearRegressorIntegrationTest(object):
    method __init__ (line 548) | def __init__(self, linear_regressor_fn, fc_lib=feature_column_v2):
    method setUp (line 552) | def setUp(self):
    method tearDown (line 555) | def tearDown(self):
    method _test_complete_flow (line 560) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 592) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 628) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 657) | def test_input_fn_from_parse_example(self):
  class BaseLinearRegressorTrainingTest (line 718) | class BaseLinearRegressorTrainingTest(object):
    method __init__ (line 720) | def __init__(self, linear_regressor_fn, fc_lib=feature_column_v2):
    method setUp (line 724) | def setUp(self):
    method tearDown (line 727) | def tearDown(self):
    method _assert_checkpoint (line 732) | def _assert_checkpoint(self,
    method testFromScratchWithDefaultOptimizer (line 757) | def testFromScratchWithDefaultOptimizer(self):
    method testTrainWithOneDimLabel (line 773) | def testTrainWithOneDimLabel(self):
    method testTrainWithOneDimWeight (line 793) | def testTrainWithOneDimWeight(self):
    method testFromScratch (line 818) | def testFromScratch(self):
    method testFromCheckpoint (line 843) | def testFromCheckpoint(self):
    method testFromCheckpointMultiBatch (line 879) | def testFromCheckpointMultiBatch(self):
  class BaseLinearClassifierTrainingTest (line 920) | class BaseLinearClassifierTrainingTest(object):
    method __init__ (line 922) | def __init__(self, linear_classifier_fn, fc_lib=feature_column_v2):
    method setUp (line 926) | def setUp(self):
    method tearDown (line 929) | def tearDown(self):
    method _assert_checkpoint (line 933) | def _assert_checkpoint(self,
    method _testFromScratchWithDefaultOptimizer (line 962) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassesFromScratchWithDefaultOptimizer (line 978) | def testBinaryClassesFromScratchWithDefaultOptimizer(self):
    method testMultiClassesFromScratchWithDefaultOptimizer (line 981) | def testMultiClassesFromScratchWithDefaultOptimizer(self):
    method _testTrainWithTwoDimsLabel (line 984) | def _testTrainWithTwoDimsLabel(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsLabel (line 1005) | def testBinaryClassesTrainWithTwoDimsLabel(self):
    method testMultiClassesTrainWithTwoDimsLabel (line 1008) | def testMultiClassesTrainWithTwoDimsLabel(self):
    method _testTrainWithOneDimLabel (line 1011) | def _testTrainWithOneDimLabel(self, n_classes):
    method testBinaryClassesTrainWithOneDimLabel (line 1030) | def testBinaryClassesTrainWithOneDimLabel(self):
    method testMultiClassesTrainWithOneDimLabel (line 1033) | def testMultiClassesTrainWithOneDimLabel(self):
    method _testTrainWithTwoDimsWeight (line 1036) | def _testTrainWithTwoDimsWeight(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsWeight (line 1061) | def testBinaryClassesTrainWithTwoDimsWeight(self):
    method testMultiClassesTrainWithTwoDimsWeight (line 1064) | def testMultiClassesTrainWithTwoDimsWeight(self):
    method _testTrainWithOneDimWeight (line 1067) | def _testTrainWithOneDimWeight(self, n_classes):
    method testBinaryClassesTrainWithOneDimWeight (line 1090) | def testBinaryClassesTrainWithOneDimWeight(self):
    method testMultiClassesTrainWithOneDimWeight (line 1093) | def testMultiClassesTrainWithOneDimWeight(self):
    method _testFromScratch (line 1096) | def _testFromScratch(self, n_classes):
    method testBinaryClassesFromScratch (line 1132) | def testBinaryClassesFromScratch(self):
    method testMultiClassesFromScratch (line 1135) | def testMultiClassesFromScratch(self):
    method _testFromCheckpoint (line 1138) | def _testFromCheckpoint(self, n_classes):
    method testBinaryClassesFromCheckpoint (line 1197) | def testBinaryClassesFromCheckpoint(self):
    method testMultiClassesFromCheckpoint (line 1200) | def testMultiClassesFromCheckpoint(self):
    method _testFromCheckpointFloatLabels (line 1203) | def _testFromCheckpointFloatLabels(self, n_classes):
    method testBinaryClassesFromCheckpointFloatLabels (line 1242) | def testBinaryClassesFromCheckpointFloatLabels(self):
    method testMultiClassesFromCheckpointFloatLabels (line 1245) | def testMultiClassesFromCheckpointFloatLabels(self):
    method _testFromCheckpointMultiBatch (line 1248) | def _testFromCheckpointMultiBatch(self, n_classes):
    method testBinaryClassesFromCheckpointMultiBatch (line 1312) | def testBinaryClassesFromCheckpointMultiBatch(self):
    method testMultiClassesFromCheckpointMultiBatch (line 1315) | def testMultiClassesFromCheckpointMultiBatch(self):
  class BaseLinearClassifierEvaluationTest (line 1319) | class BaseLinearClassifierEvaluationTest(object):
    method __init__ (line 1321) | def __init__(self, linear_classifier_fn, fc_lib=feature_column_v2):
    method setUp (line 1325) | def setUp(self):
    method tearDown (line 1328) | def tearDown(self):
    method _test_evaluation_for_simple_data (line 1332) | def _test_evaluation_for_simple_data(self, n_classes):
    method test_binary_classes_evaluation_for_simple_data (line 1394) | def test_binary_classes_evaluation_for_simple_data(self):
    method test_multi_classes_evaluation_for_simple_data (line 1397) | def test_multi_classes_evaluation_for_simple_data(self):
    method _test_evaluation_batch (line 1400) | def _test_evaluation_batch(self, n_classes):
    method test_binary_classes_evaluation_batch (line 1472) | def test_binary_classes_evaluation_batch(self):
    method test_multi_classes_evaluation_batch (line 1475) | def test_multi_classes_evaluation_batch(self):
    method _test_evaluation_weights (line 1478) | def _test_evaluation_weights(self, n_classes):
    method test_binary_classes_evaluation_weights (line 1563) | def test_binary_classes_evaluation_weights(self):
    method test_multi_classes_evaluation_weights (line 1566) | def test_multi_classes_evaluation_weights(self):
  class BaseLinearClassifierPredictTest (line 1570) | class BaseLinearClassifierPredictTest(object):
    method __init__ (line 1572) | def __init__(self, linear_classifier_fn, fc_lib=feature_column_v2):
    method setUp (line 1576) | def setUp(self):
    method tearDown (line 1579) | def tearDown(self):
    method _testPredictions (line 1583) | def _testPredictions(self, n_classes, label_vocabulary, label_output_fn):
    method testBinaryClassesWithoutLabelVocabulary (line 1658) | def testBinaryClassesWithoutLabelVocabulary(self):
    method testBinaryClassesWithLabelVocabulary (line 1665) | def testBinaryClassesWithLabelVocabulary(self):
    method testMultiClassesWithoutLabelVocabulary (line 1672) | def testMultiClassesWithoutLabelVocabulary(self):
    method testMultiClassesWithLabelVocabulary (line 1679) | def testMultiClassesWithLabelVocabulary(self):
    method testSparseCombiner (line 1686) | def testSparseCombiner(self):
  class BaseLinearClassifierIntegrationTest (line 1749) | class BaseLinearClassifierIntegrationTest(object):
    method __init__ (line 1751) | def __init__(self, linear_classifier_fn, fc_lib=feature_column_v2):
    method setUp (line 1755) | def setUp(self):
    method tearDown (line 1758) | def tearDown(self):
    method _test_complete_flow (line 1762) | def _test_complete_flow(self, n_classes, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn (line 1794) | def _test_numpy_input_fn(self, n_classes):
    method test_binary_classes_numpy_input_fn (line 1830) | def test_binary_classes_numpy_input_fn(self):
    method test_multi_classes_numpy_input_fn (line 1833) | def test_multi_classes_numpy_input_fn(self):
    method _test_pandas_input_fn (line 1836) | def _test_pandas_input_fn(self, n_classes):
    method test_binary_classes_pandas_input_fn (line 1865) | def test_binary_classes_pandas_input_fn(self):
    method test_multi_classes_pandas_input_fn (line 1868) | def test_multi_classes_pandas_input_fn(self):
    method _test_input_fn_from_parse_example (line 1871) | def _test_input_fn_from_parse_example(self, n_classes):
    method test_binary_classes_input_fn_from_parse_example (line 1930) | def test_binary_classes_input_fn_from_parse_example(self):
    method test_multi_classes_input_fn_from_parse_example (line 1933) | def test_multi_classes_input_fn_from_parse_example(self):
  class BaseLinearLogitFnTest (line 1937) | class BaseLinearLogitFnTest(object):
    method __init__ (line 1939) | def __init__(self, fc_lib=feature_column_v2):
    method test_basic_logit_correctness (line 1942) | def test_basic_logit_correctness(self):
    method test_compute_fraction_of_zero_v2 (line 1963) | def test_compute_fraction_of_zero_v2(self):
  class BaseLinearWarmStartingTest (line 1995) | class BaseLinearWarmStartingTest(object):
    method __init__ (line 1997) | def __init__(self,
    method setUp (line 2005) | def setUp(self):
    method tearDown (line 2020) | def tearDown(self):
    method test_classifier_basic_warm_starting (line 2025) | def test_classifier_basic_warm_starting(self):
    method test_regressor_basic_warm_starting (line 2059) | def test_regressor_basic_warm_starting(self):
    method test_warm_starting_selective_variables (line 2091) | def test_warm_starting_selective_variables(self):
    method test_warm_starting_with_vocab_remapping_and_partitioning (line 2124) | def test_warm_starting_with_vocab_remapping_and_partitioning(self):
    method test_warm_starting_with_naming_change (line 2202) | def test_warm_starting_with_naming_change(self):

FILE: tensorflow_estimator/python/estimator/canned/metric_keys.py
  class MetricKeys (line 24) | class MetricKeys(object):

FILE: tensorflow_estimator/python/estimator/canned/optimizers.py
  function get_optimizer_instance (line 48) | def get_optimizer_instance(opt, learning_rate=None):
  function _optimizer_has_default_learning_rate (line 90) | def _optimizer_has_default_learning_rate(opt):
  function get_optimizer_instance_v2 (line 99) | def get_optimizer_instance_v2(opt, learning_rate=None):

FILE: tensorflow_estimator/python/estimator/canned/optimizers_test.py
  class _TestOptimizer (line 26) | class _TestOptimizer(tf.compat.v1.train.Optimizer):
    method __init__ (line 28) | def __init__(self):
  class GetOptimizerInstance (line 33) | class GetOptimizerInstance(tf.test.TestCase):
    method test_unsupported_name (line 35) | def test_unsupported_name(self):
    method test_supported_name_but_learning_rate_none (line 40) | def test_supported_name_but_learning_rate_none(self):
    method test_keras_optimizer_after_tf_2_11 (line 45) | def test_keras_optimizer_after_tf_2_11(self):
    method test_adagrad (line 61) | def test_adagrad(self):
    method test_adam (line 66) | def test_adam(self):
    method test_ftrl (line 71) | def test_ftrl(self):
    method test_rmsprop (line 76) | def test_rmsprop(self):
    method test_sgd (line 81) | def test_sgd(self):
    method test_object (line 86) | def test_object(self):
    method test_object_invalid (line 90) | def test_object_invalid(self):
    method test_callable (line 95) | def test_callable(self):
    method test_lambda (line 103) | def test_lambda(self):
    method test_callable_returns_invalid (line 107) | def test_callable_returns_invalid(self):

FILE: tensorflow_estimator/python/estimator/canned/optimizers_test_v2.py
  class _TestOptimizerV2 (line 26) | class _TestOptimizerV2(tf_keras.optimizers.legacy.Optimizer):
    method __init__ (line 28) | def __init__(self):
    method get_config (line 31) | def get_config(self):
  class GetOptimizerInstanceV2 (line 35) | class GetOptimizerInstanceV2(tf.test.TestCase):
    method test_unsupported_name (line 38) | def test_unsupported_name(self):
    method test_adagrad_but_no_learning_rate (line 44) | def test_adagrad_but_no_learning_rate(self):
    method test_adam_but_no_learning_rate (line 56) | def test_adam_but_no_learning_rate(self):
    method test_adagrad (line 65) | def test_adagrad(self):
    method test_adam (line 75) | def test_adam(self):
    method test_ftrl (line 84) | def test_ftrl(self):
    method test_rmsprop (line 93) | def test_rmsprop(self):
    method test_sgd (line 103) | def test_sgd(self):
    method test_object (line 112) | def test_object(self):
    method test_object_invalid (line 116) | def test_object_invalid(self):
    method test_callable (line 122) | def test_callable(self):
    method test_lambda (line 130) | def test_lambda(self):
    method test_callable_returns_invalid (line 134) | def test_callable_returns_invalid(self):

FILE: tensorflow_estimator/python/estimator/canned/parsing_utils.py
  function classifier_parse_example_spec_v2 (line 28) | def classifier_parse_example_spec_v2(feature_columns,
  function regressor_parse_example_spec_v2 (line 148) | def regressor_parse_example_spec_v2(feature_columns,
  function _add_label_and_weight_to_parsing_spec (line 264) | def _add_label_and_weight_to_parsing_spec(parsing_spec,
  function classifier_parse_example_spec (line 316) | def classifier_parse_example_spec(feature_columns,
  function regressor_parse_example_spec (line 335) | def regressor_parse_example_spec(

FILE: tensorflow_estimator/python/estimator/canned/parsing_utils_test.py
  class BaseClassifierParseExampleSpec (line 25) | class BaseClassifierParseExampleSpec(object):
    method __init__ (line 28) | def __init__(self, parse_example_fn):
    method test_defaults (line 31) | def test_defaults(self):
    method test_string (line 40) | def test_string(self):
    method test_label_default_value (line 52) | def test_label_default_value(self):
    method test_weight_column_as_string (line 65) | def test_weight_column_as_string(self):
    method test_weight_column_as_numeric_column (line 77) | def test_weight_column_as_numeric_column(self):
    method test_label_key_should_not_be_used_as_feature (line 89) | def test_label_key_should_not_be_used_as_feature(self):
    method test_weight_column_should_not_be_used_as_feature (line 96) | def test_weight_column_should_not_be_used_as_feature(self):
    method test_weight_column_should_be_a_numeric_column (line 104) | def test_weight_column_should_be_a_numeric_column(self):
  class ClassifierParseExampleSpecV2 (line 114) | class ClassifierParseExampleSpecV2(BaseClassifierParseExampleSpec,
    method __init__ (line 117) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method test_non_v1_feature_column (line 122) | def test_non_v1_feature_column(self):
  class ClassifierParseExampleSpecV1 (line 133) | class ClassifierParseExampleSpecV1(BaseClassifierParseExampleSpec,
    method __init__ (line 136) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class BaseRegressorParseExampleSpec (line 142) | class BaseRegressorParseExampleSpec(object):
    method __init__ (line 145) | def __init__(self, parse_example_fn):
    method test_defaults (line 148) | def test_defaults(self):
    method test_int64 (line 157) | def test_int64(self):
    method test_label_default_value (line 168) | def test_label_default_value(self):
    method test_label_dimension (line 183) | def test_label_dimension(self):
    method test_weight_column_as_string (line 194) | def test_weight_column_as_string(self):
    method test_weight_column_as_numeric_column (line 206) | def test_weight_column_as_numeric_column(self):
    method test_label_key_should_not_be_used_as_feature (line 218) | def test_label_key_should_not_be_used_as_feature(self):
    method test_weight_column_should_not_be_used_as_feature (line 225) | def test_weight_column_should_not_be_used_as_feature(self):
    method test_weight_column_should_be_a_numeric_column (line 233) | def test_weight_column_should_be_a_numeric_column(self):
  class RegressorParseExampleSpecV2 (line 243) | class RegressorParseExampleSpecV2(BaseRegressorParseExampleSpec,
    method __init__ (line 246) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method test_non_v1_feature_column (line 251) | def test_non_v1_feature_column(self):
  class RegressorParseExampleSpecV1 (line 262) | class RegressorParseExampleSpecV1(BaseRegressorParseExampleSpec,
    method __init__ (line 265) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name

FILE: tensorflow_estimator/python/estimator/canned/prediction_keys.py
  class PredictionKeys (line 22) | class PredictionKeys(object):

FILE: tensorflow_estimator/python/estimator/canned/rnn.py
  function _single_rnn_cell (line 63) | def _single_rnn_cell(units, cell_type):
  function _make_rnn_cell_fn (line 81) | def _make_rnn_cell_fn(units, cell_type=_SIMPLE_RNN_KEY):
  class RNNModel (line 105) | class RNNModel(tf_keras.models.Model):
    method __init__ (line 129) | def __init__(self,
    method call (line 180) | def call(self, inputs, training=None):
    method get_config (line 227) | def get_config(self):
    method from_config (line 245) | def from_config(cls, config, custom_objects=None):
  function _get_rnn_estimator_spec (line 274) | def _get_rnn_estimator_spec(features, labels, mode, head, rnn_model, opt...
  function _verify_rnn_cell_input (line 332) | def _verify_rnn_cell_input(rnn_cell_fn, units, cell_type):
  function _make_rnn_layer (line 338) | def _make_rnn_layer(rnn_cell_fn, units, cell_type, return_sequences):
  class RNNEstimator (line 366) | class RNNEstimator(estimator.Estimator):
    method __init__ (line 426) | def __init__(self,
  class RNNClassifier (line 519) | class RNNClassifier(RNNEstimator):
    method __init__ (line 571) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/canned/rnn_test.py
  function _assert_close (line 51) | def _assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function create_checkpoint (line 66) | def create_checkpoint(kernel, recurrent, bias, dense_kernel, dense_bias,
  function _make_rnn_layer (line 101) | def _make_rnn_layer(rnn_cell_fn=None,
  class RNNLayerFnTest (line 113) | class RNNLayerFnTest(tf.test.TestCase, parameterized.TestCase):
    method testWrongClassProvided (line 116) | def testWrongClassProvided(self):
    method testWrongStringProvided (line 122) | def testWrongStringProvided(self):
    method testDefaultCellProvided (line 130) | def testDefaultCellProvided(self, cell_type):
    method testSpecificLayerTypeProvided (line 139) | def testSpecificLayerTypeProvided(self, cell_type, layer_type):
    method testSpecificLayerTypeArguments (line 144) | def testSpecificLayerTypeArguments(self):
    method testCustomCellProvided (line 157) | def testCustomCellProvided(self, mock_rnn_layer_type):
    method testMultipleCellsProvided (line 167) | def testMultipleCellsProvided(self):
    method testCustomCellFnProvided (line 176) | def testCustomCellFnProvided(self, mock_rnn_layer_type):
  function _mock_logits_layer (line 185) | def _mock_logits_layer(kernel, bias):
  function _default_features_fn (line 205) | def _default_features_fn():
  function _get_mock_head (line 213) | def _get_mock_head():
  class RNNLogitFnTest (line 221) | class RNNLogitFnTest(tf.test.TestCase, parameterized.TestCase):
    method setUp (line 224) | def setUp(self):
    method _mock_logits_layer (line 237) | def _mock_logits_layer(self):
    method _test_logits (line 240) | def _test_logits(self,
    method testOneDimLogits (line 289) | def testOneDimLogits(self, return_sequences, expected_logits, training...
    method testMultiDimLogits (line 332) | def testMultiDimLogits(self, return_sequences, expected_logits):
    method testMultiExampleMultiDim (line 383) | def testMultiExampleMultiDim(self, return_sequences, expected_logits):
    method testMultiExamplesDifferentLength (line 450) | def testMultiExamplesDifferentLength(self, return_sequences, expected_...
    method testMultiExamplesWithContext (line 496) | def testMultiExamplesWithContext(self):
    method testMultiExamplesMultiFeatures (line 539) | def testMultiExamplesMultiFeatures(self):
    method testTrainingMode (line 592) | def testTrainingMode(self, mode, expected_training_mode):
  class RNNModelTest (line 621) | class RNNModelTest(tf.test.TestCase, parameterized.TestCase):
    method setUp (line 624) | def setUp(self):
    method _get_compiled_model (line 643) | def _get_compiled_model(self,
    method testModelWeights (line 669) | def testModelWeights(self):
    method _testModelConfig (line 705) | def _testModelConfig(self, **kwargs):
    method testModelConfig (line 737) | def testModelConfig(self):
    method testModelConfigWithActivation (line 750) | def testModelConfigWithActivation(self):
    method testModelConfigWithContextFeatures (line 755) | def testModelConfigWithContextFeatures(self):
    method DISABLED_testSaveModelWeights (line 762) | def DISABLED_testSaveModelWeights(self):  # See b/129842600.
    method DISABLED_testEvaluationMetrics (line 774) | def DISABLED_testEvaluationMetrics(self):  # See b/129842600.
    method DISABLED_testEvaluationSequential (line 782) | def DISABLED_testEvaluationSequential(self):  # See b/129842600.
    method DISABLED_testPredictions (line 789) | def DISABLED_testPredictions(self):  # See b/129842600.
    method DISABLED_testPredictionsSequential (line 796) | def DISABLED_testPredictionsSequential(self):  # See b/129842600.
    method DISABLED_testTraining (line 808) | def DISABLED_testTraining(self, optimizer):  # See b/129842600.
    method DISABLED_testTrainingSequential (line 819) | def DISABLED_testTrainingSequential(self):  # See b/129842600.
  class RNNEstimatorInitTest (line 828) | class RNNEstimatorInitTest(tf.test.TestCase):
    method setUp (line 830) | def setUp(self):
    method testConflictingRNNCellFn (line 839) | def testConflictingRNNCellFn(self):
    method testNonSequentialHeadProvided (line 856) | def testNonSequentialHeadProvided(self):
    method testWrongOptimizerTypeProvided (line 865) | def testWrongOptimizerTypeProvided(self):
  class RNNClassifierTrainingTest (line 876) | class RNNClassifierTrainingTest(tf.test.TestCase):
    method setUp (line 878) | def setUp(self):
    method _assert_checkpoint (line 889) | def _assert_checkpoint(self, n_classes, input_units, cell_units,
    method _mock_optimizer (line 919) | def _mock_optimizer(self, expected_loss=None):
    method _testFromScratchWithDefaultOptimizer (line 955) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassFromScratchWithDefaultOptimizer (line 983) | def testBinaryClassFromScratchWithDefaultOptimizer(self):
    method testMultiClassFromScratchWithDefaultOptimizer (line 986) | def testMultiClassFromScratchWithDefaultOptimizer(self):
    method testFromScratchWithCustomRNNCellFn (line 989) | def testFromScratchWithCustomRNNCellFn(self):
    method _testExampleWeight (line 1022) | def _testExampleWeight(self, n_classes):
    method testBinaryClassWithExampleWeight (line 1052) | def testBinaryClassWithExampleWeight(self):
    method testMultiClassWithExampleWeight (line 1055) | def testMultiClassWithExampleWeight(self):
    method _testFromCheckpoint (line 1058) | def _testFromCheckpoint(self, input_fn, expected_loss, **kwargs):
    method testBinaryClassFromCheckpoint (line 1081) | def testBinaryClassFromCheckpoint(self):
    method testMultiClassFromCheckpoint (line 1096) | def testMultiClassFromCheckpoint(self):
    method testBinaryClassFromCheckpointSequential (line 1114) | def testBinaryClassFromCheckpointSequential(self):
    method testBinaryClassFromCheckpointSequentialWithWeights (line 1142) | def testBinaryClassFromCheckpointSequentialWithWeights(self):
    method testDefaultGradientClipping (line 1176) | def testDefaultGradientClipping(self):
  function sorted_key_dict (line 1230) | def sorted_key_dict(unsorted_dict):
  class RNNClassifierEvaluationTest (line 1235) | class RNNClassifierEvaluationTest(tf.test.TestCase):
    method setUp (line 1237) | def setUp(self):
    method _testFromCheckpoint (line 1249) | def _testFromCheckpoint(self, input_fn, **kwargs):
    method testBinaryClassEvaluationMetrics (line 1266) | def testBinaryClassEvaluationMetrics(self):
    method testBinaryClassEvaluationMetricsSequential (line 1304) | def testBinaryClassEvaluationMetricsSequential(self):
    method testMultiClassEvaluationMetrics (line 1351) | def testMultiClassEvaluationMetrics(self):
  class RNNClassifierPredictionTest (line 1391) | class RNNClassifierPredictionTest(tf.test.TestCase):
    method setUp (line 1393) | def setUp(self):
    method _testFromCheckpoint (line 1404) | def _testFromCheckpoint(self, input_fn, **kwargs):
    method testBinaryClassPredictions (line 1430) | def testBinaryClassPredictions(self):
    method testMultiClassPredictions (line 1450) | def testMultiClassPredictions(self):
    method testBinaryClassPredictionsSequential (line 1471) | def testBinaryClassPredictionsSequential(self):
  class BaseRNNClassificationIntegrationTest (line 1505) | class BaseRNNClassificationIntegrationTest(object):
    method setUp (line 1507) | def setUp(self):
    method __init__ (line 1514) | def __init__(self, _create_estimator_fn):
    method _test_complete_flow (line 1517) | def _test_complete_flow(self,
    method _testNumpyInputFn (line 1557) | def _testNumpyInputFn(self, optimizer):
    method testNumpyInputFnStringOptimizer (line 1593) | def testNumpyInputFnStringOptimizer(self):
    method testNumpyInputFnOptimizerInstance (line 1596) | def testNumpyInputFnOptimizerInstance(self):
    method testParseExampleInputFn (line 1599) | def testParseExampleInputFn(self):
  function _rnn_classifier_fn (line 1655) | def _rnn_classifier_fn(feature_columns, n_classes, cell_units, model_dir,
  class RNNClassifierIntegrationTest (line 1666) | class RNNClassifierIntegrationTest(BaseRNNClassificationIntegrationTest,
    method __init__ (line 1669) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _rnn_classifier_dropout_fn (line 1674) | def _rnn_classifier_dropout_fn(feature_columns, n_classes, cell_units,
  class RNNClassifierDropoutIntegrationTest (line 1692) | class RNNClassifierDropoutIntegrationTest(BaseRNNClassificationIntegrati...
    method __init__ (line 1695) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _rnn_estimator_fn (line 1701) | def _rnn_estimator_fn(feature_columns, n_classes, cell_units, model_dir,
  class RNNEstimatorIntegrationTest (line 1712) | class RNNEstimatorIntegrationTest(BaseRNNClassificationIntegrationTest,
    method __init__ (line 1715) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class ModelFnTest (line 1721) | class ModelFnTest(tf.test.TestCase):
    method _test_sequential_mask_in_head (line 1724) | def _test_sequential_mask_in_head(self, mask=None):
    method testSequentialMaskInHead (line 1759) | def testSequentialMaskInHead(self):
    method testSequentialMaskInHeadWithMasks (line 1762) | def testSequentialMaskInHeadWithMasks(self):

FILE: tensorflow_estimator/python/estimator/canned/saved_model_estimator.py
  class SavedModelEstimator (line 35) | class SavedModelEstimator(estimator_lib.EstimatorV2):
    method __init__ (line 121) | def __init__(self, saved_model_dir, model_dir=None):
    method _extract_available_modes (line 148) | def _extract_available_modes(self):
    method _validate_mode (line 167) | def _validate_mode(self, mode):
    method _get_meta_graph_def_for_mode (line 174) | def _get_meta_graph_def_for_mode(self, mode):
    method _get_signature_def_for_mode (line 178) | def _get_signature_def_for_mode(self, mode):
    method _get_saver_def_from_mode (line 191) | def _get_saver_def_from_mode(self, mode):
    method _create_and_assert_global_step (line 195) | def _create_and_assert_global_step(self, graph):
    method _model_fn_from_saved_model (line 205) | def _model_fn_from_saved_model(self, features, labels, mode):
    method _restore_from_saver (line 290) | def _restore_from_saver(self, scaffold, session):
    method latest_checkpoint (line 294) | def latest_checkpoint(self):
  function _get_saved_model_ckpt (line 305) | def _get_saved_model_ckpt(saved_model_dir):
  function _clear_saved_model_collections (line 316) | def _clear_saved_model_collections():
  function _generate_input_map (line 330) | def _generate_input_map(signature_def, features, labels):
  function _check_same_dtype_and_shape (line 382) | def _check_same_dtype_and_shape(tensor, tensor_info, name):
  function _extract_eval_metrics (line 408) | def _extract_eval_metrics(output_dict):
  function _validate_and_extract_outputs (line 447) | def _validate_and_extract_outputs(mode, output_dict, method_name):

FILE: tensorflow_estimator/python/estimator/canned/saved_model_estimator_test.py
  function dummy_input_fn (line 40) | def dummy_input_fn():
  function _serving_feature_dict (line 46) | def _serving_feature_dict():
  function dummy_input_fn_features_only (line 50) | def dummy_input_fn_features_only():
  function dummy_supervised_receiver_fn (line 55) | def dummy_supervised_receiver_fn():
  function dummy_serving_receiver_fn (line 60) | def dummy_serving_receiver_fn():
  function model_fn_diff_modes (line 64) | def model_fn_diff_modes(features, labels, mode):
  function model_fn_with_trackable (line 93) | def model_fn_with_trackable(features, labels, mode):
  class SavedModelEstimatorTest (line 115) | class SavedModelEstimatorTest(tf.test.TestCase):
    method setUp (line 117) | def setUp(self):
    method tearDown (line 121) | def tearDown(self):
    method _get_tmp_dir (line 128) | def _get_tmp_dir(self):
    method _export_estimator (line 133) | def _export_estimator(self,
    method test_load_all_modes (line 154) | def test_load_all_modes(self):
    method test_load_all_modes_no_train (line 171) | def test_load_all_modes_no_train(self):
    method test_partial_exported_estimator (line 183) | def test_partial_exported_estimator(self):
    method test_with_incorrect_input (line 199) | def test_with_incorrect_input(self):
    method test_input_fn_with_global_step (line 219) | def test_input_fn_with_global_step(self):
    method test_re_export_saved_model_serving_only (line 233) | def test_re_export_saved_model_serving_only(self):
    method test_re_export_saved_model (line 256) | def test_re_export_saved_model(self):
    method test_re_export_saved_model_with_trackable (line 303) | def test_re_export_saved_model_with_trackable(self):
    method test_load_saved_model_from_serving_only (line 350) | def test_load_saved_model_from_serving_only(self):
    method test_with_local_init_op (line 384) | def test_with_local_init_op(self):
    method test_with_assets (line 411) | def test_with_assets(self):
    method test_with_working_input_fn (line 454) | def test_with_working_input_fn(self):
    method test_control_dependency (line 478) | def test_control_dependency(self):
    method test_saveable_resources (line 496) | def test_saveable_resources(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/ar_model.py
  class LSTMPredictionModel (line 33) | class LSTMPredictionModel(tf_keras.models.Model):
    method __init__ (line 41) | def __init__(self,
    method call (line 65) | def call(self, input_window_features, output_window_features):
  class ARModel (line 77) | class ARModel(model.TimeSeriesModel):
    method __init__ (line 128) | def __init__(self,
    method initialize_graph (line 202) | def initialize_graph(self, input_statistics=None):
    method get_start_state (line 216) | def get_start_state(self):
    method random_model_parameters (line 227) | def random_model_parameters(self, seed=None):
    method generate (line 230) | def generate(self,
    method _predicted_covariance_op (line 237) | def _predicted_covariance_op(self, activations, num_values):
    method _predicted_mean_op (line 258) | def _predicted_mean_op(self, activations):
    method prediction_ops (line 269) | def prediction_ops(self, times, values, exogenous_regressors):
    method _output_window_predictions (line 342) | def _output_window_predictions(self, input_window_features,
    method loss_op (line 352) | def loss_op(self, targets, prediction_ops):
    method _process_exogenous_features (line 367) | def _process_exogenous_features(self, times, features):
    method predict (line 381) | def predict(self, features):
    method _process_window (line 555) | def _process_window(self, features, mode, exogenous_regressors):
    method get_batch_loss (line 617) | def get_batch_loss(self, features, mode, state):
    method _compute_time_features (line 788) | def _compute_time_features(self, time):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/ar_model_test.py
  class ARModelTest (line 32) | class ARModelTest(tf.test.TestCase):
    method test_wrong_window_size (line 34) | def test_wrong_window_size(self):
    method test_predictions_direct_lstm (line 60) | def test_predictions_direct_lstm(self):
    method test_long_eval (line 77) | def test_long_eval(self):
    method test_long_eval_discard_indivisible (line 102) | def test_long_eval_discard_indivisible(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/ar_model_training_test.py
  class InputFnBuilder (line 31) | class InputFnBuilder(object):
    method __init__ (line 33) | def __init__(self,
    method initialize_data_with_properties (line 46) | def initialize_data_with_properties(self, noise_stddev, periods, num_s...
    method train_or_test_input_fn (line 70) | def train_or_test_input_fn(self, time, data):
    method train_input_fn (line 86) | def train_input_fn(self):
    method test_input_fn (line 90) | def test_input_fn(self):
    method prediction_input_fn (line 94) | def prediction_input_fn(self):
    method true_values (line 125) | def true_values(self):
  class ARModelTrainingTest (line 134) | class ARModelTrainingTest(tf.test.TestCase):
    method train_helper (line 136) | def train_helper(self, input_window_size, loss, max_loss=None, periods...
    method test_autoregression_squared (line 199) | def test_autoregression_squared(self):
    method test_autoregression_short_input_window (line 202) | def test_autoregression_short_input_window(self):
    method test_autoregression_normal (line 205) | def test_autoregression_normal(self):
    method test_autoregression_normal_multiple_periods (line 211) | def test_autoregression_normal_multiple_periods(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/estimators.py
  class TimeSeriesRegressor (line 33) | class TimeSeriesRegressor(estimator_lib.Estimator):
    method __init__ (line 36) | def __init__(self,
    method _model_start_state_placeholders (line 75) | def _model_start_state_placeholders(self,
    method build_one_shot_parsing_serving_input_receiver_fn (line 112) | def build_one_shot_parsing_serving_input_receiver_fn(self,
    method build_raw_serving_input_receiver_fn (line 210) | def build_raw_serving_input_receiver_fn(self,
  class LSTMAutoRegressor (line 293) | class LSTMAutoRegressor(TimeSeriesRegressor):
    method __init__ (line 373) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/canned/timeseries/estimators_test.py
  class _SeedRunConfig (line 33) | class _SeedRunConfig(estimator_lib.RunConfig):
    method tf_random_seed (line 36) | def tf_random_seed(self):
  function _generate_data (line 40) | def _generate_data():
  function _build_input_fn_with_seed (line 47) | def _build_input_fn_with_seed(seed):
  class TimeSeriesRegressorTest (line 71) | class TimeSeriesRegressorTest(tf.test.TestCase):
    method _fit_restore_fit_test_template (line 73) | def _fit_restore_fit_test_template(self, estimator_fn, test_saved_model):
    method disabled_test_time_series_regressor (line 179) | def disabled_test_time_series_regressor(self):
    method test_ar_lstm_regressor (line 196) | def test_ar_lstm_regressor(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/feature_keys.py
  class State (line 24) | class State(object):
  class Times (line 34) | class Times(object):
  class Values (line 40) | class Values(object):
  class TrainEvalFeatures (line 46) | class TrainEvalFeatures(Times, Values):
  class PredictionFeatures (line 51) | class PredictionFeatures(Times, State):
  class FilteringFeatures (line 56) | class FilteringFeatures(Times, Values, State):
  class PredictionResults (line 61) | class PredictionResults(Times):
  class FilteringResults (line 66) | class FilteringResults(Times, State):
  class SavedModelLabels (line 71) | class SavedModelLabels(object):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/head.py
  class _NoStatePredictOutput (line 30) | class _NoStatePredictOutput(export_lib.PredictOutput):
    method as_signature_def (line 32) | def as_signature_def(self, receiver_tensors):
  class TimeSeriesRegressionHead (line 43) | class TimeSeriesRegressionHead(head_lib._Head):  # pylint:disable=protec...
    method __init__ (line 46) | def __init__(self,
    method name (line 68) | def name(self):
    method create_loss (line 73) | def create_loss(self, features, mode, logits=None, labels=None):
    method logits_dimension (line 82) | def logits_dimension(self):
    method _train_ops (line 86) | def _train_ops(self, features):
    method _evaluate_ops (line 100) | def _evaluate_ops(self, features):
    method _predict_ops (line 124) | def _predict_ops(self, features):
    method _serving_ops (line 133) | def _serving_ops(self, features):
    method _convert_feature_to_tensor (line 166) | def _convert_feature_to_tensor(self, name, value):
    method _gather_state (line 179) | def _gather_state(self, features):
    method _check_predict_features (line 199) | def _check_predict_features(self, features):
    method create_estimator_spec (line 222) | def create_estimator_spec(self, features, mode, labels=None):
  class OneShotPredictionHead (line 271) | class OneShotPredictionHead(TimeSeriesRegressionHead):
    method _check_predict_features (line 288) | def _check_predict_features(self, features):
    method _evaluate_ops (line 318) | def _evaluate_ops(self, features):
    method _serving_ops (line 326) | def _serving_ops(self, features):
  function _check_feature_shapes_compatible_with (line 360) | def _check_feature_shapes_compatible_with(features,
  function _check_train_eval_features (line 391) | def _check_train_eval_features(features, model):
  function _identity_metric_single (line 424) | def _identity_metric_single(name, input_tensor):
  function _identity_metric_nested (line 453) | def _identity_metric_nested(name, input_tensors):
  function state_to_dictionary (line 466) | def state_to_dictionary(state_tuple):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/head_test.py
  class HeadTest (line 41) | class HeadTest(tf.test.TestCase):
    method test_labels_provided_error (line 43) | def test_labels_provided_error(self):
    method test_unknown_mode (line 55) | def test_unknown_mode(self):
  class _TickerModel (line 61) | class _TickerModel(object):
    method initialize_graph (line 65) | def initialize_graph(self, input_statistics):
    method define_loss (line 68) | def define_loss(self, features, mode):
  class EvaluationMetricsTests (line 78) | class EvaluationMetricsTests(tf.test.TestCase):
    method test_metrics_consistent (line 80) | def test_metrics_consistent(self):
    method test_custom_metrics (line 131) | def test_custom_metrics(self):
  class _StubModel (line 169) | class _StubModel(object):
    method initialize_graph (line 173) | def initialize_graph(self, input_statistics):
  function _stub_model_fn (line 177) | def _stub_model_fn():
  class TrainEvalFeatureCheckingTests (line 184) | class TrainEvalFeatureCheckingTests(tf.test.TestCase):
    method test_no_time_feature (line 186) | def test_no_time_feature(self):
    method test_no_value_feature (line 197) | def test_no_value_feature(self):
    method test_bad_time_rank (line 208) | def test_bad_time_rank(self):
    method test_bad_value_rank (line 222) | def test_bad_value_rank(self):
    method test_bad_value_num_features (line 236) | def test_bad_value_num_features(self):
    method test_bad_exogenous_shape (line 250) | def test_bad_exogenous_shape(self):
  class PredictFeatureCheckingTests (line 265) | class PredictFeatureCheckingTests(tf.test.TestCase):
    method test_no_time_feature (line 267) | def test_no_time_feature(self):
    method test_no_start_state_feature (line 279) | def test_no_start_state_feature(self):
    method test_bad_time_rank (line 289) | def test_bad_time_rank(self):
    method test_bad_exogenous_shape (line 302) | def test_bad_exogenous_shape(self):
  class OneShotTests (line 317) | class OneShotTests(parameterized.TestCase):
    method test_one_shot_prediction_head_export (line 319) | def test_one_shot_prediction_head_export(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/math_utils.py
  function replicate_state (line 27) | def replicate_state(start_state, batch_size):
  class InputStatisticsFromMiniBatch (line 92) | class InputStatisticsFromMiniBatch(object):
    method __init__ (line 95) | def __init__(self, num_features, dtype, starting_variance_window_size=...
    method initialize_graph (line 108) | def initialize_graph(self, features, update_statistics=True):
    class _AdaptiveInputAuxiliaryStatistics (line 166) | class _AdaptiveInputAuxiliaryStatistics(
      method __new__ (line 189) | def __new__(cls, num_features, dtype):
    method _update_statistics_from_mini_batch (line 230) | def _update_statistics_from_mini_batch(self, statistics, auxiliary_var...
    method _create_variable_statistics_object (line 357) | def _create_variable_statistics_object(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/math_utils_test.py
  class InputStatisticsTests (line 26) | class InputStatisticsTests(tf.test.TestCase):
    method _input_statistics_test_template (line 28) | def _input_statistics_test_template(self,
    method test_queue (line 84) | def test_queue(self):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/model.py
  class TimeSeriesModel (line 44) | class TimeSeriesModel(object):
    method __init__ (line 47) | def __init__(self,
    method exogenous_feature_columns (line 74) | def exogenous_feature_columns(self):
    method generate (line 81) | def generate(self,
    method initialize_graph (line 111) | def initialize_graph(self, input_statistics=None):
    method _scale_data (line 126) | def _scale_data(self, data):
    method _scale_variance (line 133) | def _scale_variance(self, variance):
    method _scale_back_data (line 140) | def _scale_back_data(self, data):
    method _scale_back_variance (line 147) | def _scale_back_variance(self, variance):
    method _check_graph_initialized (line 154) | def _check_graph_initialized(self):
    method define_loss (line 161) | def define_loss(self, features, mode):
    method get_start_state (line 198) | def get_start_state(self):
    method get_batch_loss (line 208) | def get_batch_loss(self, features, mode, state):
    method predict (line 225) | def predict(self, features):
    method _get_exogenous_embedding_shape (line 251) | def _get_exogenous_embedding_shape(self):
    method _process_exogenous_features (line 272) | def _process_exogenous_features(self, times, features):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/model_utils.py
  function fully_connected (line 27) | def fully_connected(inp,
  function canonicalize_times_or_steps_from_output (line 47) | def canonicalize_times_or_steps_from_output(times, steps,

FILE: tensorflow_estimator/python/estimator/canned/timeseries/saved_model_utils.py
  function _canonicalize_numpy_data (line 34) | def _canonicalize_numpy_data(data, require_single_batch):
  function _colate_features_to_feeds_and_fetches (line 120) | def _colate_features_to_feeds_and_fetches(signature,
  function predict_continuation (line 149) | def predict_continuation(continue_from,
  function cold_start_filter (line 207) | def cold_start_filter(signatures, session, features):
  function filter_continuation (line 251) | def filter_continuation(continue_from, signatures, session, features):

FILE: tensorflow_estimator/python/estimator/canned/timeseries/state_management.py
  class PassthroughStateManager (line 26) | class PassthroughStateManager(object):
    method __init__ (line 29) | def __init__(self):
    method initialize_graph (line 33) | def initialize_graph(self, model, input_statistics=None):
    method define_loss (line 39) | def define_loss(self, model, features, mode):
  class _OverridableStateManager (line 62) | class _OverridableStateManager(PassthroughStateManager):
    method _define_loss_with_saved_state (line 66) | def _define_loss_with_saved_state(self, model, features, mode):
    method define_loss (line 69) | def define_loss(self, model, features, mode):
  class FilteringOnlyStateManager (line 88) | class FilteringOnlyStateManager(_OverridableStateManager):
    method _define_loss_with_saved_state (line 97) | def _define_loss_with_saved_state(self, model, features, mode):

FILE: tensorflow_estimator/python/estimator/canned/v1/baseline_estimator_test_v1.py
  function assert_close (line 40) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 55) | def save_variables_to_ckpt(model_dir):
  function _baseline_estimator_fn (line 62) | def _baseline_estimator_fn(weight_column=None, label_dimension=1, **kwar...
  class BaselineEstimatorEvaluationTest (line 72) | class BaselineEstimatorEvaluationTest(tf.test.TestCase):
    method setUp (line 74) | def setUp(self):
    method tearDown (line 77) | def tearDown(self):
    method test_evaluation_batch (line 82) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 109) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 139) | def test_evaluation_for_multi_dimensions(self):
  class BaselineEstimatorPredictTest (line 169) | class BaselineEstimatorPredictTest(tf.test.TestCase):
    method setUp (line 171) | def setUp(self):
    method tearDown (line 174) | def tearDown(self):
    method test_1d (line 179) | def test_1d(self):
    method testMultiDim (line 199) | def testMultiDim(self):
  class BaselineEstimatorIntegrationTest (line 226) | class BaselineEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 228) | def setUp(self):
    method tearDown (line 231) | def tearDown(self):
    method _test_complete_flow (line 236) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 267) | def test_numpy_input_fn(self):
  class BaselineEstimatorTrainingTest (line 305) | class BaselineEstimatorTrainingTest(tf.test.TestCase):
    method setUp (line 307) | def setUp(self):
    method tearDown (line 310) | def tearDown(self):
    method _mock_optimizer (line 315) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 350) | def _assert_checkpoint(self,
    method testFromScratch (line 370) | def testFromScratch(self):
    method testFromCheckpoint (line 388) | def testFromCheckpoint(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/baseline_test_v1.py
  function assert_close (line 56) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 71) | def save_variables_to_ckpt(model_dir):
  function queue_parsed_features (line 78) | def queue_parsed_features(feature_map):
  function sorted_key_dict (line 93) | def sorted_key_dict(unsorted_dict):
  function sigmoid (line 97) | def sigmoid(x):
  function _baseline_regressor_fn (line 101) | def _baseline_regressor_fn(*args, **kwargs):
  function _baseline_classifier_fn (line 105) | def _baseline_classifier_fn(*args, **kwargs):
  class BaselineRegressorEvaluationTest (line 114) | class BaselineRegressorEvaluationTest(tf.test.TestCase):
    method setUp (line 116) | def setUp(self):
    method tearDown (line 119) | def tearDown(self):
    method test_evaluation_for_simple_data (line 124) | def test_evaluation_for_simple_data(self):
    method test_evaluation_batch (line 147) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 174) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 204) | def test_evaluation_for_multi_dimensions(self):
  class BaselineRegressorPredictTest (line 234) | class BaselineRegressorPredictTest(tf.test.TestCase):
    method setUp (line 236) | def setUp(self):
    method tearDown (line 239) | def tearDown(self):
    method test_1d (line 244) | def test_1d(self):
    method testMultiDim (line 264) | def testMultiDim(self):
  class BaselineRegressorIntegrationTest (line 291) | class BaselineRegressorIntegrationTest(tf.test.TestCase):
    method setUp (line 293) | def setUp(self):
    method tearDown (line 296) | def tearDown(self):
    method _test_complete_flow (line 301) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 332) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 368) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 397) | def test_input_fn_from_parse_example(self):
  class BaselineRegressorTrainingTest (line 459) | class BaselineRegressorTrainingTest(tf.test.TestCase):
    method setUp (line 461) | def setUp(self):
    method tearDown (line 464) | def tearDown(self):
    method _mock_optimizer (line 469) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 504) | def _assert_checkpoint(self,
    method testFromScratchWithDefaultOptimizer (line 524) | def testFromScratchWithDefaultOptimizer(self):
    method testTrainWithOneDimLabel (line 536) | def testTrainWithOneDimLabel(self):
    method testTrainWithOneDimWeight (line 553) | def testTrainWithOneDimWeight(self):
    method testFromScratch (line 576) | def testFromScratch(self):
    method testFromCheckpoint (line 594) | def testFromCheckpoint(self):
    method testFromCheckpointMultiBatch (line 623) | def testFromCheckpointMultiBatch(self):
  class BaselineClassifierTrainingTest (line 660) | class BaselineClassifierTrainingTest(tf.test.TestCase):
    method setUp (line 662) | def setUp(self):
    method tearDown (line 665) | def tearDown(self):
    method _mock_optimizer (line 669) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 700) | def _assert_checkpoint(self,
    method _testFromScratchWithDefaultOptimizer (line 722) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassesFromScratchWithDefaultOptimizer (line 734) | def testBinaryClassesFromScratchWithDefaultOptimizer(self):
    method testMultiClassesFromScratchWithDefaultOptimizer (line 737) | def testMultiClassesFromScratchWithDefaultOptimizer(self):
    method _testTrainWithTwoDimsLabel (line 740) | def _testTrainWithTwoDimsLabel(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsLabel (line 759) | def testBinaryClassesTrainWithTwoDimsLabel(self):
    method testMultiClassesTrainWithTwoDimsLabel (line 762) | def testMultiClassesTrainWithTwoDimsLabel(self):
    method _testTrainWithOneDimLabel (line 765) | def _testTrainWithOneDimLabel(self, n_classes):
    method testBinaryClassesTrainWithOneDimLabel (line 782) | def testBinaryClassesTrainWithOneDimLabel(self):
    method testMultiClassesTrainWithOneDimLabel (line 785) | def testMultiClassesTrainWithOneDimLabel(self):
    method _testTrainWithTwoDimsWeight (line 788) | def _testTrainWithTwoDimsWeight(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsWeight (line 810) | def testBinaryClassesTrainWithTwoDimsWeight(self):
    method testMultiClassesTrainWithTwoDimsWeight (line 813) | def testMultiClassesTrainWithTwoDimsWeight(self):
    method _testTrainWithOneDimWeight (line 816) | def _testTrainWithOneDimWeight(self, n_classes):
    method testBinaryClassesTrainWithOneDimWeight (line 836) | def testBinaryClassesTrainWithOneDimWeight(self):
    method testMultiClassesTrainWithOneDimWeight (line 839) | def testMultiClassesTrainWithOneDimWeight(self):
    method _testFromScratch (line 842) | def _testFromScratch(self, n_classes):
    method testBinaryClassesFromScratch (line 874) | def testBinaryClassesFromScratch(self):
    method testMultiClassesFromScratch (line 877) | def testMultiClassesFromScratch(self):
    method _testFromCheckpoint (line 880) | def _testFromCheckpoint(self, n_classes):
    method testBinaryClassesFromCheckpoint (line 928) | def testBinaryClassesFromCheckpoint(self):
    method testMultiClassesFromCheckpoint (line 931) | def testMultiClassesFromCheckpoint(self):
    method _testFromCheckpointFloatLabels (line 934) | def _testFromCheckpointFloatLabels(self, n_classes):
    method testBinaryClassesFromCheckpointFloatLabels (line 968) | def testBinaryClassesFromCheckpointFloatLabels(self):
    method testMultiClassesFromCheckpointFloatLabels (line 971) | def testMultiClassesFromCheckpointFloatLabels(self):
    method _testFromCheckpointMultiBatch (line 974) | def _testFromCheckpointMultiBatch(self, n_classes):
    method testBinaryClassesFromCheckpointMultiBatch (line 1031) | def testBinaryClassesFromCheckpointMultiBatch(self):
    method testMultiClassesFromCheckpointMultiBatch (line 1034) | def testMultiClassesFromCheckpointMultiBatch(self):
  class BaselineClassifierEvaluationTest (line 1039) | class BaselineClassifierEvaluationTest(tf.test.TestCase):
    method setUp (line 1041) | def setUp(self):
    method tearDown (line 1044) | def tearDown(self):
    method _test_evaluation_for_simple_data (line 1048) | def _test_evaluation_for_simple_data(self, n_classes):
    method test_binary_classes_evaluation_for_simple_data (line 1100) | def test_binary_classes_evaluation_for_simple_data(self):
    method test_multi_classes_evaluation_for_simple_data (line 1103) | def test_multi_classes_evaluation_for_simple_data(self):
    method _test_evaluation_batch (line 1106) | def _test_evaluation_batch(self, n_classes):
    method test_binary_classes_evaluation_batch (line 1168) | def test_binary_classes_evaluation_batch(self):
    method test_multi_classes_evaluation_batch (line 1171) | def test_multi_classes_evaluation_batch(self):
    method _test_evaluation_weights (line 1174) | def _test_evaluation_weights(self, n_classes):
    method test_binary_classes_evaluation_weights (line 1250) | def test_binary_classes_evaluation_weights(self):
    method test_multi_classes_evaluation_weights (line 1253) | def test_multi_classes_evaluation_weights(self):
  class BaselineClassifierPredictTest (line 1258) | class BaselineClassifierPredictTest(tf.test.TestCase):
    method setUp (line 1260) | def setUp(self):
    method tearDown (line 1263) | def tearDown(self):
    method _testPredictions (line 1267) | def _testPredictions(self, n_classes, label_vocabulary, label_output_fn):
    method testBinaryClassesWithoutLabelVocabulary (line 1334) | def testBinaryClassesWithoutLabelVocabulary(self):
    method testBinaryClassesWithLabelVocabulary (line 1341) | def testBinaryClassesWithLabelVocabulary(self):
    method testMultiClassesWithoutLabelVocabulary (line 1348) | def testMultiClassesWithoutLabelVocabulary(self):
    method testMultiClassesWithLabelVocabulary (line 1355) | def testMultiClassesWithLabelVocabulary(self):
  class BaselineClassifierIntegrationTest (line 1364) | class BaselineClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 1366) | def setUp(self):
    method tearDown (line 1369) | def tearDown(self):
    method _test_complete_flow (line 1373) | def _test_complete_flow(self, n_classes, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn (line 1404) | def _test_numpy_input_fn(self, n_classes):
    method test_binary_classes_numpy_input_fn (line 1440) | def test_binary_classes_numpy_input_fn(self):
    method test_multi_classes_numpy_input_fn (line 1443) | def test_multi_classes_numpy_input_fn(self):
    method _test_pandas_input_fn (line 1446) | def _test_pandas_input_fn(self, n_classes):
    method test_binary_classes_pandas_input_fn (line 1475) | def test_binary_classes_pandas_input_fn(self):
    method test_multi_classes_pandas_input_fn (line 1478) | def test_multi_classes_pandas_input_fn(self):
    method _test_input_fn_from_parse_example (line 1481) | def _test_input_fn_from_parse_example(self, n_classes):
    method test_binary_classes_input_fn_from_parse_example (line 1540) | def test_binary_classes_input_fn_from_parse_example(self):
    method test_multi_classes_input_fn_from_parse_example (line 1543) | def test_multi_classes_input_fn_from_parse_example(self):
  class BaselineLogitFnTest (line 1551) | class BaselineLogitFnTest(tf.test.TestCase):
    method test_basic_logit_correctness (line 1553) | def test_basic_logit_correctness(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_estimator_test_v1.py
  function _dnn_estimator_fn (line 36) | def _dnn_estimator_fn(weight_column=None, label_dimension=1, **kwargs):
  function _dnn_estimator_classifier_fn (line 47) | def _dnn_estimator_classifier_fn(n_classes=3, **kwargs):
  class DNNEstimatorEvaluateTest (line 55) | class DNNEstimatorEvaluateTest(
    method __init__ (line 58) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorPredictTest (line 65) | class DNNEstimatorPredictTest(dnn_testing_utils_v1.BaseDNNRegressorPredi...
    method __init__ (line 68) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorTrainTest (line 75) | class DNNEstimatorTrainTest(dnn_testing_utils_v1.BaseDNNRegressorTrainTest,
    method __init__ (line 78) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorWarmStartingTest (line 85) | class DNNEstimatorWarmStartingTest(dnn_testing_utils_v1.BaseDNNWarmStart...
    method __init__ (line 88) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNEstimatorIntegrationTest (line 95) | class DNNEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 97) | def setUp(self):
    method tearDown (line 100) | def tearDown(self):
    method _test_complete_flow (line 105) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 141) | def test_numpy_input_fn(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_linear_combined_estimator_test_v1.py
  function _dnn_only_estimator_fn (line 38) | def _dnn_only_estimator_fn(hidden_units,
  class DNNOnlyEstimatorEvaluateTest (line 65) | class DNNOnlyEstimatorEvaluateTest(
    method __init__ (line 68) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyEstimatorPredictTest (line 75) | class DNNOnlyEstimatorPredictTest(
    method __init__ (line 78) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyEstimatorTrainTest (line 85) | class DNNOnlyEstimatorTrainTest(dnn_testing_utils_v1.BaseDNNRegressorTra...
    method __init__ (line 88) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _linear_only_estimator_fn (line 94) | def _linear_only_estimator_fn(feature_columns,
  class LinearOnlyEstimatorEvaluateTest (line 117) | class LinearOnlyEstimatorEvaluateTest(
    method __init__ (line 121) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyEstimatorPredictTest (line 128) | class LinearOnlyEstimatorPredictTest(
    method __init__ (line 131) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyEstimatorTrainTest (line 138) | class LinearOnlyEstimatorTrainTest(
    method __init__ (line 141) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedEstimatorIntegrationTest (line 148) | class DNNLinearCombinedEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 150) | def setUp(self):
    method tearDown (line 153) | def tearDown(self):
    method _test_complete_flow (line 158) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 199) | def test_numpy_input_fn(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_linear_combined_test_v1.py
  class DNNOnlyModelFnTest (line 65) | class DNNOnlyModelFnTest(dnn_testing_utils_v1.BaseDNNModelFnTest,
    method __init__ (line 68) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method _dnn_only_model_fn (line 73) | def _dnn_only_model_fn(self,
  function _linear_regressor_fn (line 101) | def _linear_regressor_fn(feature_columns,
  class LinearOnlyRegressorPartitionerTest (line 121) | class LinearOnlyRegressorPartitionerTest(
    method __init__ (line 125) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorPartitionerV2Test (line 132) | class LinearOnlyRegressorPartitionerV2Test(
    method __init__ (line 136) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorEvaluationTest (line 143) | class LinearOnlyRegressorEvaluationTest(
    method __init__ (line 147) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorEvaluationV2Test (line 154) | class LinearOnlyRegressorEvaluationV2Test(
    method __init__ (line 158) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorPredictTest (line 165) | class LinearOnlyRegressorPredictTest(
    method __init__ (line 168) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorPredictV2Test (line 175) | class LinearOnlyRegressorPredictV2Test(
    method __init__ (line 178) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorIntegrationTest (line 185) | class LinearOnlyRegressorIntegrationTest(
    method __init__ (line 189) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorIntegrationV2Test (line 196) | class LinearOnlyRegressorIntegrationV2Test(
    method __init__ (line 200) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorTrainingTest (line 207) | class LinearOnlyRegressorTrainingTest(
    method __init__ (line 210) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyRegressorTrainingV2Test (line 217) | class LinearOnlyRegressorTrainingV2Test(
    method __init__ (line 220) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _linear_classifier_fn (line 226) | def _linear_classifier_fn(feature_columns,
  class LinearOnlyClassifierTrainingTest (line 248) | class LinearOnlyClassifierTrainingTest(
    method __init__ (line 251) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierTrainingV2Test (line 258) | class LinearOnlyClassifierTrainingV2Test(
    method __init__ (line 261) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierClassesEvaluationTest (line 270) | class LinearOnlyClassifierClassesEvaluationTest(
    method __init__ (line 274) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierClassesEvaluationV2Test (line 281) | class LinearOnlyClassifierClassesEvaluationV2Test(
    method __init__ (line 285) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierPredictTest (line 294) | class LinearOnlyClassifierPredictTest(
    method __init__ (line 297) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierPredictV2Test (line 304) | class LinearOnlyClassifierPredictV2Test(
    method __init__ (line 307) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierIntegrationTest (line 316) | class LinearOnlyClassifierIntegrationTest(
    method __init__ (line 320) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearOnlyClassifierIntegrationV2Test (line 327) | class LinearOnlyClassifierIntegrationV2Test(
    method __init__ (line 331) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedRegressorIntegrationTest (line 341) | class DNNLinearCombinedRegressorIntegrationTest(tf.test.TestCase):
    method setUp (line 343) | def setUp(self):
    method tearDown (line 346) | def tearDown(self):
    method _test_complete_flow_helper (line 351) | def _test_complete_flow_helper(self, linear_feature_columns,
    method _test_complete_flow (line 386) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method _test_complete_flow_mix1 (line 403) | def _test_complete_flow_mix1(self, train_input_fn, eval_input_fn,
    method _test_complete_flow_mix2 (line 421) | def _test_complete_flow_mix2(self, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn_helper (line 439) | def _test_numpy_input_fn_helper(self, fc_impl, fn_to_run):
    method test_numpy_input_fn_basic (line 466) | def test_numpy_input_fn_basic(self, fc_impl):
    method test_numpy_input_fn_mix1 (line 469) | def test_numpy_input_fn_mix1(self, fc_impl):
    method test_numpy_input_fn_mix2 (line 472) | def test_numpy_input_fn_mix2(self, fc_impl):
    method _test_pandas_input_fn_helper (line 475) | def _test_pandas_input_fn_helper(self, fc_impl, fn_to_run):
    method test_pandas_input_fn_basic (line 500) | def test_pandas_input_fn_basic(self, fc_impl):
    method test_pandas_input_fn_mix1 (line 503) | def test_pandas_input_fn_mix1(self, fc_impl):
    method test_pandas_input_fn_mix2 (line 506) | def test_pandas_input_fn_mix2(self, fc_impl):
    method _test_input_fn_from_parse_example_helper (line 509) | def _test_input_fn_from_parse_example_helper(self, fc_impl, fn_to_run):
    method test_input_fn_from_parse_example_basic (line 567) | def test_input_fn_from_parse_example_basic(self, fc_impl):
    method test_input_fn_from_parse_example_mix1 (line 571) | def test_input_fn_from_parse_example_mix1(self, fc_impl):
    method test_input_fn_from_parse_example_mix2 (line 575) | def test_input_fn_from_parse_example_mix2(self, fc_impl):
  function _dnn_classifier_fn (line 581) | def _dnn_classifier_fn(hidden_units,
  class DNNOnlyClassifierEvaluateTest (line 603) | class DNNOnlyClassifierEvaluateTest(
    method __init__ (line 606) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierEvaluateV2Test (line 613) | class DNNOnlyClassifierEvaluateV2Test(
    method __init__ (line 616) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierPredictTest (line 623) | class DNNOnlyClassifierPredictTest(
    method __init__ (line 626) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierPredictV2Test (line 633) | class DNNOnlyClassifierPredictV2Test(
    method __init__ (line 636) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierTrainTest (line 643) | class DNNOnlyClassifierTrainTest(
    method __init__ (line 646) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyClassifierTrainV2Test (line 653) | class DNNOnlyClassifierTrainV2Test(
    method __init__ (line 656) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 663) | def _dnn_regressor_fn(hidden_units,
  class DNNOnlyRegressorEvaluateTest (line 683) | class DNNOnlyRegressorEvaluateTest(
    method __init__ (line 686) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorEvaluateV2Test (line 693) | class DNNOnlyRegressorEvaluateV2Test(
    method __init__ (line 696) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorPredictTest (line 703) | class DNNOnlyRegressorPredictTest(
    method __init__ (line 706) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorPredictV2Test (line 713) | class DNNOnlyRegressorPredictV2Test(
    method __init__ (line 716) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorTrainTest (line 723) | class DNNOnlyRegressorTrainTest(dnn_testing_utils_v1.BaseDNNRegressorTra...
    method __init__ (line 726) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNOnlyRegressorTrainV2Test (line 733) | class DNNOnlyRegressorTrainV2Test(
    method __init__ (line 736) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLinearCombinedClassifierIntegrationTest (line 744) | class DNNLinearCombinedClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 746) | def setUp(self):
    method tearDown (line 749) | def tearDown(self):
    method _as_label (line 754) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 757) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 798) | def test_numpy_input_fn(self, fc_impl):
    method test_pandas_input_fn (line 828) | def test_pandas_input_fn(self, fc_impl):
    method test_input_fn_from_parse_example (line 854) | def test_input_fn_from_parse_example(self, fc_impl):
  class DNNLinearCombinedTests (line 918) | class DNNLinearCombinedTests(tf.test.TestCase):
    method setUp (line 920) | def setUp(self):
    method tearDown (line 923) | def tearDown(self):
    method _mock_optimizer (line 927) | def _mock_optimizer(self, real_optimizer, var_name_prefix):
    method test_train_op_calls_both_dnn_and_linear (line 949) | def test_train_op_calls_both_dnn_and_linear(self, fc_impl):
    method test_dnn_and_linear_logits_are_added (line 974) | def test_dnn_and_linear_logits_are_added(self, fc_impl):
  class DNNLinearCombinedWarmStartingTest (line 1003) | class DNNLinearCombinedWarmStartingTest(tf.test.TestCase):
    method setUp (line 1005) | def setUp(self):
    method tearDown (line 1019) | def tearDown(self):
    method test_classifier_basic_warm_starting (line 1024) | def test_classifier_basic_warm_starting(self, fc_impl):
    method test_regressor_basic_warm_starting (line 1064) | def test_regressor_basic_warm_starting(self, fc_impl):
    method test_warm_starting_selective_variables (line 1102) | def test_warm_starting_selective_variables(self, fc_impl):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_test_fc_v1_v1.py
  function _dnn_classifier_fn (line 53) | def _dnn_classifier_fn(*args, **kwargs):
  class DNNModelFnTest (line 58) | class DNNModelFnTest(dnn_testing_utils_v1.BaseDNNModelFnTest, tf.test.Te...
    method __init__ (line 60) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLogitFnTest (line 67) | class DNNLogitFnTest(dnn_testing_utils_v1.BaseDNNLogitFnTest, tf.test.Te...
    method __init__ (line 69) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNWarmStartingTest (line 76) | class DNNWarmStartingTest(dnn_testing_utils_v1.BaseDNNWarmStartingTest,
    method __init__ (line 79) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierEvaluateTest (line 86) | class DNNClassifierEvaluateTest(
    method __init__ (line 89) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierPredictTest (line 96) | class DNNClassifierPredictTest(
    method __init__ (line 99) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierTrainTest (line 106) | class DNNClassifierTrainTest(dnn_testing_utils_v1.BaseDNNClassifierTrain...
    method __init__ (line 109) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 115) | def _dnn_regressor_fn(*args, **kwargs):
  class DNNRegressorEvaluateTest (line 120) | class DNNRegressorEvaluateTest(
    method __init__ (line 123) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorPredictTest (line 130) | class DNNRegressorPredictTest(dnn_testing_utils_v1.BaseDNNRegressorPredi...
    method __init__ (line 133) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorTrainTest (line 140) | class DNNRegressorTrainTest(dnn_testing_utils_v1.BaseDNNRegressorTrainTest,
    method __init__ (line 143) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _queue_parsed_features (line 149) | def _queue_parsed_features(feature_map):
  class DNNRegressorIntegrationTest (line 165) | class DNNRegressorIntegrationTest(tf.test.TestCase, parameterized.TestCa...
    method setUp (line 167) | def setUp(self):
    method tearDown (line 170) | def tearDown(self):
    method _test_complete_flow (line 175) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 212) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 238) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 262) | def test_input_fn_from_parse_example(self):
  class DNNClassifierIntegrationTest (line 321) | class DNNClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 323) | def setUp(self):
    method tearDown (line 326) | def tearDown(self):
    method _as_label (line 331) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 334) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 371) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 400) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 425) | def test_input_fn_from_parse_example(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_test_fc_v2_v1.py
  function _dnn_classifier_fn (line 50) | def _dnn_classifier_fn(*args, **kwargs):
  class DNNModelFnV2Test (line 55) | class DNNModelFnV2Test(dnn_testing_utils_v1.BaseDNNModelFnTest,
    method __init__ (line 58) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNLogitFnV2Test (line 65) | class DNNLogitFnV2Test(dnn_testing_utils_v1.BaseDNNLogitFnTest,
    method __init__ (line 68) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNWarmStartingV2Test (line 75) | class DNNWarmStartingV2Test(dnn_testing_utils_v1.BaseDNNWarmStartingTest,
    method __init__ (line 78) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierEvaluateV2Test (line 85) | class DNNClassifierEvaluateV2Test(
    method __init__ (line 88) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierPredictV2Test (line 95) | class DNNClassifierPredictV2Test(
    method __init__ (line 98) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNClassifierTrainV2Test (line 105) | class DNNClassifierTrainV2Test(dnn_testing_utils_v1.BaseDNNClassifierTra...
    method __init__ (line 108) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _dnn_regressor_fn (line 114) | def _dnn_regressor_fn(*args, **kwargs):
  class DNNRegressorEvaluateV2Test (line 119) | class DNNRegressorEvaluateV2Test(
    method __init__ (line 122) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorPredictV2Test (line 129) | class DNNRegressorPredictV2Test(
    method __init__ (line 132) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class DNNRegressorTrainV2Test (line 139) | class DNNRegressorTrainV2Test(dnn_testing_utils_v1.BaseDNNRegressorTrain...
    method __init__ (line 142) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  function _queue_parsed_features (line 148) | def _queue_parsed_features(feature_map):
  class DNNRegressorIntegrationTest (line 164) | class DNNRegressorIntegrationTest(tf.test.TestCase, parameterized.TestCa...
    method setUp (line 166) | def setUp(self):
    method tearDown (line 169) | def tearDown(self):
    method _test_complete_flow (line 174) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 210) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 236) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 260) | def test_input_fn_from_parse_example(self):
  class DNNClassifierIntegrationTest (line 319) | class DNNClassifierIntegrationTest(tf.test.TestCase):
    method setUp (line 321) | def setUp(self):
    method tearDown (line 324) | def tearDown(self):
    method _as_label (line 329) | def _as_label(self, data_in_float):
    method _test_complete_flow (line 332) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 368) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 397) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 422) | def test_input_fn_from_parse_example(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/dnn_testing_utils_v1.py
  function assert_close (line 66) | def assert_close(expected, actual, rtol=1e-04, message='', name='assert_...
  function create_checkpoint (line 82) | def create_checkpoint(weights_and_biases,
  function mock_head (line 130) | def mock_head(testcase, hidden_units, logits_dimension, expected_logits):
  function mock_optimizer (line 190) | def mock_optimizer(testcase, hidden_units, expected_loss=None):
  class BaseDNNModelFnTest (line 240) | class BaseDNNModelFnTest(object):
    method __init__ (line 243) | def __init__(self, dnn_model_fn, fc_impl=feature_column):
    method setUp (line 247) | def setUp(self):
    method tearDown (line 250) | def tearDown(self):
    method _test_logits (line 255) | def _test_logits(self, mode, hidden_units, logits_dimension, inputs,
    method test_one_dim_logits (line 287) | def test_one_dim_logits(self):
    method test_multi_dim_logits (line 311) | def test_multi_dim_logits(self):
    method test_multi_example_multi_dim_logits (line 336) | def test_multi_example_multi_dim_logits(self):
    method test_multi_dim_input_one_dim_logits (line 365) | def test_multi_dim_input_one_dim_logits(self):
    method test_multi_dim_input_multi_dim_logits (line 390) | def test_multi_dim_input_multi_dim_logits(self):
    method test_multi_feature_column_multi_dim_logits (line 415) | def test_multi_feature_column_multi_dim_logits(self):
    method test_multi_feature_column_mix_multi_dim_logits (line 466) | def test_multi_feature_column_mix_multi_dim_logits(self):
    method test_features_tensor_raises_value_error (line 517) | def test_features_tensor_raises_value_error(self):
  class BaseDNNLogitFnTest (line 545) | class BaseDNNLogitFnTest(object):
    method __init__ (line 548) | def __init__(self, dnn_logit_fn_builder, fc_impl=feature_column):
    method setUp (line 552) | def setUp(self):
    method tearDown (line 555) | def tearDown(self):
    method _test_logits (line 560) | def _test_logits(self,
    method test_one_dim_logits (line 594) | def test_one_dim_logits(self):
    method test_one_dim_logits_with_batch_norm (line 617) | def test_one_dim_logits_with_batch_norm(self):
    method test_multi_dim_logits (line 682) | def test_multi_dim_logits(self):
    method test_multi_example_multi_dim_logits (line 706) | def test_multi_example_multi_dim_logits(self):
    method test_multi_dim_input_one_dim_logits (line 734) | def test_multi_dim_input_one_dim_logits(self):
    method test_multi_dim_input_multi_dim_logits (line 759) | def test_multi_dim_input_multi_dim_logits(self):
    method test_multi_feature_column_multi_dim_logits (line 783) | def test_multi_feature_column_multi_dim_logits(self):
    method test_multi_feature_column_mix_multi_dim_logits (line 834) | def test_multi_feature_column_mix_multi_dim_logits(self):
  class BaseDNNWarmStartingTest (line 886) | class BaseDNNWarmStartingTest(object):
    method __init__ (line 888) | def __init__(self,
    method setUp (line 896) | def setUp(self):
    method tearDown (line 911) | def tearDown(self):
    method assertAllNotClose (line 916) | def assertAllNotClose(self, t1, t2):
    method test_classifier_basic_warm_starting (line 927) | def test_classifier_basic_warm_starting(self):
    method test_regressor_basic_warm_starting (line 960) | def test_regressor_basic_warm_starting(self):
    method test_warm_starting_selective_variables (line 991) | def test_warm_starting_selective_variables(self):
    method test_warm_starting_with_vocab_remapping_and_partitioning (line 1041) | def test_warm_starting_with_vocab_remapping_and_partitioning(self):
    method test_warm_starting_with_naming_change (line 1139) | def test_warm_starting_with_naming_change(self):
  class BaseDNNClassifierEvaluateTest (line 1190) | class BaseDNNClassifierEvaluateTest(object):
    method __init__ (line 1192) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column):
    method setUp (line 1196) | def setUp(self):
    method tearDown (line 1199) | def tearDown(self):
    method test_one_dim (line 1204) | def test_one_dim(self):
    method test_multi_dim (line 1246) | def test_multi_dim(self):
    method test_float_labels (line 1283) | def test_float_labels(self):
    method test_multi_dim_weights (line 1310) | def test_multi_dim_weights(self):
  class BaseDNNRegressorEvaluateTest (line 1341) | class BaseDNNRegressorEvaluateTest(object):
    method __init__ (line 1343) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column):
    method setUp (line 1347) | def setUp(self):
    method tearDown (line 1350) | def tearDown(self):
    method test_one_dim (line 1355) | def test_one_dim(self):
    method test_multi_dim (line 1387) | def test_multi_dim(self):
    method test_multi_dim_weights (line 1422) | def test_multi_dim_weights(self):
  class BaseDNNClassifierPredictTest (line 1452) | class BaseDNNClassifierPredictTest(object):
    method __init__ (line 1454) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column):
    method setUp (line 1458) | def setUp(self):
    method tearDown (line 1461) | def tearDown(self):
    method _test_one_dim (line 1466) | def _test_one_dim(self, label_vocabulary, label_output_fn):
    method test_one_dim_without_label_vocabulary (line 1502) | def test_one_dim_without_label_vocabulary(self):
    method test_one_dim_with_label_vocabulary (line 1506) | def test_one_dim_with_label_vocabulary(self):
    method _test_multi_dim_with_3_classes (line 1512) | def _test_multi_dim_with_3_classes(self, label_vocabulary, label_outpu...
    method test_multi_dim_with_3_classes_but_no_label_vocab (line 1559) | def test_multi_dim_with_3_classes_but_no_label_vocab(self):
    method test_multi_dim_with_3_classes_and_label_vocab (line 1563) | def test_multi_dim_with_3_classes_and_label_vocab(self):
  class BaseDNNRegressorPredictTest (line 1570) | class BaseDNNRegressorPredictTest(object):
    method __init__ (line 1572) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column):
    method setUp (line 1576) | def setUp(self):
    method tearDown (line 1579) | def tearDown(self):
    method test_one_dim (line 1584) | def test_one_dim(self):
    method test_multi_dim (line 1608) | def test_multi_dim(self):
  class _SummaryHook (line 1637) | class _SummaryHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 1640) | def __init__(self):
    method begin (line 1643) | def begin(self):
    method before_run (line 1646) | def before_run(self, run_context):
    method after_run (line 1649) | def after_run(self, run_context, run_values):
    method summaries (line 1654) | def summaries(self):
  function _assert_checkpoint (line 1658) | def _assert_checkpoint(testcase, global_step, input_units, hidden_units,
  function _assert_simple_summary (line 1694) | def _assert_simple_summary(testcase, expected_values, actual_summary):
  class BaseDNNClassifierTrainTest (line 1710) | class BaseDNNClassifierTrainTest(object):
    method __init__ (line 1712) | def __init__(self, dnn_classifier_fn, fc_impl=feature_column):
    method setUp (line 1716) | def setUp(self):
    method tearDown (line 1719) | def tearDown(self):
    method test_from_scratch_with_default_optimizer_binary (line 1724) | def test_from_scratch_with_default_optimizer_binary(self):
    method test_from_scratch_with_default_optimizer_multi_class (line 1745) | def test_from_scratch_with_default_optimizer_multi_class(self):
    method test_from_scratch_validate_summary (line 1768) | def test_from_scratch_validate_summary(self):
    method test_binary_classification (line 1803) | def test_binary_classification(self):
    method test_binary_classification_float_labels (line 1856) | def test_binary_classification_float_labels(self):
    method test_multi_class (line 1888) | def test_multi_class(self):
  class BaseDNNRegressorTrainTest (line 1944) | class BaseDNNRegressorTrainTest(object):
    method __init__ (line 1946) | def __init__(self, dnn_regressor_fn, fc_impl=feature_column):
    method setUp (line 1950) | def setUp(self):
    method tearDown (line 1953) | def tearDown(self):
    method test_from_scratch_with_default_optimizer (line 1958) | def test_from_scratch_with_default_optimizer(self):
    method test_from_scratch (line 1979) | def test_from_scratch(self):
    method test_one_dim (line 2014) | def test_one_dim(self):
    method test_multi_dim (line 2068) | def test_multi_dim(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/linear_estimator_test_v1.py
  function _linear_estimator_fn (line 36) | def _linear_estimator_fn(weight_column=None, label_dimension=1, **kwargs):
  class LinearEstimatorEvaluateTest (line 48) | class LinearEstimatorEvaluateTest(
    method __init__ (line 52) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorPredictTest (line 59) | class LinearEstimatorPredictTest(
    method __init__ (line 62) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorTrainTest (line 69) | class LinearEstimatorTrainTest(
    method __init__ (line 72) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearEstimatorIntegrationTest (line 79) | class LinearEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 81) | def setUp(self):
    method tearDown (line 84) | def tearDown(self):
    method _test_complete_flow (line 89) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 124) | def test_numpy_input_fn(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/linear_test_v1.py
  function _linear_regressor_fn (line 31) | def _linear_regressor_fn(*args, **kwargs):
  function _linear_classifier_fn (line 35) | def _linear_classifier_fn(*args, **kwargs):
  class LinearRegressorPartitionerTest (line 43) | class LinearRegressorPartitionerTest(
    method __init__ (line 47) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorPartitionerV2Test (line 54) | class LinearRegressorPartitionerV2Test(
    method __init__ (line 58) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorEvaluationTest (line 65) | class LinearRegressorEvaluationTest(
    method __init__ (line 69) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorEvaluationV2Test (line 76) | class LinearRegressorEvaluationV2Test(
    method __init__ (line 80) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorPredictTest (line 87) | class LinearRegressorPredictTest(
    method __init__ (line 90) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorPredictV2Test (line 97) | class LinearRegressorPredictV2Test(
    method __init__ (line 100) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorIntegrationTest (line 107) | class LinearRegressorIntegrationTest(
    method __init__ (line 111) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorIntegrationV2Test (line 118) | class LinearRegressorIntegrationV2Test(
    method __init__ (line 122) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorTrainingTest (line 129) | class LinearRegressorTrainingTest(
    method __init__ (line 132) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearRegressorTrainingV2Test (line 139) | class LinearRegressorTrainingV2Test(
    method __init__ (line 142) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierTrainingTest (line 150) | class LinearClassifierTrainingTest(
    method __init__ (line 153) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierTrainingV2Test (line 160) | class LinearClassifierTrainingV2Test(
    method __init__ (line 163) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierEvaluationTest (line 172) | class LinearClassifierEvaluationTest(
    method __init__ (line 176) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierEvaluationV2Test (line 183) | class LinearClassifierEvaluationV2Test(
    method __init__ (line 187) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierPredictTest (line 196) | class LinearClassifierPredictTest(
    method __init__ (line 199) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierPredictV2Test (line 206) | class LinearClassifierPredictV2Test(
    method __init__ (line 209) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierIntegrationTest (line 218) | class LinearClassifierIntegrationTest(
    method __init__ (line 222) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearClassifierIntegrationV2Test (line 229) | class LinearClassifierIntegrationV2Test(
    method __init__ (line 233) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearLogitFnTest (line 243) | class LinearLogitFnTest(linear_testing_utils_v1.BaseLinearLogitFnTest,
    method __init__ (line 246) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearLogitFnV2Test (line 253) | class LinearLogitFnV2Test(linear_testing_utils_v1.BaseLinearLogitFnTest,
    method __init__ (line 256) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearWarmStartingTest (line 264) | class LinearWarmStartingTest(linear_testing_utils_v1.BaseLinearWarmStart...
    method __init__ (line 267) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class LinearWarmStartingV2Test (line 277) | class LinearWarmStartingV2Test(
    method __init__ (line 280) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
  class ComputeFractionOfZeroTest (line 290) | class ComputeFractionOfZeroTest(tf.test.TestCase):
    method _assertSparsity (line 292) | def _assertSparsity(self, expected_sparsity, tensor):
    method test_small_float32 (line 297) | def test_small_float32(self):
    method test_small_int32 (line 303) | def test_small_int32(self):
    method test_small_float64 (line 307) | def test_small_float64(self):
    method test_small_int64 (line 311) | def test_small_int64(self):
    method test_nested (line 315) | def test_nested(self):
    method test_none (line 320) | def test_none(self):
    method test_empty (line 324) | def test_empty(self):
    method test_multiple_empty (line 331) | def test_multiple_empty(self):
    method test_some_empty (line 341) | def test_some_empty(self):
    method test_mixed_types (line 349) | def test_mixed_types(self):
    method test_2_27_zeros__using_512_MiB_of_ram (line 356) | def test_2_27_zeros__using_512_MiB_of_ram(self):
    method test_2_27_ones__using_512_MiB_of_ram (line 360) | def test_2_27_ones__using_512_MiB_of_ram(self):

FILE: tensorflow_estimator/python/estimator/canned/v1/linear_testing_utils_v1.py
  function assert_close (line 72) | def assert_close(expected, actual, rtol=1e-04, name='assert_close'):
  function save_variables_to_ckpt (line 87) | def save_variables_to_ckpt(model_dir):
  function queue_parsed_features (line 94) | def queue_parsed_features(feature_map):
  function sorted_key_dict (line 109) | def sorted_key_dict(unsorted_dict):
  function sigmoid (line 113) | def sigmoid(x):
  class CheckPartitionerVarHook (line 117) | class CheckPartitionerVarHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 120) | def __init__(self, test_case, var_name, var_dim, partitions):
    method begin (line 126) | def begin(self):
  class BaseLinearRegressorPartitionerTest (line 139) | class BaseLinearRegressorPartitionerTest(object):
    method __init__ (line 141) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 145) | def setUp(self):
    method tearDown (line 148) | def tearDown(self):
    method testPartitioner (line 153) | def testPartitioner(self):
    method testDefaultPartitionerWithMultiplePsReplicas (line 181) | def testDefaultPartitionerWithMultiplePsReplicas(self):
  class BaseLinearRegressorEvaluationTest (line 219) | class BaseLinearRegressorEvaluationTest(object):
    method __init__ (line 221) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 225) | def setUp(self):
    method tearDown (line 228) | def tearDown(self):
    method test_evaluation_for_simple_data (line 233) | def test_evaluation_for_simple_data(self):
    method test_evaluation_batch (line 259) | def test_evaluation_batch(self):
    method test_evaluation_weights (line 289) | def test_evaluation_weights(self):
    method test_evaluation_for_multi_dimensions (line 322) | def test_evaluation_for_multi_dimensions(self):
    method test_evaluation_for_multiple_feature_columns (line 358) | def test_evaluation_for_multiple_feature_columns(self):
    method test_evaluation_for_multiple_feature_columns_mix (line 396) | def test_evaluation_for_multiple_feature_columns_mix(self):
  class BaseLinearRegressorPredictTest (line 436) | class BaseLinearRegressorPredictTest(object):
    method __init__ (line 438) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 442) | def setUp(self):
    method tearDown (line 445) | def tearDown(self):
    method test_1d (line 450) | def test_1d(self):
    method testMultiDim (line 473) | def testMultiDim(self):
    method testTwoFeatureColumns (line 506) | def testTwoFeatureColumns(self):
    method testTwoFeatureColumnsMix (line 534) | def testTwoFeatureColumnsMix(self):
    method testSparseCombiner (line 559) | def testSparseCombiner(self):
  class BaseLinearRegressorIntegrationTest (line 622) | class BaseLinearRegressorIntegrationTest(object):
    method __init__ (line 624) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 628) | def setUp(self):
    method tearDown (line 631) | def tearDown(self):
    method _test_complete_flow (line 636) | def _test_complete_flow(self, train_input_fn, eval_input_fn, predict_i...
    method test_numpy_input_fn (line 669) | def test_numpy_input_fn(self):
    method test_pandas_input_fn (line 705) | def test_pandas_input_fn(self):
    method test_input_fn_from_parse_example (line 734) | def test_input_fn_from_parse_example(self):
  class BaseLinearRegressorTrainingTest (line 795) | class BaseLinearRegressorTrainingTest(object):
    method __init__ (line 797) | def __init__(self, linear_regressor_fn, fc_lib=feature_column):
    method setUp (line 801) | def setUp(self):
    method tearDown (line 804) | def tearDown(self):
    method _mock_optimizer (line 809) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 847) | def _assert_checkpoint(self,
    method testFromScratchWithDefaultOptimizer (line 872) | def testFromScratchWithDefaultOptimizer(self):
    method testTrainWithOneDimLabel (line 888) | def testTrainWithOneDimLabel(self):
    method testTrainWithOneDimWeight (line 908) | def testTrainWithOneDimWeight(self):
    method testFromScratch (line 933) | def testFromScratch(self):
    method testFromCheckpoint (line 957) | def testFromCheckpoint(self):
    method testFromCheckpointMultiBatch (line 992) | def testFromCheckpointMultiBatch(self):
  class BaseLinearClassifierTrainingTest (line 1031) | class BaseLinearClassifierTrainingTest(object):
    method __init__ (line 1033) | def __init__(self, linear_classifier_fn, fc_lib=feature_column):
    method setUp (line 1037) | def setUp(self):
    method tearDown (line 1040) | def tearDown(self):
    method _mock_optimizer (line 1044) | def _mock_optimizer(self, expected_loss=None):
    method _assert_checkpoint (line 1078) | def _assert_checkpoint(self,
    method _testFromScratchWithDefaultOptimizer (line 1107) | def _testFromScratchWithDefaultOptimizer(self, n_classes):
    method testBinaryClassesFromScratchWithDefaultOptimizer (line 1123) | def testBinaryClassesFromScratchWithDefaultOptimizer(self):
    method testMultiClassesFromScratchWithDefaultOptimizer (line 1126) | def testMultiClassesFromScratchWithDefaultOptimizer(self):
    method _testTrainWithTwoDimsLabel (line 1129) | def _testTrainWithTwoDimsLabel(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsLabel (line 1150) | def testBinaryClassesTrainWithTwoDimsLabel(self):
    method testMultiClassesTrainWithTwoDimsLabel (line 1153) | def testMultiClassesTrainWithTwoDimsLabel(self):
    method _testTrainWithOneDimLabel (line 1156) | def _testTrainWithOneDimLabel(self, n_classes):
    method testBinaryClassesTrainWithOneDimLabel (line 1175) | def testBinaryClassesTrainWithOneDimLabel(self):
    method testMultiClassesTrainWithOneDimLabel (line 1178) | def testMultiClassesTrainWithOneDimLabel(self):
    method _testTrainWithTwoDimsWeight (line 1181) | def _testTrainWithTwoDimsWeight(self, n_classes):
    method testBinaryClassesTrainWithTwoDimsWeight (line 1206) | def testBinaryClassesTrainWithTwoDimsWeight(self):
    method testMultiClassesTrainWithTwoDimsWeight (line 1209) | def testMultiClassesTrainWithTwoDimsWeight(self):
    method _testTrainWithOneDimWeight (line 1212) | def _testTrainWithOneDimWeight(self, n_classes):
    method testBinaryClassesTrainWithOneDimWeight (line 1235) | def testBinaryClassesTrainWithOneDimWeight(self):
    method testMultiClassesTrainWithOneDimWeight (line 1238) | def testMultiClassesTrainWithOneDimWeight(self):
    method _testFromScratch (line 1241) | def _testFromScratch(self, n_classes):
    method testBinaryClassesFromScratch (line 1277) | def testBinaryClassesFromScratch(self):
    method testMultiClassesFromScratch (line 1280) | def testMultiClassesFromScratch(self):
    method _testFromCheckpoint (line 1283) | def _testFromCheckpoint(self, n_classes):
    method testBinaryClassesFromCheckpoint (line 1342) | def testBinaryClassesFromCheckpoint(self):
    method testMultiClassesFromCheckpoint (line 1345) | def testMultiClassesFromCheckpoint(self):
    method _testFromCheckpointFloatLabels (line 1348) | def _testFromCheckpointFloatLabels(self, n_classes):
    method testBinaryClassesFromCheckpointFloatLabels (line 1387) | def testBinaryClassesFromCheckpointFloatLabels(self):
    method testMultiClassesFromCheckpointFloatLabels (line 1390) | def testMultiClassesFromCheckpointFloatLabels(self):
    method _testFromCheckpointMultiBatch (line 1393) | def _testFromCheckpointMultiBatch(self, n_classes):
    method testBinaryClassesFromCheckpointMultiBatch (line 1457) | def testBinaryClassesFromCheckpointMultiBatch(self):
    method testMultiClassesFromCheckpointMultiBatch (line 1460) | def testMultiClassesFromCheckpointMultiBatch(self):
  class BaseLinearClassifierEvaluationTest (line 1464) | class BaseLinearClassifierEvaluationTest(object):
    method __init__ (line 1466) | def __init__(self, linear_classifier_fn, fc_lib=feature_column):
    method setUp (line 1470) | def setUp(self):
    method tearDown (line 1473) | def tearDown(self):
    method _test_evaluation_for_simple_data (line 1477) | def _test_evaluation_for_simple_data(self, n_classes):
    method test_binary_classes_evaluation_for_simple_data (line 1539) | def test_binary_classes_evaluation_for_simple_data(self):
    method test_multi_classes_evaluation_for_simple_data (line 1542) | def test_multi_classes_evaluation_for_simple_data(self):
    method _test_evaluation_batch (line 1545) | def _test_evaluation_batch(self, n_classes):
    method test_binary_classes_evaluation_batch (line 1617) | def test_binary_classes_evaluation_batch(self):
    method test_multi_classes_evaluation_batch (line 1620) | def test_multi_classes_evaluation_batch(self):
    method _test_evaluation_weights (line 1623) | def _test_evaluation_weights(self, n_classes):
    method test_binary_classes_evaluation_weights (line 1708) | def test_binary_classes_evaluation_weights(self):
    method test_multi_classes_evaluation_weights (line 1711) | def test_multi_classes_evaluation_weights(self):
  class BaseLinearClassifierPredictTest (line 1715) | class BaseLinearClassifierPredictTest(object):
    method __init__ (line 1717) | def __init__(self, linear_classifier_fn, fc_lib=feature_column):
    method setUp (line 1721) | def setUp(self):
    method tearDown (line 1724) | def tearDown(self):
    method _testPredictions (line 1728) | def _testPredictions(self, n_classes, label_vocabulary, label_output_fn):
    method testBinaryClassesWithoutLabelVocabulary (line 1803) | def testBinaryClassesWithoutLabelVocabulary(self):
    method testBinaryClassesWithLabelVocabulary (line 1810) | def testBinaryClassesWithLabelVocabulary(self):
    method testMultiClassesWithoutLabelVocabulary (line 1817) | def testMultiClassesWithoutLabelVocabulary(self):
    method testMultiClassesWithLabelVocabulary (line 1824) | def testMultiClassesWithLabelVocabulary(self):
    method testSparseCombiner (line 1831) | def testSparseCombiner(self):
  class BaseLinearClassifierIntegrationTest (line 1894) | class BaseLinearClassifierIntegrationTest(object):
    method __init__ (line 1896) | def __init__(self, linear_classifier_fn, fc_lib=feature_column):
    method setUp (line 1900) | def setUp(self):
    method tearDown (line 1903) | def tearDown(self):
    method _test_complete_flow (line 1907) | def _test_complete_flow(self, n_classes, train_input_fn, eval_input_fn,
    method _test_numpy_input_fn (line 1940) | def _test_numpy_input_fn(self, n_classes):
    method test_binary_classes_numpy_input_fn (line 1976) | def test_binary_classes_numpy_input_fn(self):
    method test_multi_classes_numpy_input_fn (line 1979) | def test_multi_classes_numpy_input_fn(self):
    method _test_pandas_input_fn (line 1982) | def _test_pandas_input_fn(self, n_classes):
    method test_binary_classes_pandas_input_fn (line 2011) | def test_binary_classes_pandas_input_fn(self):
    method test_multi_classes_pandas_input_fn (line 2014) | def test_multi_classes_pandas_input_fn(self):
    method _test_input_fn_from_parse_example (line 2017) | def _test_input_fn_from_parse_example(self, n_classes):
    method test_binary_classes_input_fn_from_parse_example (line 2076) | def test_binary_classes_input_fn_from_parse_example(self):
    method test_multi_classes_input_fn_from_parse_example (line 2079) | def test_multi_classes_input_fn_from_parse_example(self):
  class BaseLinearLogitFnTest (line 2083) | class BaseLinearLogitFnTest(object):
    method __init__ (line 2085) | def __init__(self, fc_lib=feature_column):
    method test_basic_logit_correctness (line 2088) | def test_basic_logit_correctness(self):
    method test_compute_fraction_of_zero (line 2109) | def test_compute_fraction_of_zero(self):
    method test_compute_fraction_of_zero_v2 (line 2141) | def test_compute_fraction_of_zero_v2(self):
  class BaseLinearWarmStartingTest (line 2173) | class BaseLinearWarmStartingTest(object):
    method __init__ (line 2175) | def __init__(self,
    method setUp (line 2183) | def setUp(self):
    method tearDown (line 2198) | def tearDown(self):
    method test_classifier_basic_warm_starting (line 2203) | def test_classifier_basic_warm_starting(self):
    method test_regressor_basic_warm_starting (line 2231) | def test_regressor_basic_warm_starting(self):
    method test_warm_starting_selective_variables (line 2257) | def test_warm_starting_selective_variables(self):
    method test_warm_starting_with_vocab_remapping_and_partitioning (line 2291) | def test_warm_starting_with_vocab_remapping_and_partitioning(self):
    method test_warm_starting_with_naming_change (line 2373) | def test_warm_starting_with_naming_change(self):

FILE: tensorflow_estimator/python/estimator/distribute_strategy_estimator_integration_test.py
  class DNNLinearCombinedClassifierIntegrationTest (line 36) | class DNNLinearCombinedClassifierIntegrationTest(tf.test.TestCase,
    method setUp (line 39) | def setUp(self):
    method dataset_input_fn (line 42) | def dataset_input_fn(self, x, y, batch_size, shuffle):
    method test_estimator_with_strategy_hooks (line 62) | def test_estimator_with_strategy_hooks(self, distribution,
    method test_complete_flow_with_mode (line 117) | def test_complete_flow_with_mode(self, distribution, use_train_and_eva...
    method tearDown (line 181) | def tearDown(self):

FILE: tensorflow_estimator/python/estimator/distribute_strategy_estimator_training_test.py
  class DistributeCoordinatorIntegrationTest (line 61) | class DistributeCoordinatorIntegrationTest(
    method setUpClass (line 65) | def setUpClass(cls):
    method setUp (line 71) | def setUp(self):
    method dataset_input_fn (line 75) | def dataset_input_fn(self, x, y, batch_size, shuffle):
    method _get_exporter (line 86) | def _get_exporter(self, name, fc):
    method _extract_loss_and_global_step (line 93) | def _extract_loss_and_global_step(self, event_folder):
    method _get_estimator (line 119) | def _get_estimator(self,
    method _complete_flow (line 145) | def _complete_flow(self,
    method _inspect_train_and_eval_events (line 205) | def _inspect_train_and_eval_events(self, estimator):
    method _make_cross_device_ops (line 236) | def _make_cross_device_ops(self, num_gpus_per_worker):
    method _get_strategy_object (line 239) | def _get_strategy_object(self,
    method test_complete_flow_standalone_client (line 283) | def test_complete_flow_standalone_client(self, train_distribute_cls,
    method test_complete_flow_standalone_client_collective_nccl (line 313) | def test_complete_flow_standalone_client_collective_nccl(
    method test_estimator_standalone_client (line 343) | def test_estimator_standalone_client(self, train_distribute_cls,
    method _mock_run_std_server (line 363) | def _mock_run_std_server(self, *args, **kwargs):
    method _independent_worker_fn (line 370) | def _independent_worker_fn(
    method test_complete_flow_independent_worker_between_graph (line 395) | def test_complete_flow_independent_worker_between_graph(
    method test_complete_flow_independent_worker_in_graph (line 447) | def test_complete_flow_independent_worker_in_graph(self, train_distrib...
  class RunConfigTest (line 493) | class RunConfigTest(tf.compat.v1.test.TestCase):
    method test_previously_unexpected_cluster_spec (line 495) | def test_previously_unexpected_cluster_spec(self):
    method test_should_run_distribute_coordinator (line 503) | def test_should_run_distribute_coordinator(self):
    method test_init_run_config_duplicate_distribute (line 546) | def test_init_run_config_duplicate_distribute(self):
    method test_init_run_config_none_distribute_coordinator_mode (line 559) | def test_init_run_config_none_distribute_coordinator_mode(self):
    method test_init_run_config_independent_worker (line 580) | def test_init_run_config_independent_worker(self):
    method test_init_run_config_standalone_client (line 590) | def test_init_run_config_standalone_client(self):
  function new_init (line 606) | def new_init(*args, **kwargs):

FILE: tensorflow_estimator/python/estimator/early_stopping.py
  function make_early_stopping_hook (line 31) | def make_early_stopping_hook(estimator,
  function stop_if_higher_hook (line 100) | def stop_if_higher_hook(estimator,
  function stop_if_lower_hook (line 157) | def stop_if_lower_hook(estimator,
  function stop_if_no_increase_hook (line 214) | def stop_if_no_increase_hook(estimator,
  function stop_if_no_decrease_hook (line 272) | def stop_if_no_decrease_hook(estimator,
  function read_eval_metrics (line 329) | def read_eval_metrics(eval_dir):
  function _stop_if_threshold_crossed_hook (line 352) | def _stop_if_threshold_crossed_hook(estimator, metric_name, threshold,
  function _stop_if_no_metric_improvement_hook (line 387) | def _stop_if_no_metric_improvement_hook(estimator, metric_name,
  function _summaries (line 429) | def _summaries(eval_dir):
  function _get_or_create_stop_var (line 453) | def _get_or_create_stop_var():
  class _StopOnPredicateHook (line 467) | class _StopOnPredicateHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 470) | def __init__(self, should_stop_fn, run_every_secs=60, run_every_steps=...
    method begin (line 481) | def begin(self):
    method before_run (line 486) | def before_run(self, run_context):
    method after_run (line 490) | def after_run(self, run_context, run_values):
  class _CheckForStoppingHook (line 501) | class _CheckForStoppingHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 504) | def __init__(self):
    method begin (line 507) | def begin(self):
    method before_run (line 510) | def before_run(self, run_context):
    method after_run (line 514) | def after_run(self, run_context, run_values):
  class _MultiWorkerEarlyStoppingHook (line 521) | class _MultiWorkerEarlyStoppingHook(tf.compat.v1.train.SessionRunHook):
    method _get_or_create_stop_var_with_aggregation (line 524) | def _get_or_create_stop_var_with_aggregation(self):
    method __init__ (line 539) | def __init__(self, should_stop_fn, run_every_steps=None):
    method begin (line 551) | def begin(self):
    method before_run (line 569) | def before_run(self, run_context):
    method after_run (line 576) | def after_run(self, run_context, run_values):

FILE: tensorflow_estimator/python/estimator/early_stopping_test.py
  class _FakeRunConfig (line 33) | class _FakeRunConfig(run_config.RunConfig):
    method __init__ (line 35) | def __init__(self, is_chief):
    method is_chief (line 40) | def is_chief(self):
  function _dummy_model_fn (line 44) | def _dummy_model_fn(features, labels, params):
  class _FakeEstimator (line 48) | class _FakeEstimator(estimator.Estimator):
    method __init__ (line 51) | def __init__(self, config):
  function _write_events (line 56) | def _write_events(eval_dir, params):
  class ReadEvalMetricsTest (line 66) | class ReadEvalMetricsTest(tf.test.TestCase):
    method test_read_eval_metrics (line 68) | def test_read_eval_metrics(self):
    method test_data_loss_error_ignored (line 94) | def test_data_loss_error_ignored(self):
    method test_read_eval_metrics_when_no_events (line 126) | def test_read_eval_metrics_when_no_events(self):
  class EarlyStoppingHooksTest (line 140) | class EarlyStoppingHooksTest(tf.test.TestCase, parameterized.TestCase):
    method setUp (line 142) | def setUp(self):
    method run_session (line 158) | def run_session(self, hooks, should_stop):
    method test_stop_if_higher_hook (line 168) | def test_stop_if_higher_hook(self, threshold, min_steps, should_stop):
    method test_stop_if_lower_hook (line 177) | def test_stop_if_lower_hook(self, threshold, min_steps, should_stop):
    method test_stop_if_no_increase_hook (line 187) | def test_stop_if_no_increase_hook(self, max_steps, min_steps, should_s...
    method test_stop_if_no_decrease_hook (line 197) | def test_stop_if_no_decrease_hook(self, max_steps, min_steps, should_s...
    method test_multiple_hooks (line 207) | def test_multiple_hooks(self, max_steps, loss_threshold, should_stop):
    method test_make_early_stopping_hook (line 218) | def test_make_early_stopping_hook(self, should_stop):
    method test_make_early_stopping_hook_typeerror (line 224) | def test_make_early_stopping_hook_typeerror(self):
    method test_make_early_stopping_hook_valueerror (line 229) | def test_make_early_stopping_hook_valueerror(self):
  class StopOnPredicateHookTest (line 238) | class StopOnPredicateHookTest(tf.test.TestCase):
    method test_stop (line 240) | def test_stop(self):
  class CheckForStoppingHookTest (line 264) | class CheckForStoppingHookTest(tf.test.TestCase):
    method test_stop (line 266) | def test_stop(self):

FILE: tensorflow_estimator/python/estimator/estimator.py
  class Estimator (line 75) | class Estimator(object):
    method __init__ (line 131) | def __init__(self,
    method model_dir (line 215) | def model_dir(self):
    method config (line 219) | def config(self):
    method params (line 223) | def params(self):
    method model_fn (line 227) | def model_fn(self):
    method get_variable_value (line 241) | def get_variable_value(self, name):
    method get_variable_names (line 257) | def get_variable_names(self):
    method latest_checkpoint (line 270) | def latest_checkpoint(self):
    method train (line 280) | def train(self,
    method _convert_train_steps_to_hooks (line 362) | def _convert_train_steps_to_hooks(self, steps, max_steps):
    method eval_dir (line 387) | def eval_dir(self, name=None):
    method evaluate (line 401) | def evaluate(self,
    method _actual_eval (line 478) | def _actual_eval(self,
    method _convert_eval_steps_to_hooks (line 521) | def _convert_eval_steps_to_hooks(self, steps):
    method predict (line 553) | def predict(self,
    method _assert_members_are_not_overridden (line 653) | def _assert_members_are_not_overridden(self):
    method export_saved_model (line 657) | def export_saved_model(self,
    method experimental_export_all_saved_models (line 736) | def experimental_export_all_saved_models(self,
    method _export_all_saved_models (line 810) | def _export_all_saved_models(self,
    method _add_meta_graph_for_mode (line 893) | def _add_meta_graph_for_mode(self,
    method _get_features_from_input_fn (line 1014) | def _get_features_from_input_fn(self, input_fn, mode):
    method _validate_features_in_predict_input (line 1021) | def _validate_features_in_predict_input(self, result):
    method _get_iterator_from_input_fn (line 1029) | def _get_iterator_from_input_fn(self, input_fn, mode, distribution=None):
    method _get_features_and_labels_from_input_fn (line 1045) | def _get_features_and_labels_from_input_fn(self, input_fn, mode):
    method _extract_batch_length (line 1050) | def _extract_batch_length(self, preds_evaluated):
    method _extract_keys (line 1060) | def _extract_keys(self, predictions, predict_keys):
    method _create_global_step (line 1078) | def _create_global_step(self, graph):
    method _create_and_assert_global_step (line 1092) | def _create_and_assert_global_step(self, graph):
    method _call_input_fn (line 1106) | def _call_input_fn(self, input_fn, mode, input_context=None):
    method _call_model_fn (line 1143) | def _call_model_fn(self, features, labels, mode, config):
    method _train_model (line 1182) | def _train_model(self, input_fn, hooks, saving_listeners):
    method _train_model_default (line 1188) | def _train_model_default(self, input_fn, hooks, saving_listeners):
    method _train_model_distributed (line 1221) | def _train_model_distributed(self, input_fn, hooks, saving_listeners):
    method _actual_train_model_distributed (line 1249) | def _actual_train_model_distributed(self, strategy, input_fn, hooks,
    method _train_with_estimator_spec_distributed (line 1362) | def _train_with_estimator_spec_distributed(self, estimator_spec, worke...
    method _train_with_estimator_spec (line 1396) | def _train_with_estimator_spec(self, estimator_spec, worker_hooks, hooks,
    method _evaluate_build_graph (line 1539) | def _evaluate_build_graph(self, input_fn, hooks=None, checkpoint_path=...
    method _call_model_fn_eval (line 1579) | def _call_model_fn_eval(self, input_fn, config):
    method _call_model_fn_eval_distributed (line 1592) | def _call_model_fn_eval_distributed(self, input_fn, config):
    method _evaluate_run (line 1655) | def _evaluate_run(self, checkpoint_path, scaffold, update_op, eval_dict,
    method _maybe_warm_start (line 1682) | def _maybe_warm_start(self, checkpoint_path):
    method export_savedmodel (line 1690) | def export_savedmodel(self,
  class EstimatorV2 (line 1766) | class EstimatorV2(Estimator):
    method _assert_members_are_not_overridden (line 1773) | def _assert_members_are_not_overridden(self):
  function _get_loss_reduce_op_for_reporting (line 1778) | def _get_loss_reduce_op_for_reporting():
  function _assert_members_are_not_overridden (line 1785) | def _assert_members_are_not_overridden(cls, obj):
  function _verify_and_create_loss_metric (line 1810) | def _verify_and_create_loss_metric(eval_metric_ops, loss, distribution=N...
  function maybe_overwrite_model_dir_and_session_config (line 1827) | def maybe_overwrite_model_dir_and_session_config(config, model_dir):
  function create_per_replica_ready_for_local_init_op (line 1871) | def create_per_replica_ready_for_local_init_op(scaffold):
  function _combine_distributed_scaffold (line 1885) | def _combine_distributed_scaffold(grouped_scaffold, distribution):
  function _check_checkpoint_available (line 1963) | def _check_checkpoint_available(model_dir):
  function _check_hooks_type (line 1970) | def _check_hooks_type(hooks):
  function _check_listeners_type (line 1979) | def _check_listeners_type(saving_listeners):
  function _get_replica_device_setter (line 1990) | def _get_replica_device_setter(config):
  function _verify_model_fn_args (line 2019) | def _verify_model_fn_args(model_fn, params):
  function _load_global_step_from_checkpoint_dir (line 2038) | def _load_global_step_from_checkpoint_dir(checkpoint_dir):
  function _extract_metric_update_ops (line 2047) | def _extract_metric_update_ops(eval_dict, distribution=None):
  function _dict_to_str (line 2061) | def _dict_to_str(dictionary):
  function _write_dict_to_summary (line 2075) | def _write_dict_to_summary(output_dir, dictionary, current_global_step):
  function _write_checkpoint_path_to_summary (line 2131) | def _write_checkpoint_path_to_summary(output_dir, checkpoint_path,
  function _has_dataset_or_queue_runner (line 2155) | def _has_dataset_or_queue_runner(maybe_tensor):
  class WarmStartSettings (line 2175) | class WarmStartSettings(
    method __new__ (line 2355) | def __new__(cls,
  function _get_default_warm_start_settings (line 2372) | def _get_default_warm_start_settings(warm_start_from):

FILE: tensorflow_estimator/python/estimator/estimator_export.py
  class estimator_export (line 63) | class estimator_export(tf_export.api_export):  # pylint: disable=invalid...
    method __init__ (line 66) | def __init__(self, *args: str, v1: Optional[Sequence[str]] = None):
    method __call__ (line 79) | def __call__(self, func: T) -> T:

FILE: tensorflow_estimator/python/estimator/estimator_export_test.py
  class TestClass (line 26) | class TestClass(object):
  class ValidateExportTest (line 30) | class ValidateExportTest(tf.test.TestCase):
    method setUp (line 33) | def setUp(self):
    method tearDown (line 37) | def tearDown(self):
    method testExportDeprecated (line 50) | def testExportDeprecated(self, mock_warning):

FILE: tensorflow_estimator/python/estimator/estimator_test.py
  function dummy_model_fn (line 64) | def dummy_model_fn(features, labels, params):
  function summaries_with_matching_keyword (line 68) | def summaries_with_matching_keyword(keyword, dir_):
  function check_eventfile_for_keyword (line 81) | def check_eventfile_for_keyword(keyword, dir_):
  function get_mock_saver (line 86) | def get_mock_saver():
  class EstimatorInheritanceConstraintTest (line 91) | class EstimatorInheritanceConstraintTest(tf.test.TestCase):
    method random_estimator (line 95) | def random_estimator(self):
    method test_override_a_method (line 99) | def test_override_a_method(self):
    method test_extension_of_api_is_ok (line 113) | def test_extension_of_api_is_ok(self):
    method test_override_allowed_method (line 125) | def test_override_allowed_method(self):
  class EstimatorConstructorTest (line 138) | class EstimatorConstructorTest(tf.test.TestCase):
    method test_config_must_be_a_run_config (line 140) | def test_config_must_be_a_run_config(self):
    method test_model_fn_must_be_provided (line 144) | def test_model_fn_must_be_provided(self):
    method test_property_accessors (line 148) | def test_property_accessors(self):
    method test_default_config (line 163) | def test_default_config(self):
    method test_default_model_dir (line 175) | def test_default_model_dir(self):
    method test_model_dir_in_constructor (line 185) | def test_model_dir_in_constructor(self):
    method test_empty_model_dir (line 194) | def test_empty_model_dir(self):
    method test_model_dir_in_run_config (line 204) | def test_model_dir_in_run_config(self):
    method test_same_model_dir_in_constructor_and_run_config (line 219) | def test_same_model_dir_in_constructor_and_run_config(self):
    method test_different_model_dir_in_constructor_and_run_config (line 235) | def test_different_model_dir_in_constructor_and_run_config(self):
    method test_model_fn_args_must_include_features (line 253) | def test_model_fn_args_must_include_features(self):
    method test_model_fn_args_labels_is_optional (line 261) | def test_model_fn_args_labels_is_optional(self):
    method test_if_params_provided_then_model_fn_should_accept_it (line 268) | def test_if_params_provided_then_model_fn_should_accept_it(self):
    method test_internal_params_is_a_deepcopy (line 277) | def test_internal_params_is_a_deepcopy(self):
    method test_not_known_model_fn_args (line 288) | def test_not_known_model_fn_args(self):
    method test_not_known_model_fn_args_handled_by_lambda (line 296) | def test_not_known_model_fn_args_handled_by_lambda(self):
    method test_if_model_fn_is_a_member_function_of_a_class (line 305) | def test_if_model_fn_is_a_member_function_of_a_class(self):
    method test_model_fn_property_binds_params (line 317) | def test_model_fn_property_binds_params(self):
    method test_model_fn_property_returns_fixed_signature (line 327) | def test_model_fn_property_returns_fixed_signature(self):
  function dummy_input_fn (line 338) | def dummy_input_fn():
  function model_fn_global_step_incrementer (line 342) | def model_fn_global_step_incrementer(features, labels, mode):
  function assert_features_op (line 349) | def assert_features_op(expected_features, actual_features):
  function _estimator_spec (line 357) | def _estimator_spec(expected_features, expected_labels, actual_features,
  function _make_input_fn (line 373) | def _make_input_fn(features, labels):
  class EstimatorTrainTest (line 382) | class EstimatorTrainTest(tf.test.TestCase):
    method test_callable_model_fn (line 384) | def test_callable_model_fn(self):
    method test_callable_input_fn (line 409) | def test_callable_input_fn(self):
    method test_nested_input_fn (line 436) | def test_nested_input_fn(self):
    method test_input_fn_args (line 459) | def test_input_fn_args(self):
    method test_minimal_model_fn_args (line 482) | def test_minimal_model_fn_args(self):
    method test_labels_should_be_none_if_model_fn_does_not_use_labels (line 506) | def test_labels_should_be_none_if_model_fn_does_not_use_labels(self):
    method test_input_fn_len_should_be_2_if_tuple_or_list (line 523) | def test_input_fn_len_should_be_2_if_tuple_or_list(self):
    method test_all_model_fn_args (line 535) | def test_all_model_fn_args(self):
    method test_partial_model_fn_args (line 563) | def test_partial_model_fn_args(self):
    method test_model_fn_must_return_estimator_spec (line 599) | def test_model_fn_must_return_estimator_spec(self):
    method test_run_train_op_and_saves_at_the_end (line 609) | def test_run_train_op_and_saves_at_the_end(self):
    method test_loss_summary (line 615) | def test_loss_summary(self):
    method test_latest_checkpoint (line 628) | def test_latest_checkpoint(self):
    method test_steps_and_saves_reloads (line 635) | def test_steps_and_saves_reloads(self):
    method test_warm_starts (line 644) | def test_warm_starts(self):
    method test_warm_starts_from_savedmodel (line 673) | def test_warm_starts_from_savedmodel(self):
    method test_max_step (line 715) | def test_max_step(self):
    method test_checkpoint_contains_relative_paths (line 724) | def test_checkpoint_contains_relative_paths(self):
    method test_train_save_copy_reload (line 740) | def test_train_save_copy_reload(self):
    method test_steps0_raises_error (line 761) | def test_steps0_raises_error(self):
    method test_steps_negative_raises_error (line 766) | def test_steps_negative_raises_error(self):
    method test_max_steps0_raises_error (line 771) | def test_max_steps0_raises_error(self):
    method test_max_steps_negative_raises_error (line 776) | def test_max_steps_negative_raises_error(self):
    method test_scaffold_is_used (line 781) | def test_scaffold_is_used(self):
    method test_hooks_should_be_session_run_hook (line 800) | def test_hooks_should_be_session_run_hook(self):
    method test_training_hooks_are_used (line 805) | def test_training_hooks_are_used(self):
    method test_saving_listeners_are_used (line 827) | def test_saving_listeners_are_used(self):
    method test_saver_hook_should_exist_to_use_saving_listeners (line 837) | def test_saver_hook_should_exist_to_use_saving_listeners(self):
    method test_listeners_should_be_listeners (line 847) | def test_listeners_should_be_listeners(self):
    method test_chief_only_hook_should_not_be_called_on_non_chief (line 853) | def test_chief_only_hook_should_not_be_called_on_non_chief(self):
    method test_features_labels_mode (line 896) | def test_features_labels_mode(self):
    method test_graph_initialization_global_step_and_random_seed (line 917) | def test_graph_initialization_global_step_and_random_seed(self):
    method test_config_should_not_be_evaluator_or_ps (line 933) | def test_config_should_not_be_evaluator_or_ps(self):
    method test_master_distributed_hooks (line 946) | def test_master_distributed_hooks(self):
    method test_master_distributed_hooks_for_worker_0 (line 977) | def test_master_distributed_hooks_for_worker_0(self):
    method test_master_distributed_hooks_for_worker_nonzero (line 1008) | def test_master_distributed_hooks_for_worker_nonzero(self):
    method test_master_hooks_single_replica (line 1039) | def test_master_hooks_single_replica(self):
    method test_master_hooks_single_replica_with_ps (line 1069) | def test_master_hooks_single_replica_with_ps(self):
    method test_hooks_with_distributed_collective_ops (line 1100) | def test_hooks_with_distributed_collective_ops(self):
  function _model_fn_with_eval_metric_ops (line 1155) | def _model_fn_with_eval_metric_ops(features, labels, mode, params):
  class _StepCounterHook (line 1181) | class _StepCounterHook(tf.train.SessionRunHook):
    method __init__ (line 1184) | def __init__(self):
    method before_run (line 1187) | def before_run(self, run_context):
    method steps (line 1192) | def steps(self):
  class EstimatorGetVariablesTest (line 1196) | class EstimatorGetVariablesTest(tf.test.TestCase):
    method test_model_should_be_trained (line 1198) | def test_model_should_be_trained(self):
    method test_get_variable_utils (line 1214) | def test_get_variable_utils(self):
  class EstimatorTraceTest (line 1233) | class EstimatorTraceTest(tf.test.TestCase, parameterized.TestCase):
    method setUp (line 1235) | def setUp(self):
    class ModelFn (line 1243) | class ModelFn(object):
      method __call__ (line 1245) | def __call__(self, features, labels):
  class EstimatorDatasetIntegrationTest (line 1251) | class EstimatorDatasetIntegrationTest(tf.test.TestCase):
    method test_returned_by_input_fn (line 1254) | def test_returned_by_input_fn(self):
    method test_with_none_labels (line 1270) | def test_with_none_labels(self):
    method test_with_predict (line 1287) | def test_with_predict(self):
    method test_batching (line 1304) | def test_batching(self):
  class EstimatorEvaluateTest (line 1325) | class EstimatorEvaluateTest(tf.test.TestCase):
    method test_eval_dir (line 1327) | def test_eval_dir(self):
    method test_input_fn_args (line 1335) | def test_input_fn_args(self):
    method test_model_fn_must_return_estimator_spec (line 1359) | def test_model_fn_must_return_estimator_spec(self):
    method test_no_checkpoint_uses_init (line 1376) | def test_no_checkpoint_uses_init(self):
    method test_no_checkpoint_uses_init_with_warm_starting (line 1398) | def test_no_checkpoint_uses_init_with_warm_starting(self):
    method test_scores (line 1459) | def test_scores(self):
    method test_tuple_metrics (line 1475) | def test_tuple_metrics(self):
    method test_steps0_raises_error (line 1498) | def test_steps0_raises_error(self):
    method test_steps_negative_raises_error (line 1504) | def test_steps_negative_raises_error(self):
    method test_global_step_metric_raises_error (line 1510) | def test_global_step_metric_raises_error(self):
    method test_global_step_is_reported (line 1522) | def test_global_step_is_reported(self):
    method test_loss_metric_is_reported (line 1536) | def test_loss_metric_is_reported(self):
    method test_hooks_should_be_session_run_hook (line 1556) | def test_hooks_should_be_session_run_hook(self):
    method test_hooks_are_used (line 1562) | def test_hooks_are_used(self):
    method test_evaluate_from_checkpoint (line 1570) | def test_evaluate_from_checkpoint(self):
    method test_wrong_shape_throws_reasonable_error (line 1587) | def test_wrong_shape_throws_reasonable_error(self):
    method test_scaffold_is_used (line 1617) | def test_scaffold_is_used(self):
    method test_features_labels_mode (line 1635) | def test_features_labels_mode(self):
    method test_graph_initialization_global_step_and_random_seed (line 1657) | def test_graph_initialization_global_step_and_random_seed(self):
    method test_evaluation_hooks_are_used (line 1674) | def test_evaluation_hooks_are_used(self):
    method test_summary_writing_with_summary_proto (line 1692) | def test_summary_writing_with_summary_proto(self):
    method test_summary_writing_with_tensor (line 1740) | def test_summary_writing_with_tensor(self):
  class EstimatorPredictTest (line 1783) | class EstimatorPredictTest(tf.test.TestCase):
    method test_input_fn_args (line 1785) | def test_input_fn_args(self):
    method test_no_checkpoint_uses_init (line 1813) | def test_no_checkpoint_uses_init(self):
    method test_no_checkpoint_uses_init_with_warm_starting (line 1826) | def test_no_checkpoint_uses_init_with_warm_starting(self):
    method test_no_trained_model_invalid_checkpoint_path (line 1876) | def test_no_trained_model_invalid_checkpoint_path(self):
    method test_tensor_predictions (line 1884) | def test_tensor_predictions(self):
    method test_predictionhooks_are_used (line 1898) | def test_predictionhooks_are_used(self):
    method test_warn_if_no_queue_runner (line 1917) | def test_warn_if_no_queue_runner(self):
    method test_skip_warn_if_dataset_returns_features (line 1935) | def test_skip_warn_if_dataset_returns_features(self):
    method test_skip_warn_if_dataset_returns_features_dict (line 1957) | def test_skip_warn_if_dataset_returns_features_dict(self):
    method test_input_fn_can_return_just_features (line 1980) | def test_input_fn_can_return_just_features(self):
    method test_batch_size_mismatch (line 1998) | def test_batch_size_mismatch(self):
    method test_iterate_batches (line 2017) | def test_iterate_batches(self):
    method test_predict_keys_defined_for_tensor (line 2038) | def test_predict_keys_defined_for_tensor(self):
    method test_predict_keys_does_not_exists (line 2055) | def test_predict_keys_does_not_exists(self):
    method test_return_given_predict_keys (line 2074) | def test_return_given_predict_keys(self):
    method test_yield_rows_of_tensor (line 2093) | def test_yield_rows_of_tensor(self):
    method test_yield_rows_of_dict (line 2109) | def test_yield_rows_of_dict(self):
    method test_hooks_should_be_session_run_hook (line 2128) | def test_hooks_should_be_session_run_hook(self):
    method test_hooks_are_used (line 2134) | def test_hooks_are_used(self):
    method test_predict_from_old_model_dir (line 2156) | def test_predict_from_old_model_dir(self):
    method test_predict_from_checkpoint_path (line 2173) | def test_predict_from_checkpoint_path(self):
    method test_scaffold_is_used (line 2194) | def test_scaffold_is_used(self):
    method test_features_labels_mode (line 2212) | def test_features_labels_mode(self):
    method test_graph_initialization_global_step_and_random_seed (line 2234) | def test_graph_initialization_global_step_and_random_seed(self):
  function _model_fn_for_export_tests (line 2252) | def _model_fn_for_export_tests(features, labels, mode):
  function _x_y_input_fn (line 2268) | def _x_y_input_fn():
  function _model_fn_with_x_y (line 2275) | def _model_fn_with_x_y(features, labels, mode):
  function _model_fn_with_saveables_for_export_tests (line 2312) | def _model_fn_with_saveables_for_export_tests(features, labels, mode):
  function _get_serving_input_receiver_fn (line 2329) | def _get_serving_input_receiver_fn():
  function _get_supervised_input_receiver_fn (line 2337) | def _get_supervised_input_receiver_fn():
  class EstimatorExportTest (line 2347) | class EstimatorExportTest(tf.test.TestCase):
    method test_export_saved_model_proto_roundtrip_raw_receiver (line 2349) | def test_export_saved_model_proto_roundtrip_raw_receiver(self):
    method test_export_saved_model_train (line 2374) | def test_export_saved_model_train(self):
    method test_export_saved_model_eval (line 2378) | def test_export_saved_model_eval(self):
    method test_export_saved_model_predict (line 2382) | def test_export_saved_model_predict(self):
    method _test_export_saved_model_for_mode (line 2386) | def _test_export_saved_model_for_mode(self, input_receiver_fn, mode):
    method test_export_all_saved_models_proto_roundtrip_receiver_map (line 2413) | def test_export_all_saved_models_proto_roundtrip_receiver_map(self):
    method test_export_all_saved_models_proto_roundtrip_train_only (line 2430) | def test_export_all_saved_models_proto_roundtrip_train_only(self):
    method test_export_all_saved_models_proto_roundtrip_eval_only (line 2450) | def test_export_all_saved_models_proto_roundtrip_eval_only(self):
    method test_export_all_saved_models_proto_roundtrip_no_serving (line 2468) | def test_export_all_saved_models_proto_roundtrip_no_serving(self):
    method test_export_all_saved_models_proto_roundtrip_three_defs (line 2498) | def test_export_all_saved_models_proto_roundtrip_three_defs(self):
    method test_export_all_saved_models_proto_roundtrip_all_vars (line 2520) | def test_export_all_saved_models_proto_roundtrip_all_vars(self):
    method test_export_all_saved_models_name_collision (line 2545) | def test_export_all_saved_models_name_collision(self):
    method _test_export_all_saved_models (line 2581) | def _test_export_all_saved_models(self, input_receiver_fn_map):
    method _validate_exported_files (line 2599) | def _validate_exported_files(self, export_dir):
    method test_export_all_saved_models_var_not_found (line 2622) | def test_export_all_saved_models_var_not_found(self):
    method test_export_all_saved_models_metric_operation (line 2659) | def test_export_all_saved_models_metric_operation(self):
    method test_export_saved_model_with_saveables_proto_roundtrip (line 2701) | def test_export_saved_model_with_saveables_proto_roundtrip(self):
    method test_export_saved_model_assets (line 2756) | def test_export_saved_model_assets(self):
    method test_export_saved_model_extra_assets (line 2821) | def test_export_saved_model_extra_assets(self):
    method test_export_saved_model_tensor_features (line 2863) | def test_export_saved_model_tensor_features(self):
    method test_export_saved_model_int_feature_keys (line 2917) | def test_export_saved_model_int_feature_keys(self):
    method test_scaffold_is_used_for_saver (line 2977) | def test_scaffold_is_used_for_saver(self):
    method test_scaffold_is_used_for_saver_multiple_modes (line 3011) | def test_scaffold_is_used_for_saver_multiple_modes(self):
    method test_scaffold_is_used_for_local_init (line 3058) | def test_scaffold_is_used_for_local_init(self):
    method test_scaffold_is_used_for_local_init_multiple_modes (line 3106) | def test_scaffold_is_used_for_local_init_multiple_modes(self):
    method test_features_labels_mode (line 3161) | def test_features_labels_mode(self):
    method test_graph_initialization_global_step_and_random_seed (line 3186) | def test_graph_initialization_global_step_and_random_seed(self):
    method test_export_saved_model_respects_soft_placement (line 3211) | def test_export_saved_model_respects_soft_placement(self):
    method _validate_strip_default_attrs (line 3284) | def _validate_strip_default_attrs(self, estimator_cls, export_fn,
    method test_export_saved_model_proto_strip_default_attrs (line 3332) | def test_export_saved_model_proto_strip_default_attrs(self):
    method test_export_saved_model_no_export_outputs (line 3350) | def test_export_saved_model_no_export_outputs(self):
    method test_export_from_warm_start (line 3390) | def test_export_from_warm_start(self):
    method test_export_saved_model_symbol_deprecated (line 3416) | def test_export_saved_model_symbol_deprecated(self):
  class EstimatorHookOrderingTest (line 3423) | class EstimatorHookOrderingTest(tf.test.TestCase):
    method testCustomHooksAreCalledBeforeNanTensorHook (line 3425) | def testCustomHooksAreCalledBeforeNanTensorHook(self):
  class EstimatorIntegrationTest (line 3462) | class EstimatorIntegrationTest(tf.test.TestCase):
    method test_complete_flow_with_a_simple_linear_model (line 3464) | def test_complete_flow_with_a_simple_linear_model(self):
  class EstimatorInputContextTest (line 3525) | class EstimatorInputContextTest(tf.test.TestCase):
    method test_with_input_fn (line 3527) | def test_with_input_fn(self):

FILE: tensorflow_estimator/python/estimator/export/export.py
  function wrap_and_check_input_tensors (line 47) | def wrap_and_check_input_tensors(tensors, field_name, allow_int_keys=Fal...
  function _check_tensor (line 75) | def _check_tensor(tensor, name, error_label='feature'):
  function _check_tensor_key (line 99) | def _check_tensor_key(name, error_label='feature', allow_ints=False):
  class ServingInputReceiver (line 109) | class ServingInputReceiver(
    method __new__ (line 136) | def __new__(cls,
  class TensorServingInputReceiver (line 166) | class TensorServingInputReceiver(
    method __new__ (line 203) | def __new__(cls,
  class UnsupervisedInputReceiver (line 223) | class UnsupervisedInputReceiver(ServingInputReceiver):
    method __new__ (line 238) | def __new__(cls, features, receiver_tensors):
  class SupervisedInputReceiver (line 246) | class SupervisedInputReceiver(
    method __new__ (line 268) | def __new__(cls, features, labels, receiver_tensors):
  function build_parsing_serving_input_receiver_fn (line 286) | def build_parsing_serving_input_receiver_fn(feature_spec,
  function _placeholder_from_tensor (line 317) | def _placeholder_from_tensor(t, default_batch_size=None):
  function _placeholders_from_receiver_tensors_dict (line 347) | def _placeholders_from_receiver_tensors_dict(input_vals,
  function build_raw_serving_input_receiver_fn (line 356) | def build_raw_serving_input_receiver_fn(features, default_batch_size=None):
  function build_raw_supervised_input_receiver_fn (line 382) | def build_raw_supervised_input_receiver_fn(features,
  function build_supervised_input_receiver_fn_from_input_fn (line 444) | def build_supervised_input_receiver_fn_from_input_fn(input_fn, **input_f...

FILE: tensorflow_estimator/python/estimator/export/export_test.py
  class LabeledTensorMock (line 32) | class LabeledTensorMock(object):
    method __init__ (line 35) | def __init__(self):
  function _convert_labeled_tensor_mock_to_tensor (line 39) | def _convert_labeled_tensor_mock_to_tensor(value, *args, **kwargs):
  class ServingInputReceiverTest (line 47) | class ServingInputReceiverTest(tf.test.TestCase):
    method test_serving_input_receiver_constructor (line 49) | def test_serving_input_receiver_constructor(self):
    method test_serving_input_receiver_features_invalid (line 65) | def test_serving_input_receiver_features_invalid(self):
    method test_serving_input_receiver_receiver_tensors_invalid (line 86) | def test_serving_input_receiver_receiver_tensors_invalid(self):
    method test_single_feature_single_receiver (line 109) | def test_single_feature_single_receiver(self):
    method test_multi_feature_single_receiver (line 120) | def test_multi_feature_single_receiver(self):
    method test_multi_feature_multi_receiver (line 125) | def test_multi_feature_multi_receiver(self):
    method test_feature_wrong_type (line 130) | def test_feature_wrong_type(self):
    method test_feature_labeled_tensor (line 136) | def test_feature_labeled_tensor(self):
    method test_receiver_wrong_type (line 141) | def test_receiver_wrong_type(self):
  class UnsupervisedInputReceiverTest (line 148) | class UnsupervisedInputReceiverTest(tf.test.TestCase):
    method test_unsupervised_input_receiver_constructor (line 153) | def test_unsupervised_input_receiver_constructor(self):
  class SupervisedInputReceiverTest (line 173) | class SupervisedInputReceiverTest(tf.test.TestCase):
    method test_input_receiver_constructor (line 175) | def test_input_receiver_constructor(self):
    method test_input_receiver_raw_values (line 200) | def test_input_receiver_raw_values(self):
    method test_input_receiver_features_invalid (line 232) | def test_input_receiver_features_invalid(self):
    method test_input_receiver_receiver_tensors_invalid (line 278) | def test_input_receiver_receiver_tensors_invalid(self):
    method test_single_feature_single_receiver (line 307) | def test_single_feature_single_receiver(self):
    method test_multi_feature_single_receiver (line 318) | def test_multi_feature_single_receiver(self):
    method test_multi_feature_multi_receiver (line 324) | def test_multi_feature_multi_receiver(self):
    method test_feature_labeled_tensor (line 330) | def test_feature_labeled_tensor(self):
  class ExportTest (line 337) | class ExportTest(tf.test.TestCase):
    method test_build_parsing_serving_input_receiver_fn (line 341) | def test_build_parsing_serving_input_receiver_fn(self):
    method test_build_raw_serving_input_receiver_fn_name (line 394) | def test_build_raw_serving_input_receiver_fn_name(self):
    method test_build_raw_serving_input_receiver_fn_without_shape (line 407) | def test_build_raw_serving_input_receiver_fn_without_shape(self):
    method test_build_raw_serving_input_receiver_fn (line 421) | def test_build_raw_serving_input_receiver_fn(self):
    method test_build_raw_supervised_input_receiver_fn (line 443) | def test_build_raw_supervised_input_receiver_fn(self):
    method test_build_raw_supervised_input_receiver_fn_raw_tensors (line 464) | def test_build_raw_supervised_input_receiver_fn_raw_tensors(self):
    method test_build_raw_supervised_input_receiver_fn_batch_size (line 488) | def test_build_raw_supervised_input_receiver_fn_batch_size(self):
    method test_build_raw_supervised_input_receiver_fn_overlapping_keys (line 501) | def test_build_raw_supervised_input_receiver_fn_overlapping_keys(self):
    method test_build_supervised_input_receiver_fn_from_input_fn (line 510) | def test_build_supervised_input_receiver_fn_from_input_fn(self):
    method test_build_supervised_input_receiver_fn_from_input_fn_args (line 528) | def test_build_supervised_input_receiver_fn_from_input_fn_args(self):
  class TensorServingReceiverTest (line 550) | class TensorServingReceiverTest(tf.test.TestCase):
    method test_tensor_serving_input_receiver_constructor (line 552) | def test_tensor_serving_input_receiver_constructor(self):
    method test_tensor_serving_input_receiver_sparse (line 562) | def test_tensor_serving_input_receiver_sparse(self):
    method test_serving_input_receiver_features_invalid (line 573) | def test_serving_input_receiver_features_invalid(self):
    method test_serving_input_receiver_receiver_tensors_invalid (line 587) | def test_serving_input_receiver_receiver_tensors_invalid(self):

FILE: tensorflow_estimator/python/estimator/export/function.py
  class ModelFunction (line 33) | class ModelFunction(tf.compat.v2.__internal__.tracking.AutoTrackable):
    method __init__ (line 39) | def __init__(self, config=None, params=None):
    method from_function (line 51) | def from_function(model_fn, all_modes=None, config=None, params=None):
    method variables (line 64) | def variables(self):
    method add_mode (line 67) | def add_mode(self, fn, mode, input_signature=None):
    method train (line 82) | def train(self, features, labels):
    method evaluate (line 85) | def evaluate(self, features, labels):
    method predict (line 88) | def predict(self, features):
    method call (line 91) | def call(self, mode, features, labels=None):
  function _wrap_and_verify_model_fn (line 104) | def _wrap_and_verify_model_fn(model_fn,
  class EstimatorSpecFunction (line 151) | class EstimatorSpecFunction(tf.compat.v2.__internal__.function.Function):
    method __init__ (line 158) | def __init__(self,
    method _defun (line 185) | def _defun(self, fn):
  class _EstimatorSpecFunction (line 195) | class _EstimatorSpecFunction(tf.compat.v2.__internal__.function.Function):
    method __init__ (line 201) | def __init__(self, python_function, name, variable_holder=None, **kwar...
    method _create_graph_function (line 206) | def _create_graph_function(self, args, kwargs, **other_kwargs):
  class _EstimatorWrappedGraph (line 217) | class _EstimatorWrappedGraph(wrap_function.WrappedGraph):
    method __init__ (line 220) | def __init__(self, *args, **kwargs):
    method _global_step (line 232) | def _global_step(self):
    method global_step (line 236) | def global_step(self):
    method model_fn (line 240) | def model_fn(self):
    method estimator_spec (line 244) | def estimator_spec(self):
    method wrap_model_fn (line 249) | def wrap_model_fn(self,
    method wrap_input_receiver_fn (line 273) | def wrap_input_receiver_fn(self, input_receiver_fn):
  function _filter_estimator_spec_outputs (line 361) | def _filter_estimator_spec_outputs(spec):
  function _canonicalize_receiver_tensors (line 372) | def _canonicalize_receiver_tensors(receiver_tensors):
  function _input_receiver_fn_name (line 385) | def _input_receiver_fn_name(name):
  function _prune_receiver_tensors (line 392) | def _prune_receiver_tensors(wrapped_function, receiver_tensors, outputs,...

FILE: tensorflow_estimator/python/estimator/export/function_test.py
  function _string_fix (line 32) | def _string_fix(obj):
  function _model_fn (line 38) | def _model_fn(features, labels, mode):
  function _model_fn_train_only (line 55) | def _model_fn_train_only(features, labels):
  function _model_fn_predict_only (line 64) | def _model_fn_predict_only(features):
  class ModelFunctionTest (line 71) | class ModelFunctionTest(object):
    method test_from_function (line 73) | def test_from_function(self):
    method test_model_fn_train_only (line 85) | def test_model_fn_train_only(self):
    method test_model_fn_predict_only (line 97) | def test_model_fn_predict_only(self):
    method test_save_and_load (line 109) | def test_save_and_load(self):
  function _model_fn_callable_variable_initializers (line 147) | def _model_fn_callable_variable_initializers(features, labels, mode):
  class EstimatorWrappedGraphTest (line 167) | class EstimatorWrappedGraphTest(tf.test.TestCase):
    method test_wrap_model_fn_train (line 169) | def test_wrap_model_fn_train(self):
    method test_wrap_model_fn_eval (line 186) | def test_wrap_model_fn_eval(self):
    method test_wrap_model_fn_predict (line 199) | def test_wrap_model_fn_predict(self):
    method test_wrap_input_receiver_fn (line 210) | def test_wrap_input_receiver_fn(self):

FILE: tensorflow_estimator/python/estimator/exporter.py
  class Exporter (line 31) | class Exporter(object):
    method name (line 35) | def name(self):
    method export (line 44) | def export(self, estimator, export_path, checkpoint_path, eval_result,
  class _SavedModelExporter (line 65) | class _SavedModelExporter(Exporter):
    method __init__ (line 72) | def __init__(self,
    method name (line 103) | def name(self):
    method export (line 106) | def export(self, estimator, export_path, checkpoint_path, eval_result,
  function _loss_smaller (line 120) | def _loss_smaller(best_eval_result, current_eval_result):
  function _verify_compare_fn_args (line 148) | def _verify_compare_fn_args(compare_fn):
  class BestExporter (line 165) | class BestExporter(Exporter):
    method __init__ (line 172) | def __init__(self,
    method name (line 276) | def name(self):
    method export (line 279) | def export(self, estimator, export_path, checkpoint_path, eval_result,
    method _garbage_collect_exports (line 309) | def _garbage_collect_exports(self, export_dir_base):
    method _get_best_eval_result (line 341) | def _get_best_eval_result(self, event_files):
  class FinalExporter (line 369) | class FinalExporter(Exporter):
    method __init__ (line 375) | def __init__(self,
    method name (line 405) | def name(self):
    method export (line 408) | def export(self, estimator, export_path, checkpoint_path, eval_result,
  class LatestExporter (line 422) | class LatestExporter(Exporter):
    method __init__ (line 428) | def __init__(self,
    method name (line 466) | def name(self):
    method export (line 469) | def export(self, estimator, export_path, checkpoint_path, eval_result,
    method _garbage_collect_exports (line 479) | def _garbage_collect_exports(self, export_dir_base):

FILE: tensorflow_estimator/python/estimator/exporter_test.py
  class BestExporterTest (line 31) | class BestExporterTest(tf.test.TestCase):
    method test_error_out_if_exports_to_keep_is_zero (line 33) | def test_error_out_if_exports_to_keep_is_zero(self):
    method test_best_exporter (line 45) | def test_best_exporter(self):
    method test_best_export_is_saved (line 76) | def test_best_export_is_saved(self):
    method test_best_exporter_with_preemption (line 110) | def test_best_exporter_with_preemption(self):
    method test_best_exporter_with_empty_event (line 151) | def test_best_exporter_with_empty_event(self):
    method test_the_first_export (line 185) | def test_the_first_export(self):
    method test_garbage_collect_exports (line 220) | def test_garbage_collect_exports(self):
  class LatestExporterTest (line 255) | class LatestExporterTest(tf.test.TestCase):
    method test_error_out_if_exports_to_keep_is_zero (line 257) | def test_error_out_if_exports_to_keep_is_zero(self):
    method test_latest_exporter (line 269) | def test_latest_exporter(self):
    method test_only_the_last_export_is_saved (line 297) | def test_only_the_last_export_is_saved(self):
    method test_garbage_collect_exports (line 330) | def test_garbage_collect_exports(self):
    method test_garbage_collect_exports_with_trailing_delimiter (line 360) | def test_garbage_collect_exports_with_trailing_delimiter(self):
  function _create_test_export_dir (line 399) | def _create_test_export_dir(export_dir_base):
  function _get_timestamped_export_dir (line 406) | def _get_timestamped_export_dir(export_dir_base):

FILE: tensorflow_estimator/python/estimator/extenders.py
  function add_metrics (line 30) | def add_metrics(estimator, metric_fn):
  function _verify_metric_fn_args (line 103) | def _verify_metric_fn_args(metric_fn):
  function _call_metric_fn (line 111) | def _call_metric_fn(metric_fn, features, labels, predictions, config):

FILE: tensorflow_estimator/python/estimator/extenders_test.py
  function get_input_fn (line 29) | def get_input_fn(x, y):
  class AddMetricsTest (line 41) | class AddMetricsTest(tf.test.TestCase):
    method test_should_add_metrics (line 43) | def test_should_add_metrics(self):
    method test_should_error_out_for_not_recognized_args (line 68) | def test_should_error_out_for_not_recognized_args(self):
    method test_all_supported_args (line 79) | def test_all_supported_args(self):
    method test_all_supported_args_in_different_order (line 96) | def test_all_supported_args_in_different_order(self):
    method test_all_args_are_optional (line 113) | def test_all_args_are_optional(self):
    method test_overrides_existing_metrics (line 132) | def test_overrides_existing_metrics(self):

FILE: tensorflow_estimator/python/estimator/gc.py
  function _largest_export_versions (line 77) | def _largest_export_versions(n):
  function _one_of_every_n_export_versions (line 98) | def _one_of_every_n_export_versions(n):
  function _mod_export_version (line 130) | def _mod_export_version(n):
  function _union (line 150) | def _union(lf, rf):
  function _negation (line 169) | def _negation(f):
  function _get_paths (line 187) | def _get_paths(base_dir, parser):

FILE: tensorflow_estimator/python/estimator/gc_test.py
  function _create_parser (line 30) | def _create_parser(base_dir):
  class GcTest (line 48) | class GcTest(tf.test.TestCase):
    method testLargestExportVersions (line 50) | def testLargestExportVersions(self):
    method testLargestExportVersionsDoesNotDeleteZeroFolder (line 56) | def testLargestExportVersionsDoesNotDeleteZeroFolder(self):
    method testModExportVersion (line 62) | def testModExportVersion(self):
    method testOneOfEveryNExportVersions (line 74) | def testOneOfEveryNExportVersions(self):
    method testOneOfEveryNExportVersionsZero (line 94) | def testOneOfEveryNExportVersionsZero(self):
    method testUnion (line 101) | def testUnion(self):
    method testNegation (line 116) | def testNegation(self):
    method testPathsWithParse (line 128) | def testPathsWithParse(self):
    method testMixedStrTypes (line 144) | def testMixedStrTypes(self):
    method testGcsDirWithSeparator (line 157) | def testGcsDirWithSeparator(self):

FILE: tensorflow_estimator/python/estimator/head/base_head.py
  class Head (line 46) | class Head(object):
    method name (line 100) | def name(self):
    method logits_dimension (line 109) | def logits_dimension(self):
    method loss_reduction (line 121) | def loss_reduction(self):
    method loss (line 132) | def loss(self,
    method predictions (line 162) | def predictions(self, logits, keys=None):
    method metrics (line 177) | def metrics(self, regularization_losses=None):
    method update_metrics (line 191) | def update_metrics(self,
    method _summary_key (line 222) | def _summary_key(self, key):
    method create_estimator_spec (line 225) | def create_estimator_spec(self,
    method _create_tpu_estimator_spec (line 295) | def _create_tpu_estimator_spec(
  function check_dense_labels_match_logits_and_reshape (line 380) | def check_dense_labels_match_logits_and_reshape(labels, logits,
  function get_weights_and_check_match_logits (line 468) | def get_weights_and_check_match_logits(features,
  function check_logits_final_dim (line 581) | def check_logits_final_dim(logits, expected_logits_dimension):
  function validate_loss_fn_args (line 623) | def validate_loss_fn_args(loss_fn):
  function validate_loss_reduction (line 647) | def validate_loss_reduction(loss_reduction):
  function validate_update_ops (line 655) | def validate_update_ops(update_ops=None):
  function validate_v2_optimizer (line 661) | def validate_v2_optimizer(optimizer):
  function validate_trainable_variables (line 670) | def validate_trainable_variables(trainable_variables=None):
  function validate_n_classes (line 680) | def validate_n_classes(n_classes):
  function call_loss_fn (line 706) | def call_loss_fn(loss_fn, labels, logits, features, expected_loss_dim=1):
  function check_prediction_keys (line 760) | def check_prediction_keys(pred_keys, valid_keys):
  function all_class_ids (line 768) | def all_class_ids(logits, n_classes):
  function all_classes (line 776) | def all_classes(logits, n_classes, label_vocabulary=None):
  function classification_output (line 786) | def classification_output(scores, n_classes, label_vocabulary=None):
  function check_label_range (line 793) | def check_label_range(labels, n_classes, message=None):
  function update_metric_with_broadcast_weights (line 817) | def update_metric_with_broadcast_weights(eval_metric, values, weights):
  function create_eval_metrics_tuple (line 824) | def create_eval_metrics_tuple(fn, kwargs):
  function create_estimator_spec_train_op (line 855) | def create_estimator_spec_train_op(
  function create_estimator_spec_summary (line 919) | def create_estimator_spec_summary(regularized_training_loss,

FILE: tensorflow_estimator/python/estimator/head/base_head_test.py
  function _assert_simple_summaries (line 34) | def _assert_simple_summaries(test_case,
  function _assert_no_hooks (line 54) | def _assert_no_hooks(test_case, spec):
  class CreateEstimatorSpecTest (line 60) | class CreateEstimatorSpecTest(tf.test.TestCase):
    class _HeadWithTPUSupport (line 62) | class _HeadWithTPUSupport(base_head.Head):
      method name (line 65) | def name(self):
      method logits_dimension (line 68) | def logits_dimension(self):
      method loss_reduction (line 71) | def loss_reduction(self):
      method loss (line 74) | def loss(self, features, mode, logits, labels):
      method predictions (line 77) | def predictions(self, logits):
      method metrics (line 80) | def metrics(self, regularization_losses=None):
      method update_metrics (line 83) | def update_metrics(self,
      method _create_tpu_estimator_spec (line 92) | def _create_tpu_estimator_spec(
    class _HeadWithOutTPUSupport (line 107) | class _HeadWithOutTPUSupport(base_head.Head):
      method name (line 110) | def name(self):
      method logits_dimension (line 113) | def logits_dimension(self):
      method loss_reduction (line 116) | def loss_reduction(self):
      method loss (line 119) | def loss(self, features, mode, logits, labels):
      method predictions (line 122) | def predictions(self, logits):
      method metrics (line 125) | def metrics(self, regularization_losses=None):
      method update_metrics (line 128) | def update_metrics(self,
      method create_estimator_spec (line 137) | def create_estimator_spec(
    class _InvalidHead (line 152) | class _InvalidHead(base_head.Head):
      method name (line 155) | def name(self):
      method logits_dimension (line 158) | def logits_dimension(self):
      method loss_reduction (line 161) | def loss_reduction(self):
      method loss (line 164) | def loss(self, features, mode, logits, labels):
      method predictions (line 167) | def predictions(self, logits):
      method metrics (line 170) | def metrics(self, regularization_losses=None):
      method update_metrics (line 173) | def update_metrics(self,
    method test_head_override_tpu_estimator_spec (line 182) | def test_head_override_tpu_estimator_spec(self):
    method test_head_override_estimator_spec (line 192) | def test_head_override_estimator_spec(self):
    method test_invalid_head_class (line 203) | def test_invalid_head_class(self):
    method test_tensor_shape_checking_in_graph_mode (line 217) | def test_tensor_shape_checking_in_graph_mode(self):
    method test_optimizer_v2_variable_name (line 249) | def test_optimizer_v2_variable_name(self):
    method test_head_with_invalid_optimizer (line 292) | def test_head_with_invalid_optimizer(self):

FILE: tensorflow_estimator/python/estimator/head/binary_class_head.py
  class BinaryClassHead (line 36) | class BinaryClassHead(base_head.Head):
    method __init__ (line 143) | def __init__(self,
    method name (line 195) | def name(self):
    method logits_dimension (line 200) | def logits_dimension(self):
    method loss_reduction (line 205) | def loss_reduction(self):
    method _class_id_table (line 218) | def _class_id_table(self):
    method _class_string_table (line 234) | def _class_string_table(self):
    method _processed_labels (line 251) | def _processed_labels(self, logits, labels):
    method _unweighted_loss_and_weights (line 260) | def _unweighted_loss_and_weights(self, logits, labels, features):
    method loss (line 276) | def loss(self,
    method predictions (line 301) | def predictions(self, logits, keys=None):
    method metrics (line 363) | def metrics(self, regularization_losses=None):
    method _update_accuracy_baseline (line 399) | def _update_accuracy_baseline(self, eval_metrics):
    method _update_auc (line 434) | def _update_auc(self, auc_metric, labels, predictions, weights=None):
    method update_metrics (line 441) | def update_metrics(self,
    method _create_tpu_estimator_spec (line 494) | def _create_tpu_estimator_spec(self,

FILE: tensorflow_estimator/python/estimator/head/binary_class_head_test.py
  class BinaryClassHeadTest (line 36) | class BinaryClassHeadTest(tf.test.TestCase):
    method test_threshold_too_small (line 38) | def test_threshold_too_small(self):
    method test_threshold_too_large (line 42) | def test_threshold_too_large(self):
    method test_invalid_loss_reduction (line 46) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 53) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 63) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 73) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 79) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits_shape (line 88) | def test_invalid_logits_shape(self):
    method test_invalid_labels_shape (line 114) | def test_invalid_labels_shape(self):
    method test_incompatible_labels_shape (line 151) | def test_incompatible_labels_shape(self):
    method test_predict (line 216) | def test_predict(self):
    method test_predict_with_vocabulary_list (line 295) | def test_predict_with_vocabulary_list(self):
    method test_eval_create_loss (line 311) | def test_eval_create_loss(self):
    method test_eval_create_loss_loss_fn (line 329) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 352) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_eval_labels_none (line 381) | def test_eval_labels_none(self):
    method test_eval (line 393) | def test_eval(self):
    method test_eval_metric_ops_with_head_name (line 453) | def test_eval_metric_ops_with_head_name(self):
    method test_eval_with_regularization_losses (line 475) | def test_eval_with_regularization_losses(self):
    method test_eval_with_vocabulary_list_create_loss (line 535) | def test_eval_with_vocabulary_list_create_loss(self):
    method test_eval_with_vocabulary_list (line 556) | def test_eval_with_vocabulary_list(self):
    method test_eval_with_thresholds_create_loss (line 586) | def test_eval_with_thresholds_create_loss(self):
    method test_eval_with_thresholds (line 608) | def test_eval_with_thresholds(self):
    method test_train_create_loss (line 683) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 704) | def test_train_create_loss_loss_reduction(self):
    method test_train_labels_none (line 726) | def test_train_labels_none(self):
    method test_train (line 737) | def test_train(self):
    method test_train_one_dim_create_loss (line 789) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 814) | def test_train_one_dim(self):
    method test_train_with_regularization_losses (line 873) | def test_train_with_regularization_losses(self):
    method test_float_labels_invalid_values (line 929) | def test_float_labels_invalid_values(self):
    method test_float_labels_train_create_loss (line 949) | def test_float_labels_train_create_loss(self):
    method test_float_labels_train (line 968) | def test_float_labels_train(self):
    method test_float_labels_eval_create_loss (line 1011) | def test_float_labels_eval_create_loss(self):
    method test_float_labels_eval (line 1034) | def test_float_labels_eval(self):
    method test_weighted_multi_example_predict (line 1080) | def test_weighted_multi_example_predict(self):
    method test_weighted_multi_example_eval (line 1103) | def test_weighted_multi_example_eval(self):
    method test_weighted_multi_example_train (line 1171) | def test_weighted_multi_example_train(self):
    method test_multi_dim_weighted_train_create_loss (line 1222) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 1248) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 1292) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 1326) | def test_multi_dim_train_weights_wrong_outer_dim(self):
    method test_multi_dim_weighted_eval (line 1360) | def test_multi_dim_weighted_eval(self):
  class BinaryClassHeadForEstimator (line 1430) | class BinaryClassHeadForEstimator(tf.test.TestCase):
    method test_invalid_trainable_variables (line 1433) | def test_invalid_trainable_variables(self):
    method test_train_with_optimizer (line 1472) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 1512) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 1543) | def test_train_summaries_with_head_name(self):
    method test_lookup_tables_in_graph (line 1575) | def test_lookup_tables_in_graph(self):

FILE: tensorflow_estimator/python/estimator/head/head_utils.py
  function binary_or_multi_class_head (line 28) | def binary_or_multi_class_head(n_classes, weight_column, label_vocabulary,
  function _initialize_variables (line 67) | def _initialize_variables(test_case, scaffold):
  function _assert_simple_summaries (line 78) | def _assert_simple_summaries(test_case,
  function _assert_no_hooks (line 98) | def _assert_no_hooks(test_case, spec):

FILE: tensorflow_estimator/python/estimator/head/multi_class_head.py
  class MultiClassHead (line 36) | class MultiClassHead(base_head.Head):
    method __init__ (line 136) | def __init__(self,
    method name (line 171) | def name(self):
    method logits_dimension (line 176) | def logits_dimension(self):
    method loss_reduction (line 181) | def loss_reduction(self):
    method _class_id_table (line 194) | def _class_id_table(self):
    method _class_string_table (line 210) | def _class_string_table(self):
    method _processed_labels (line 227) | def _processed_labels(self, logits, labels):
    method _unweighted_loss_and_weights (line 244) | def _unweighted_loss_and_weights(self, logits, label_ids, features):
    method loss (line 264) | def loss(self,
    method predictions (line 289) | def predictions(self, logits, keys=None):
    method metrics (line 347) | def metrics(self, regularization_losses=None):
    method update_metrics (line 363) | def update_metrics(self,
    method _create_tpu_estimator_spec (line 389) | def _create_tpu_estimator_spec(self,

FILE: tensorflow_estimator/python/estimator/head/multi_class_head_test.py
  class MultiClassHead (line 35) | class MultiClassHead(tf.test.TestCase):
    method test_n_classes_is_none (line 37) | def test_n_classes_is_none(self):
    method test_n_classes_is_2 (line 41) | def test_n_classes_is_2(self):
    method test_invalid_loss_reduction (line 45) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 54) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 64) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 74) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 81) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits_shape (line 90) | def test_invalid_logits_shape(self):
    method test_invalid_labels_shape (line 122) | def test_invalid_labels_shape(self):
    method test_invalid_labels_type (line 167) | def test_invalid_labels_type(self):
    method test_invalid_labels_values (line 204) | def test_invalid_labels_values(self):
    method test_invalid_labels_sparse_tensor (line 261) | def test_invalid_labels_sparse_tensor(self):
    method test_incompatible_labels_shape (line 284) | def test_incompatible_labels_shape(self):
    method test_predict (line 338) | def test_predict(self):
    method test_predict_with_tensor_n_classes (line 398) | def test_predict_with_tensor_n_classes(self):
    method test_predict_with_invalid_keys (line 458) | def test_predict_with_invalid_keys(self):
    method test_predict_with_vocabulary_list (line 470) | def test_predict_with_vocabulary_list(self):
    method test_weight_should_not_impact_prediction (line 499) | def test_weight_should_not_impact_prediction(self):
    method test_eval_create_loss (line 532) | def test_eval_create_loss(self):
    method test_eval_create_loss_loss_fn (line 554) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 577) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_eval_labels_none (line 602) | def test_eval_labels_none(self):
    method test_eval (line 617) | def test_eval(self):
    method test_eval_metric_ops_with_head_name (line 680) | def test_eval_metric_ops_with_head_name(self):
    method test_eval_with_regularization_losses (line 699) | def test_eval_with_regularization_losses(self):
    method test_eval_with_label_vocabulary_create_loss (line 762) | def test_eval_with_label_vocabulary_create_loss(self):
    method test_eval_with_label_vocabulary (line 784) | def test_eval_with_label_vocabulary(self):
    method test_weighted_multi_example_eval (line 835) | def test_weighted_multi_example_eval(self):
    method test_train_create_loss (line 908) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 946) | def test_train_create_loss_loss_reduction(self):
    method test_train_labels_none (line 987) | def test_train_labels_none(self):
    method test_train (line 1002) | def test_train(self):
    method test_train_with_regularization_losses (line 1060) | def test_train_with_regularization_losses(self):
    method test_train_one_dim_create_loss (line 1123) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 1165) | def test_train_one_dim(self):
    method test_train_with_vocabulary_create_loss (line 1241) | def test_train_with_vocabulary_create_loss(self):
    method test_train_with_vocabulary (line 1265) | def test_train_with_vocabulary(self):
    method test_weighted_multi_example_train (line 1299) | def test_weighted_multi_example_train(self):
    method test_multi_dim_weighted_train_create_loss (line 1361) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 1388) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 1435) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 1470) | def test_multi_dim_train_weights_wrong_outer_dim(self):
    method test_multi_dim_weighted_eval (line 1508) | def test_multi_dim_weighted_eval(self):
  class MultiClassHeadForEstimator (line 1571) | class MultiClassHeadForEstimator(tf.test.TestCase):
    method test_invalid_trainable_variables (line 1574) | def test_invalid_trainable_variables(self):
    method test_train_with_optimizer (line 1620) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 1667) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 1703) | def test_train_summaries_with_head_name(self):
    method test_lookup_tables_in_graph (line 1740) | def test_lookup_tables_in_graph(self):
    method test_missmatch_n_classes_label_vocabulary (line 1769) | def test_missmatch_n_classes_label_vocabulary(self):

FILE: tensorflow_estimator/python/estimator/head/multi_head.py
  function _no_op_train_fn (line 33) | def _no_op_train_fn(loss):
  function _default_export_output (line 38) | def _default_export_output(export_outputs, head_name):
  class MultiHead (line 54) | class MultiHead(base_head.Head):
    method __init__ (line 177) | def __init__(self, heads, head_weights=None):
    method name (line 203) | def name(self):
    method logits_dimension (line 208) | def logits_dimension(self):
    method loss_reduction (line 213) | def loss_reduction(self):
    method _split_logits (line 222) | def _split_logits(self, logits):
    method _check_logits_and_labels (line 284) | def _check_logits_and_labels(self, logits, labels=None):
    method loss (line 308) | def loss(self,
    method predictions (line 344) | def predictions(self, logits, keys=None):
    method metrics (line 355) | def metrics(self, regularization_losses=None):
    method update_metrics (line 369) | def update_metrics(self,
    method create_estimator_spec (line 397) | def create_estimator_spec(self,
    method _merge_predict_export_outputs (line 514) | def _merge_predict_export_outputs(self, all_estimator_spec):

FILE: tensorflow_estimator/python/estimator/head/multi_head_test.py
  class MultiHeadTest (line 36) | class MultiHeadTest(tf.test.TestCase):
    method test_no_heads (line 38) | def test_no_heads(self):
    method test_head_name_missing (line 43) | def test_head_name_missing(self):
    method test_head_weights_wrong_size (line 50) | def test_head_weights_wrong_size(self):
    method test_name (line 58) | def test_name(self):
    method test_predict_two_heads_logits_dict (line 64) | def test_predict_two_heads_logits_dict(self):
    method test_predict_two_heads_logits_tensor (line 140) | def test_predict_two_heads_logits_tensor(self):
    method test_predict_two_heads_logits_tensor_multi_dim (line 201) | def test_predict_two_heads_logits_tensor_multi_dim(self):
    method test_eval_two_heads_with_weights (line 253) | def test_eval_two_heads_with_weights(self):
    method test_train_loss_one_head (line 334) | def test_train_loss_one_head(self):
    method test_train_loss_two_heads_with_weights (line 351) | def test_train_loss_two_heads_with_weights(self):
    method test_train_loss_logits_tensor (line 392) | def test_train_loss_logits_tensor(self):
    method test_train_loss_logits_tensor_wrong_shape (line 430) | def test_train_loss_logits_tensor_wrong_shape(self):
    method test_train_loss_logits_tensor_multi_dim (line 459) | def test_train_loss_logits_tensor_multi_dim(self):
    method test_train_loss_logits_tensor_multi_dim_wrong_shape (line 496) | def test_train_loss_logits_tensor_multi_dim_wrong_shape(self):
    method test_train_one_head (line 519) | def test_train_one_head(self):
    method test_train_one_head_with_optimizer (line 584) | def test_train_one_head_with_optimizer(self):
    method test_train_two_heads_with_weights (line 637) | def test_train_two_heads_with_weights(self):
    method test_train_with_regularization_losses (line 724) | def test_train_with_regularization_losses(self):
  class MultiHeadForEstimator (line 829) | class MultiHeadForEstimator(tf.test.TestCase):
    method test_loss_reduction_must_be_same (line 832) | def test_loss_reduction_must_be_same(self):

FILE: tensorflow_estimator/python/estimator/head/multi_label_head.py
  class MultiLabelHead (line 37) | class MultiLabelHead(base_head.Head):
    method __init__ (line 152) | def __init__(self,
    method name (line 246) | def name(self):
    method logits_dimension (line 251) | def logits_dimension(self):
    method loss_reduction (line 256) | def loss_reduction(self):
    method _class_id_table (line 267) | def _class_id_table(self):
    method _processed_labels (line 282) | def _processed_labels(self, logits, labels):
    method _unweighted_loss_and_weights (line 320) | def _unweighted_loss_and_weights(self, logits, processed_labels, featu...
    method loss (line 341) | def loss(self,
    method predictions (line 366) | def predictions(self, logits, keys=None):
    method metrics (line 401) | def metrics(self, regularization_losses=None):
    method update_metrics (line 433) | def update_metrics(self,
    method _create_tpu_estimator_spec (line 486) | def _create_tpu_estimator_spec(self,

FILE: tensorflow_estimator/python/estimator/head/multi_label_head_test.py
  function _sigmoid_cross_entropy (line 34) | def _sigmoid_cross_entropy(labels, logits):
  class MultiLabelHead (line 44) | class MultiLabelHead(tf.test.TestCase):
    method test_n_classes_is_none (line 46) | def test_n_classes_is_none(self):
    method test_n_classes_is_1 (line 52) | def test_n_classes_is_1(self):
    method test_threshold_too_small (line 58) | def test_threshold_too_small(self):
    method test_threshold_too_large (line 63) | def test_threshold_too_large(self):
    method test_label_vocabulary_dict (line 68) | def test_label_vocabulary_dict(self):
    method test_label_vocabulary_wrong_size (line 74) | def test_label_vocabulary_wrong_size(self):
    method test_invalid_loss_reduction (line 80) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 89) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 99) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 109) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 116) | def test_loss_fn_arg_invalid(self):
    method test_classes_for_class_based_metrics_invalid (line 125) | def test_classes_for_class_based_metrics_invalid(self):
    method test_classes_for_class_based_metrics_string_invalid (line 133) | def test_classes_for_class_based_metrics_string_invalid(self):
    method test_predict (line 140) | def test_predict(self):
    method test_weight_should_not_impact_prediction (line 185) | def test_weight_should_not_impact_prediction(self):
    method test_eval_create_loss (line 227) | def test_eval_create_loss(self):
    method test_eval_create_loss_large_logits (line 244) | def test_eval_create_loss_large_logits(self):
    method test_eval_create_loss_labels_wrong_shape (line 264) | def test_eval_create_loss_labels_wrong_shape(self):
    method test_eval_create_loss_loss_fn (line 306) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 329) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_eval_labels_none (line 356) | def test_eval_labels_none(self):
    method _test_eval (line 367) | def _test_eval(self,
    method test_eval (line 428) | def test_eval(self):
    method test_eval_sparse_labels (line 454) | def test_eval_sparse_labels(self):
    method test_eval_with_regularization_losses (line 483) | def test_eval_with_regularization_losses(self):
    method test_eval_with_label_vocabulary (line 514) | def test_eval_with_label_vocabulary(self):
    method test_eval_with_label_vocabulary_with_multi_hot_input (line 546) | def test_eval_with_label_vocabulary_with_multi_hot_input(self):
    method test_eval_with_thresholds (line 573) | def test_eval_with_thresholds(self):
    method test_eval_with_classes_for_class_based_metrics (line 612) | def test_eval_with_classes_for_class_based_metrics(self):
    method test_eval_with_classes_for_class_based_metrics_string (line 649) | def test_eval_with_classes_for_class_based_metrics_string(self):
    method test_eval_with_weights (line 692) | def test_eval_with_weights(self):
    method test_train_create_loss_large_logits (line 726) | def test_train_create_loss_large_logits(self):
    method test_train_create_loss_loss_reduction (line 751) | def test_train_create_loss_loss_reduction(self):
    method test_train_labels_none (line 781) | def test_train_labels_none(self):
    method test_train_invalid_indicator_labels (line 793) | def test_train_invalid_indicator_labels(self):
    method test_train_invalid_sparse_labels (line 828) | def test_train_invalid_sparse_labels(self):
    method _test_train (line 862) | def _test_train(self, head, logits, labels, expected_loss):
    method test_train (line 905) | def test_train(self):
    method test_train_sparse_labels (line 919) | def test_train_sparse_labels(self):
    method test_train_with_label_vocabulary (line 935) | def test_train_with_label_vocabulary(self):
    method test_train_with_regularization_losses (line 954) | def test_train_with_regularization_losses(self):
    method test_train_with_weights (line 1011) | def test_train_with_weights(self):
    method test_multi_dim_weighted_train_create_loss (line 1071) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 1098) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_weights_wrong_inner_dim (line 1147) | def test_multi_dim_weights_wrong_inner_dim(self):
    method test_multi_dim_weights_wrong_outer_dim (line 1185) | def test_multi_dim_weights_wrong_outer_dim(self):
    method test_multi_dim_weighted_eval (line 1227) | def test_multi_dim_weighted_eval(self):
  class MultiLabelHeadForEstimator (line 1259) | class MultiLabelHeadForEstimator(tf.test.TestCase):
    method test_invalid_trainable_variables (line 1262) | def test_invalid_trainable_variables(self):
    method test_train_with_optimizer (line 1301) | def test_train_with_optimizer(self):
    method test_predict_with_label_vocabulary (line 1347) | def test_predict_with_label_vocabulary(self):
    method test_train_with_update_ops (line 1373) | def test_train_with_update_ops(self):
    method test_lookup_tables_in_graph (line 1404) | def test_lookup_tables_in_graph(self):

FILE: tensorflow_estimator/python/estimator/head/regression_head.py
  class RegressionHead (line 36) | class RegressionHead(base_head.Head):
    method __init__ (line 136) | def __init__(self,
    method name (line 162) | def name(self):
    method logits_dimension (line 167) | def logits_dimension(self):
    method loss_reduction (line 172) | def loss_reduction(self):
    method _processed_labels (line 176) | def _processed_labels(self, logits, labels):
    method _unweighted_loss_and_weights (line 184) | def _unweighted_loss_and_weights(self, logits, labels, features):
    method loss (line 205) | def loss(self,
    method predictions (line 230) | def predictions(self, logits):
    method metrics (line 256) | def metrics(self, regularization_losses=None):
    method update_metrics (line 273) | def update_metrics(self,
    method _create_tpu_estimator_spec (line 301) | def _create_tpu_estimator_spec(self,
  class PoissonRegressionHead (line 412) | class PoissonRegressionHead(RegressionHead):
    method __init__ (line 479) | def __init__(self,
    method _poisson_loss (line 494) | def _poisson_loss(self, labels, logits):
  class LogisticRegressionHead (line 502) | class LogisticRegressionHead(RegressionHead):
    method _logistic_loss (line 569) | def _logistic_loss(self, labels, logits):
    method __init__ (line 575) | def __init__(self,

FILE: tensorflow_estimator/python/estimator/head/regression_head_test.py
  class RegressionHead (line 34) | class RegressionHead(tf.test.TestCase):
    method test_invalid_label_dimension (line 36) | def test_invalid_label_dimension(self):
    method test_invalid_loss_reduction (line 42) | def test_invalid_loss_reduction(self):
    method test_loss_fn_arg_labels_missing (line 49) | def test_loss_fn_arg_labels_missing(self):
    method test_loss_fn_arg_logits_missing (line 59) | def test_loss_fn_arg_logits_missing(self):
    method test_loss_fn_arg_features_ok (line 69) | def test_loss_fn_arg_features_ok(self):
    method test_loss_fn_arg_invalid (line 75) | def test_loss_fn_arg_invalid(self):
    method test_invalid_logits (line 84) | def test_invalid_logits(self):
    method test_incompatible_labels_eval (line 109) | def test_incompatible_labels_eval(self):
    method test_incompatible_labels_train (line 165) | def test_incompatible_labels_train(self):
    method test_predict (line 221) | def test_predict(self):
    method test_predict_with_inverse_link_fn (line 259) | def test_predict_with_inverse_link_fn(self):
    method test_eval_create_loss (line 304) | def test_eval_create_loss(self):
    method test_eval_create_loss_loss_fn (line 314) | def test_eval_create_loss_loss_fn(self):
    method test_eval_create_loss_loss_fn_wrong_shape (line 337) | def test_eval_create_loss_loss_fn_wrong_shape(self):
    method test_eval_labels_none (line 363) | def test_eval_labels_none(self):
    method test_eval (line 378) | def test_eval(self):
    method test_eval_metric_ops_with_head_name_for_regression (line 441) | def test_eval_metric_ops_with_head_name_for_regression(self):
    method test_eval_with_regularization_losses (line 458) | def test_eval_with_regularization_losses(self):
    method test_train_create_loss (line 523) | def test_train_create_loss(self):
    method test_train_create_loss_loss_reduction (line 535) | def test_train_create_loss_loss_reduction(self):
    method test_train_labels_none (line 548) | def test_train_labels_none(self):
    method test_train (line 571) | def test_train(self):
    method test_train_with_regularization_losses (line 629) | def test_train_with_regularization_losses(self):
    method test_weighted_multi_example_eval (line 691) | def test_weighted_multi_example_eval(self):
    method test_weight_with_numeric_column (line 754) | def test_weight_with_numeric_column(self):
    method test_weighted_multi_example_train (line 773) | def test_weighted_multi_example_train(self):
    method test_train_one_dim_create_loss (line 837) | def test_train_one_dim_create_loss(self):
    method test_train_one_dim (line 856) | def test_train_one_dim(self):
    method test_weighted_multi_value_eval_create_loss (line 924) | def test_weighted_multi_value_eval_create_loss(self):
    method test_weighted_multi_value_eval (line 942) | def test_weighted_multi_value_eval(self):
    method test_weighted_multi_value_train_create_loss (line 1012) | def test_weighted_multi_value_train_create_loss(self):
    method test_weighted_multi_value_train (line 1028) | def test_weighted_multi_value_train(self):
    method test_weighted_multi_batch_eval_eager (line 1092) | def test_weighted_multi_batch_eval_eager(self):
    method test_weighted_multi_batch_train_eager (line 1130) | def test_weighted_multi_batch_train_eager(self):
    method test_multi_dim_weighted_train_create_loss (line 1154) | def test_multi_dim_weighted_train_create_loss(self):
    method test_multi_dim_weighted_train (line 1181) | def test_multi_dim_weighted_train(self):
    method test_multi_dim_train_weights_wrong_inner_dim (line 1220) | def test_multi_dim_train_weights_wrong_inner_dim(self):
    method test_multi_dim_train_weights_wrong_outer_dim (line 1258) | def test_multi_dim_train_weights_wrong_outer_dim(self):
  class RegressionHeadForEstimator (line 1307) | class RegressionHeadForEstimator(tf.test.TestCase):
    method test_invalid_trainable_variables (line 1310) | def test_invalid_trainable_variables(self):
    method test_train_with_optimizer (line 1349) | def test_train_with_optimizer(self):
    method test_train_with_update_ops (line 1389) | def test_train_with_update_ops(self):
    method test_train_summaries_with_head_name (line 1420) | def test_train_summaries_with_head_name(self):
    method test_weighted_multi_batch_train (line 1454) | def test_weighted_multi_batch_train(self):
    method test_weighted_multi_batch_eval (line 1503) | def test_weighted_multi_batch_eval(self):
  class PoissonRegressionHead (line 1576) | class PoissonRegressionHead(tf.test.TestCase):
    method test_train (line 1578) | def test_train(self):
    method test_predict (line 1628) | def test_predict(self):
  class LogisticRegressionHead (line 1644) | class LogisticRegressionHead(tf.test.TestCase):
    method test_train (line 1646) | def test_train(self):
    method test_train_labels_too_large (line 1694) | def test_train_labels_too_large(self):
    method test_train_labels_negative (line 1729) | def test_train_labels_negative(self):
    method test_predict (line 1765) | def test_predict(self):

FILE: tensorflow_estimator/python/estimator/head/sequential_head.py
  class _SequentialHead (line 37) | class _SequentialHead(base_head.Head):
    method input_sequence_mask_key (line 70) | def input_sequence_mask_key(self):
  class SequentialHeadWrapper (line 79) | class SequentialHeadWrapper(_SequentialHead):
    method __init__ (line 91) | def __init__(self,
    method _flatten (line 181) | def _flatten(self, labels, logits, features):
    method loss (line 235) | def loss(self,
    method create_estimator_spec (line 274) | def create_estimator_spec(self,
    method update_metrics (line 351) | def update_metrics(self,
    method _create_tpu_estimator_spec (line 389) | def _create_tpu_estimator_spec(self,
    method predictions (line 401) | def predictions(self, logits, keys=None):
    method metrics (line 405) | def metrics(self, regularization_losses=None):
    method input_sequence_mask_key (line 410) | def input_sequence_mask_key(self):
    method logits_dimension (line 415) | def logits_dimension(self):
    method loss_reduction (line 420) | def loss_reduction(self):
    method name (line 425) | def name(self):
    method static_head (line 432) | def static_head(self):
  function _flatten_tensor (line 437) | def _flatten_tensor(tensor, sequence_mask, expected_length):

FILE: tensorflow_estimator/python/estimator/head/sequential_head_test.py
  function _convert_to_tensor (line 39) | def _convert_to_tensor(features):
  class TestFlatten (line 51) | class TestFlatten(tf.test.TestCase, parameterized.TestCase):
    method test_flatten_tensor (line 89) | def test_flatten_tensor(self, tensor, expected):
    method _test_flatten_method (line 106) | def _test_flatten_method(self, features, feature_columns):
    method test_flatten_method (line 128) | def test_flatten_method(self):
    method test_flatten_with_one_feature_columns (line 135) | def test_flatten_with_one_feature_columns(self):
    method test_flatten_with_multiple_feature_columns (line 147) | def test_flatten_with_multiple_feature_columns(self):
    method test_flatten_no_mask (line 161) | def test_flatten_no_mask(self):
    method test_flatten_missing_feature (line 169) | def test_flatten_missing_feature(self):
    method test_flatten_tensor_wrong_feature_dim (line 176) | def test_flatten_tensor_wrong_feature_dim(self):
    method test_flatten_tensor_wrong_feature_mask (line 186) | def test_flatten_tensor_wrong_feature_mask(self):
    method test_flatten_tensor_wrong_mask_dim (line 196) | def test_flatten_tensor_wrong_mask_dim(self):
  class _MockHead (line 204) | class _MockHead(object):
    method metrics (line 207) | def metrics(self, regularization_losses=None):
    method loss (line 210) | def loss(self, **kwargs):
    method create_estimator_spec (line 213) | def create_estimator_spec(self, **kwargs):
  class TestSequentialHead (line 219) | class TestSequentialHead(tf.test.TestCase):
    method _assert_equal (line 222) | def _assert_equal(self, d, dref, session=None):
    method test_predictions (line 245) | def test_predictions(self):
    method test_metrics (line 288) | def test_metrics(self):
    method test_loss_args (line 303) | def test_loss_args(self):
    method test_create_estimator_spec_args (line 327) | def test_create_estimator_spec_args(self):
    method test_head_properties (line 363) | def test_head_properties(self):
    method test_loss_reduction (line 375) | def test_loss_reduction(self):
    method test_metrics_computation (line 434) | def test_metrics_computation(self):
    method test_wrong_mask_type (line 508) | def test_wrong_mask_type(self):
    method test_wrong_feature_column_type (line 514) | def test_wrong_feature_column_type(self):
    method test_wrong_feature_column_type_in_iterable (line 520) | def test_wrong_feature_column_type_in_iterable(self):
    method test_multi_head_provided (line 526) | def test_multi_head_provided(self):

FILE: tensorflow_estimator/python/estimator/hooks/basic_session_run_hooks_test.py
  class MockCheckpointSaverListener (line 40) | class MockCheckpointSaverListener(
    method __init__ (line 43) | def __init__(self):
    method begin (line 50) | def begin(self):
    method before_save (line 53) | def before_save(self, session, global_step):
    method after_save (line 56) | def after_save(self, session, global_step):
    method end (line 61) | def end(self, session, global_step):
    method get_counts (line 64) | def get_counts(self):
  class SecondOrStepTimerTest (line 74) | class SecondOrStepTimerTest(tf.test.TestCase):
    method test_raise_in_both_secs_and_steps (line 76) | def test_raise_in_both_secs_and_steps(self):
    method test_raise_in_none_secs_and_steps (line 80) | def test_raise_in_none_secs_and_steps(self):
    method test_every_secs (line 85) | def test_every_secs(self, mock_time):
    method test_every_steps (line 98) | def test_every_steps(self):
    method test_update_last_triggered_step (line 108) | def test_update_last_triggered_step(self):
  class StopAtStepTest (line 125) | class StopAtStepTest(tf.test.TestCase):
    method test_raise_in_both_last_step_and_num_steps (line 127) | def test_raise_in_both_last_step_and_num_steps(self):
    method test_stop_based_on_last_step (line 131) | def test_stop_based_on_last_step(self):
    method test_stop_based_on_num_step (line 154) | def test_stop_based_on_num_step(self):
    method test_stop_based_with_multiple_steps (line 181) | def test_stop_based_with_multiple_steps(self):
  class LoggingTensorHookTest (line 200) | class LoggingTensorHookTest(tf.test.TestCase):
    method setUp (line 202) | def setUp(self):
    method tearDown (line 214) | def tearDown(self):
    method test_illegal_args (line 217) | def test_illegal_args(self):
    method test_print_at_end_only (line 228) | def test_print_at_end_only(self):
    method _validate_print_every_n_steps (line 246) | def _validate_print_every_n_steps(self, sess, at_end):
    method test_print_every_n_steps (line 280) | def test_print_every_n_steps(self):
    method test_print_every_n_steps_and_end (line 286) | def test_print_every_n_steps_and_end(self):
    method test_print_first_step (line 292) | def test_print_first_step(self):
    method _validate_print_every_n_secs (line 307) | def _validate_print_every_n_secs(self, sess, at_end, mock_time):
    method test_print_every_n_secs (line 339) | def test_print_every_n_secs(self, mock_time):
    method test_print_every_n_secs_and_end (line 347) | def test_print_every_n_secs_and_end(self, mock_time):
    method test_print_formatter (line 354) | def test_print_formatter(self):
  class CheckpointSaverHookTest (line 370) | class CheckpointSaverHookTest(tf.test.TestCase):
    method setUp (line 372) | def setUp(self):
    method tearDown (line 380) | def tearDown(self):
    method test_saves_when_saver_and_scaffold_both_missing (line 383) | def test_saves_when_saver_and_scaffold_both_missing(self):
    method test_raise_when_saver_and_scaffold_both_present (line 396) | def test_raise_when_saver_and_scaffold_both_present(self):
    method test_raise_in_both_secs_and_steps (line 401) | def test_raise_in_both_secs_and_steps(self):
    method test_raise_in_none_secs_and_steps (line 406) | def test_raise_in_none_secs_and_steps(self):
    method test_save_secs_saves_in_first_step (line 410) | def test_save_secs_saves_in_first_step(self):
    method test_save_secs_calls_listeners_at_begin_and_end (line 423) | def test_save_secs_calls_listeners_at_begin_and_end(self):
    method test_listener_with_monitored_session (line 446) | def test_listener_with_monitored_session(self):
    method test_listener_stops_training_in_after_save (line 469) | def test_listener_stops_training_in_after_save(self):
    method test_listener_with_default_saver (line 488) | def test_listener_with_default_saver(self):
    method test_two_listeners_with_default_saver (line 516) | def test_two_listeners_with_default_saver(self):
    method test_save_secs_saves_periodically (line 548) | def test_save_secs_saves_periodically(self, mock_time):
    method test_save_secs_calls_listeners_periodically (line 589) | def test_save_secs_calls_listeners_periodically(self, mock_time):
    method test_save_steps_saves_in_first_step (line 634) | def test_save_steps_saves_in_first_step(self):
    method test_save_steps_saves_periodically (line 647) | def test_save_steps_saves_periodically(self):
    method test_save_saves_at_end (line 674) | def test_save_saves_at_end(self):
    method test_summary_writer_defs (line 689) | def test_summary_writer_defs(self):
    method test_save_checkpoint_before_first_train_step (line 715) | def test_save_checkpoint_before_first_train_step(self):
  class CheckpointSaverHookMultiStepTest (line 739) | class CheckpointSaverHookMultiStepTest(tf.test.TestCase):
    method setUp (line 741) | def setUp(self):
    method tearDown (line 750) | def tearDown(self):
    method test_save_steps_saves_in_first_step (line 753) | def test_save_steps_saves_in_first_step(self):
    method test_save_steps_saves_periodically (line 769) | def test_save_steps_saves_periodically(self):
    method test_save_steps_saves_at_end (line 806) | def test_save_steps_saves_at_end(self):
  class ResourceCheckpointSaverHookTest (line 826) | class ResourceCheckpointSaverHookTest(tf.test.TestCase):
    method setUp (line 828) | def setUp(self):
    method test_save_steps_saves_periodically (line 837) | def test_save_steps_saves_periodically(self):
  class StepCounterHookTest (line 866) | class StepCounterHookTest(tf.test.TestCase):
    method setUp (line 868) | def setUp(self):
    method tearDown (line 871) | def tearDown(self):
    method test_step_counter_every_n_steps (line 875) | def test_step_counter_every_n_steps(self, mock_time):
    method test_step_counter_every_n_secs (line 906) | def test_step_counter_every_n_secs(self, mock_time):
    method test_global_step_name (line 937) | def test_global_step_name(self):
    method test_log_warning_if_global_step_not_increased (line 970) | def test_log_warning_if_global_step_not_increased(self):
    method _setup_steps_per_run_test (line 988) | def _setup_steps_per_run_test(self, every_n_steps, steps_per_run, graph,
    method test_steps_per_run_less_than_every_n_steps (line 1002) | def test_steps_per_run_less_than_every_n_steps(self, mock_time):
    method test_steps_per_run_equal_every_n_steps (line 1025) | def test_steps_per_run_equal_every_n_steps(self, mock_time):
    method test_steps_per_run_greater_than_every_n_steps (line 1049) | def test_steps_per_run_greater_than_every_n_steps(self, mock_time):
  class SummarySaverHookTest (line 1074) | class SummarySaverHookTest(tf.test.TestCase):
    method setUp (line 1076) | def setUp(self):
    method test_raise_when_scaffold_and_summary_op_both_missing (line 1091) | def test_raise_when_scaffold_and_summary_op_both_missing(self):
    method test_raise_when_scaffold_and_summary_op_both_present (line 1095) | def test_raise_when_scaffold_and_summary_op_both_present(self):
    method test_raise_in_both_secs_and_steps (line 1100) | def test_raise_in_both_secs_and_steps(self):
    method test_raise_in_none_secs_and_steps (line 1105) | def test_raise_in_none_secs_and_steps(self):
    method test_save_steps (line 1110) | def test_save_steps(self):
    method test_multiple_summaries (line 1142) | def test_multiple_summaries(self):
    method test_save_secs_saving_once_every_step (line 1171) | def test_save_secs_saving_once_every_step(self, mock_time):
    method test_save_secs_saving_once_every_three_steps (line 1206) | def test_save_secs_saving_once_every_three_steps(self, mock_time):
  class GlobalStepWaiterHookTest (line 1240) | class GlobalStepWaiterHookTest(tf.test.TestCase):
    method test_not_wait_for_step_zero (line 1242) | def test_not_wait_for_step_zero(self):
    method test_wait_for_step (line 1254) | def test_wait_for_step(self, mock_sleep):
  class FinalOpsHookTest (line 1293) | class FinalOpsHookTest(tf.test.TestCase):
    method test_final_ops_is_scalar_tensor (line 1295) | def test_final_ops_is_scalar_tensor(self):
    method test_final_ops_is_tensor (line 1307) | def test_final_ops_is_tensor(self):
    method test_final_ops_triggers_out_of_range_error (line 1319) | def test_final_ops_triggers_out_of_range_error(self):
    method test_final_ops_with_dictionary (line 1339) | def test_final_ops_with_dictionary(self):
  class ResourceSummarySaverHookTest (line 1355) | class ResourceSummarySaverHookTest(tf.test.TestCase):
    method setUp (line 1357) | def setUp(self):
    method test_save_steps (line 1371) | def test_save_steps(self):
  class FeedFnHookTest (line 1405) | class FeedFnHookTest(tf.test.TestCase):
    method test_feeding_placeholder (line 1407) | def test_feeding_placeholder(self):
  class ProfilerHookTest (line 1418) | class ProfilerHookTest(tf.test.TestCase):
    method setUp (line 1420) | def setUp(self):
    method tearDown (line 1429) | def tearDown(self):
    method _count_timeline_files (line 1433) | def _count_timeline_files(self):
    method test_raise_in_both_secs_and_steps (line 1436) | def test_raise_in_both_secs_and_steps(self):
    method test_raise_in_none_secs_and_steps (line 1440) | def test_raise_in_none_secs_and_steps(self):
    method test_save_secs_does_not_save_in_first_step (line 1444) | def test_save_secs_does_not_save_in_first_step(self):
    method test_save_secs_saves_periodically (line 1453) | def test_save_secs_saves_periodically(self, mock_time):
    method test_save_steps_does_not_save_in_first_step (line 1479) | def test_save_steps_does_not_save_in_first_step(self):
    method test_save_steps_saves_periodically (line 1487) | def test_save_steps_saves_periodically(self):
    method test_run_metadata_saves (line 1504) | def test_run_metadata_saves(self):

FILE: tensorflow_estimator/python/estimator/hooks/fake_summary_writer.py
  class FakeSummaryWriter (line 27) | class FakeSummaryWriter(object):
    method install (line 33) | def install(cls):
    method uninstall (line 41) | def uninstall(cls):
    method __init__ (line 48) | def __init__(self, logdir, graph=None):
    method summaries (line 58) | def summaries(self):
    method assert_summaries (line 61) | def assert_summaries(self,
    method add_summary (line 98) | def add_summary(self, summ, current_global_step):
    method add_graph (line 112) | def add_graph(self, graph, global_step=None, graph_def=None):
    method add_meta_graph (line 120) | def add_meta_graph(self, meta_graph_def, global_step=None):
    method add_session_log (line 127) | def add_session_log(self, session_log, global_step=None):
    method add_run_metadata (line 131) | def add_run_metadata(self, run_metadata, tag, global_step=None):
    method flush (line 136) | def flush(self):
    method reopen (line 139) | def reopen(self):
    method close (line 142) | def close(self):

FILE: tensorflow_estimator/python/estimator/hooks/hooks.py
  class InMemoryEvaluatorHook (line 31) | class InMemoryEvaluatorHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 62) | def __init__(self,
    method begin (line 122) | def begin(self):
    method after_create_session (line 146) | def after_create_session(self, session, coord):  # pylint: disable=unu...
    method _evaluate (line 178) | def _evaluate(self, train_session):
    method after_run (line 203) | def after_run(self, run_context, run_values):  # pylint: disable=unuse...
    method end (line 209) | def end(self, session):  # pylint: disable=unused-argument
  class _StopAtCheckpointStepHook (line 214) | class _StopAtCheckpointStepHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 221) | def __init__(self, model_dir, last_step, wait_after_file_check_secs=30):
    method begin (line 246) | def begin(self):
    method before_run (line 252) | def before_run(self, run_context):  # pylint: disable=unused-argument
    method after_run (line 255) | def after_run(self, run_context, run_values):
  function make_stop_at_checkpoint_step_hook (line 270) | def make_stop_at_checkpoint_step_hook(estimator,

FILE: tensorflow_estimator/python/estimator/hooks/hooks_test.py
  function summary_step_keyword_to_value_mapping (line 34) | def summary_step_keyword_to_value_mapping(dir_):
  function get_summary_value (line 50) | def get_summary_value(dir_, step, keyword):
  class InMemoryEvaluatorHookTest (line 66) | class InMemoryEvaluatorHookTest(tf.test.TestCase):
    method test_runs_eval_metrics (line 68) | def test_runs_eval_metrics(self):
    method test_uses_latest_variable_value (line 111) | def test_uses_latest_variable_value(self):
    method test_dnn_classifier (line 160) | def test_dnn_classifier(self):
    method test_raise_error_with_multi_worker (line 192) | def test_raise_error_with_multi_worker(self):
    method test_raise_error_with_ps (line 215) | def test_raise_error_with_ps(self):
    method test_raise_error_with_custom_saver_in_eval (line 238) | def test_raise_error_with_custom_saver_in_eval(self):
    method test_raise_error_with_custom_init_fn_in_eval (line 263) | def test_raise_error_with_custom_init_fn_in_eval(self):
    method test_raise_error_with_saveables_other_than_global_variables (line 291) | def test_raise_error_with_saveables_other_than_global_variables(self):
  class StopAtCheckpointStepHookTest (line 325) | class StopAtCheckpointStepHookTest(tf.test.TestCase):
    method test_do_not_stop_if_checkpoint_is_not_there (line 327) | def test_do_not_stop_if_checkpoint_is_not_there(self):
    method test_do_not_stop_if_checkpoint_step_is_smaller (line 342) | def test_do_not_stop_if_checkpoint_step_is_smaller(self):
    method test_stop_if_checkpoint_step_is_laststep (line 363) | def test_stop_if_checkpoint_step_is_laststep(self):
    method test_creates_regular_stop_at_step_hook_for_chief (line 383) | def test_creates_regular_stop_at_step_hook_for_chief(self):
    method test_creates_checkpoint_hook_for_workers (line 392) | def test_creates_checkpoint_hook_for_workers(self):

FILE: tensorflow_estimator/python/estimator/inputs/numpy_io.py
  function _get_unique_target_key (line 33) | def _get_unique_target_key(features):
  function _validate_and_convert_features (line 56) | def _validate_and_convert_features(x):
  function numpy_input_fn (line 90) | def numpy_input_fn(x,

FILE: tensorflow_estimator/python/estimator/inputs/numpy_io_test.py
  class NumpyIoTest (line 29) | class NumpyIoTest(tf.test.TestCase):
    method testNumpyInputFn (line 31) | def testNumpyInputFn(self):
    method testNumpyInputFnWithVeryLargeBatchSizeAndMultipleEpochs (line 58) | def testNumpyInputFnWithVeryLargeBatchSizeAndMultipleEpochs(self):
    method testNumpyInputFnWithZeroEpochs (line 84) | def testNumpyInputFnWithZeroEpochs(self):
    method testNumpyInputFnWithBatchSizeNotDividedByDataSize (line 105) | def testNumpyInputFnWithBatchSizeNotDividedByDataSize(self):
    method testNumpyInputFnWithBatchSizeNotDividedByDataSizeAndMultipleEpochs (line 142) | def testNumpyInputFnWithBatchSizeNotDividedByDataSizeAndMultipleEpochs...
    method testNumpyInputFnWithBatchSizeLargerThanDataSize (line 189) | def testNumpyInputFnWithBatchSizeLargerThanDataSize(self):
    method testNumpyInputFnWithDifferentDimensionsOfFeatures (line 216) | def testNumpyInputFnWithDifferentDimensionsOfFeatures(self):
    method testNumpyInputFnWithXAsNonDict (line 239) | def testNumpyInputFnWithXAsNonDict(self):
    method testNumpyInputFnWithXIsEmptyDict (line 248) | def testNumpyInputFnWithXIsEmptyDict(self):
    method testNumpyInputFnWithXIsEmptyArray (line 256) | def testNumpyInputFnWithXIsEmptyArray(self):
    method testNumpyInputFnWithYIsNone (line 264) | def testNumpyInputFnWithYIsNone(self):
    method testNumpyInputFnWithNonBoolShuffle (line 291) | def testNumpyInputFnWithNonBoolShuffle(self):
    method testNumpyInputFnWithTargetKeyAlreadyInX (line 301) | def testNumpyInputFnWithTargetKeyAlreadyInX(self):
    method testNumpyInputFnWithMismatchLengthOfInputs (line 314) | def testNumpyInputFnWithMismatchLengthOfInputs(self):
    method testNumpyInputFnWithYAsDict (line 338) | def testNumpyInputFnWithYAsDict(self):
    method testNumpyInputFnWithYIsEmptyDict (line 368) | def testNumpyInputFnWithYIsEmptyDict(self):
    method testNumpyInputFnWithDuplicateKeysInXAndY (line 378) | def testNumpyInputFnWithDuplicateKeysInXAndY(self):
    method testNumpyInputFnWithXIsArray (line 389) | def testNumpyInputFnWithXIsArray(self):
    method testNumpyInputFnWithXIsNDArray (line 406) | def testNumpyInputFnWithXIsNDArray(self):
    method testNumpyInputFnWithXIsArrayYIsDict (line 424) | def testNumpyInputFnWithXIsArrayYIsDict(self):
    method testArrayAndDictGiveSameOutput (line 443) | def testArrayAndDictGiveSameOutput(self):
  class FeatureColumnIntegrationTest (line 467) | class FeatureColumnIntegrationTest(tf.test.TestCase):
    method _initialized_session (line 469) | def _initialized_session(self, config=None):
    method _get_linear_model_bias (line 475) | def _get_linear_model_bias(self, name='linear_model'):
    method _get_linear_model_column_var (line 479) | def _get_linear_model_column_var(self, column, name='linear_model'):
    method _get_keras_linear_model_predictions (line 483) | def _get_keras_linear_model_predictions(self,
    method test_linear_model_numpy_input_fn (line 503) | def test_linear_model_numpy_input_fn(self):
    method test_linear_model_impl_numpy_input_fn (line 539) | def test_linear_model_impl_numpy_input_fn(self):
    method test_functional_input_layer_with_numpy_input_fn (line 579) | def test_functional_input_layer_with_numpy_input_fn(self):

FILE: tensorflow_estimator/python/estimator/inputs/pandas_io.py
  function _get_unique_target_key (line 39) | def _get_unique_target_key(features, target_column_name):
  function pandas_input_fn (line 56) | def pandas_input_fn(x,

FILE: tensorflow_estimator/python/estimator/inputs/pandas_io_test.py
  class PandasIoTest (line 38) | class PandasIoTest(tf.test.TestCase):
    method makeTestDataFrame (line 40) | def makeTestDataFrame(self):
    method makeTestDataFrameWithYAsDataFrame (line 48) | def makeTestDataFrameWithYAsDataFrame(self):
    method callInputFnOnce (line 58) | def callInputFnOnce(self, input_fn, session):
    method testPandasInputFn_IndexMismatch (line 68) | def testPandasInputFn_IndexMismatch(self):
    method testPandasInputFn_RaisesWhenTargetColumnIsAList (line 77) | def testPandasInputFn_RaisesWhenTargetColumnIsAList(self):
    method testPandasInputFn_NonBoolShuffle (line 93) | def testPandasInputFn_NonBoolShuffle(self):
    method testPandasInputFn_ProducesExpectedOutputs (line 104) | def testPandasInputFn_ProducesExpectedOutputs(self):
    method testPandasInputFnWhenYIsDataFrame_ProducesExpectedOutput (line 118) | def testPandasInputFnWhenYIsDataFrame_ProducesExpectedOutput(self):
    method testPandasInputFnYIsDataFrame_HandlesOverlappingColumns (line 133) | def testPandasInputFnYIsDataFrame_HandlesOverlappingColumns(self):
    method testPandasInputFnYIsDataFrame_HandlesOverlappingColumnsInTargets (line 149) | def testPandasInputFnYIsDataFrame_HandlesOverlappingColumnsInTargets(s...
    method testPandasInputFn_ProducesOutputsForLargeBatchAndMultipleEpochs (line 165) | def testPandasInputFn_ProducesOutputsForLargeBatchAndMultipleEpochs(se...
    method testPandasInputFn_ProducesOutputsWhenDataSizeNotDividedByBatchSize (line 194) | def testPandasInputFn_ProducesOutputsWhenDataSizeNotDividedByBatchSize...
    method testPandasInputFn_OnlyX (line 234) | def testPandasInputFn_OnlyX(self):
    method testPandasInputFn_ExcludesIndex (line 247) | def testPandasInputFn_ExcludesIndex(self):
    method assertInputsCallableNTimes (line 259) | def assertInputsCallableNTimes(self, input_fn, session, n):
    method testPandasInputFn_RespectsEpoch_NoShuffle (line 271) | def testPandasInputFn_RespectsEpoch_NoShuffle(self):
    method testPandasInputFn_RespectsEpoch_WithShuffle (line 281) | def testPandasInputFn_RespectsEpoch_WithShuffle(self):
    method testPandasInputFn_RespectsEpoch_WithShuffleAutosize (line 291) | def testPandasInputFn_RespectsEpoch_WithShuffleAutosize(self):
    method testPandasInputFn_RespectsEpochUnevenBatches (line 301) | def testPandasInputFn_RespectsEpochUnevenBatches(self):
    method testPandasInputFn_Idempotent (line 312) | def testPandasInputFn_Idempotent(self):

FILE: tensorflow_estimator/python/estimator/inputs/queues/feeding_functions.py
  function _fill_array (line 41) | def _fill_array(arr, seq, fillvalue=0):
  function _pad_if_needed (line 63) | def _pad_if_needed(batch_key_item, fillvalue=0):
  function _get_integer_indices_for_next_batch (line 99) | def _get_integer_indices_for_next_batch(batch_indices_start, batch_size,
  class _ArrayFeedFn (line 148) | class _ArrayFeedFn(object):
    method __init__ (line 151) | def __init__(self,
    method __call__ (line 171) | def __call__(self):
  class _OrderedDictNumpyFeedFn (line 187) | class _OrderedDictNumpyFeedFn(object):
    method __init__ (line 190) | def __init__(self,
    method __call__ (line 214) | def __call__(self):
  class _PandasFeedFn (line 233) | class _PandasFeedFn(object):
    method __init__ (line 236) | def __init__(self,
    method __call__ (line 257) | def __call__(self):
  class _GeneratorFeedFn (line 274) | class _GeneratorFeedFn(object):
    method __init__ (line 277) | def __init__(self,
    method __call__ (line 299) | def __call__(self):
  function _enqueue_data (line 333) | def _enqueue_data(data,

FILE: tensorflow_estimator/python/estimator/inputs/queues/feeding_functions_test.py
  function vals_to_list (line 38) | def vals_to_list(a):
  class _FeedingFunctionsTestCase (line 45) | class _FeedingFunctionsTestCase(tf.test.TestCase):
    method testArrayFeedFnBatchOne (line 48) | def testArrayFeedFnBatchOne(self):
    method testArrayFeedFnBatchFive (line 63) | def testArrayFeedFnBatchFive(self):
    method testArrayFeedFnBatchTwoWithOneEpoch (line 79) | def testArrayFeedFnBatchTwoWithOneEpoch(self):
    method testArrayFeedFnBatchOneHundred (line 96) | def testArrayFeedFnBatchOneHundred(self):
    method testArrayFeedFnBatchOneHundredWithSmallerArrayAndMultipleEpochs (line 111) | def testArrayFeedFnBatchOneHundredWithSmallerArrayAndMultipleEpochs(se...
    method testPandasFeedFnBatchOne (line 123) | def testPandasFeedFnBatchOne(self):
    method testPandasFeedFnBatchFive (line 143) | def testPandasFeedFnBatchFive(self):
    method testPandasFeedFnBatchTwoWithOneEpoch (line 164) | def testPandasFeedFnBatchTwoWithOneEpoch(self):
    method testPandasFeedFnBatchOneHundred (line 197) | def testPandasFeedFnBatchOneHundred(self):
    method testPandasFeedFnBatchOneHundredWithSmallDataArrayAndMultipleEpochs (line 214) | def testPandasFeedFnBatchOneHundredWithSmallDataArrayAndMultipleEpochs...
    method testOrderedDictNumpyFeedFnBatchTwoWithOneEpoch (line 231) | def testOrderedDictNumpyFeedFnBatchTwoWithOneEpoch(self):
    method testOrderedDictNumpyFeedFnLargeBatchWithSmallArrayAndMultipleEpochs (line 265) | def testOrderedDictNumpyFeedFnLargeBatchWithSmallArrayAndMultipleEpoch...
    method testFillArraySmall (line 283) | def testFillArraySmall(self):
    method testFillArrayLarge (line 293) | def testFillArrayLarge(self):
    method testFillArraySmallWithSpecifiedValue (line 303) | def testFillArraySmallWithSpecifiedValue(self):
    method testFillArrayLargeWithSpecifiedValue (line 314) | def testFillArrayLargeWithSpecifiedValue(self):
    method testPadIfNeededSmall (line 325) | def testPadIfNeededSmall(self):
    method testPadIfNeededLarge (line 335) | def testPadIfNeededLarge(self):
    method testPadIfNeededSmallWithSpecifiedValue (line 345) | def testPadIfNeededSmallWithSpecifiedValue(self):
    method testPadIfNeededLargeWithSpecifiedValue (line 356) | def testPadIfNeededLargeWithSpecifiedValue(self):
    method testPadIfNeededSmallWithSpecifiedNonNumericValue (line 367) | def testPadIfNeededSmallWithSpecifiedNonNumericValue(self):
    method testPadIfNeededLargeWithSpecifiedNonNumericValue (line 378) | def testPadIfNeededLargeWithSpecifiedNonNumericValue(self):

FILE: tensorflow_estimator/python/estimator/inputs/queues/feeding_queue_runner.py
  class _FeedingQueueRunner (line 25) | class _FeedingQueueRunner(tf.compat.v1.train.queue_runner.QueueRunner):
    method __init__ (line 28) | def __init__(self,
    method _run (line 76) | def _run(self, sess, enqueue_op, feed_fn, coord=None):
    method create_threads (line 124) | def create_threads(self, sess, coord=None, daemon=False, start=False):
    method _init_from_proto (line 176) | def _init_from_proto(self, queue_runner_def):
    method to_proto (line 181) | def to_proto(self):

FILE: tensorflow_estimator/python/estimator/inputs/queues/feeding_queue_runner_test.py
  function get_rows (line 37) | def get_rows(array, row_indices):
  class FeedingQueueRunnerTestCase (line 43) | class FeedingQueueRunnerTestCase(tf.test.TestCase):
    method testArrayFeeding (line 46) | def testArrayFeeding(self):
    method testArrayFeedingMultiThread (line 68) | def testArrayFeedingMultiThread(self):
    method testPandasFeeding (line 86) | def testPandasFeeding(self):
    method testPandasFeedingMultiThread (line 115) | def testPandasFeedingMultiThread(self):

FILE: tensorflow_estimator/python/estimator/keras_distribute_strategy_test.py
  function simple_sequential_model (line 37) | def simple_sequential_model():
  function simple_functional_model (line 45) | def simple_functional_model():
  function multi_inputs_multi_outputs_model (line 54) | def multi_inputs_multi_outputs_model():
  function get_ds_train_input_fn (line 82) | def get_ds_train_input_fn():
  function get_ds_test_input_fn (line 96) | def get_ds_test_input_fn():
  function get_multi_inputs_multi_outputs_data (line 110) | def get_multi_inputs_multi_outputs_data():
  class TestEstimatorDistributionStrategy (line 153) | class TestEstimatorDistributionStrategy(tf.test.TestCase,
    method setUp (line 156) | def setUp(self):
    method tearDown (line 165) | def tearDown(self):
    method test_train_functional_with_distribution_strategy (line 178) | def test_train_functional_with_distribution_strategy(self, distribution,
    method test_train_sequential_with_distribution_strategy (line 211) | def test_train_sequential_with_distribution_strategy(self, distribution,
    method test_multi_inputs_multi_outputs_with_input_fn_as_dict (line 242) | def test_multi_inputs_multi_outputs_with_input_fn_as_dict(self, distri...
    method do_test_multi_inputs_multi_outputs_with_input_fn (line 274) | def do_test_multi_inputs_multi_outputs_with_input_fn(self, distribution,
  function get_test_data (line 291) | def get_test_data(train_samples,

FILE: tensorflow_estimator/python/estimator/keras_lib.py
  class FormattedKeyError (line 40) | class FormattedKeyError(KeyError):
    method __init__ (line 50) | def __init__(self, message):
    method __str__ (line 53) | def __str__(self):
  function _cast_tensor_to_floatx (line 57) | def _cast_tensor_to_floatx(x):
  function _convert_tensor (line 65) | def _convert_tensor(x):
  function _any_weight_initialized (line 73) | def _any_weight_initialized(keras_model):
  function _convert_estimator_io_to_keras (line 94) | def _convert_estimator_io_to_keras(keras_model, features, labels):
  function _extract_sample_weight_tensors (line 187) | def _extract_sample_weight_tensors(features):
  function _clone_and_build_model (line 198) | def _clone_and_build_model(mode,
  function _convert_keras_metrics_to_estimator (line 253) | def _convert_keras_metrics_to_estimator(model, metric_names_map=None):
  function _create_keras_model_fn (line 294) | def _create_keras_model_fn(keras_model,
  function _save_first_checkpoint (line 436) | def _save_first_checkpoint(keras_model, custom_objects, config,
  function _get_file_from_google_storage (line 494) | def _get_file_from_google_storage(keras_model_path, model_dir):
  function model_to_estimator (line 532) | def model_to_estimator(keras_model=None,
  function _assert_valid_model (line 751) | def _assert_valid_model(model, custom_objects=None):
  function standardize_sample_weights (line 765) | def standardize_sample_weights(x_weight, output_names):

FILE: tensorflow_estimator/python/estimator/keras_premade_model_test.py
  function gen_input_fn (line 33) | def gen_input_fn(x, y=None, batch_size=32, num_epochs=10, shuffle=False):
  function get_resource_for_simple_model (line 45) | def get_resource_for_simple_model():
  function randomize_io_type (line 68) | def randomize_io_type(array, name):
  class KerasPremadeModelTest (line 76) | class KerasPremadeModelTest(tf.test.TestCase):
    method setUp (line 78) | def setUp(self):
    method tearDown (line 85) | def tearDown(self):
    method test_train_premade_linear_model_with_dense_features (line 93) | def test_train_premade_linear_model_with_dense_features(self):
    method test_train_premade_linear_model (line 127) | def test_train_premade_linear_model(self):
    method test_train_premade_widedeep_model_with_feature_layers (line 146) | def test_train_premade_widedeep_model_with_feature_layers(self):

FILE: tensorflow_estimator/python/estimator/keras_test.py
  function simple_sequential_model (line 55) | def simple_sequential_model():
  function simple_functional_model (line 63) | def simple_functional_model(activation='relu'):
  function simple_subclassed_model (line 72) | def simple_subclassed_model():
  function gen_input_fn (line 97) | def gen_input_fn(x, y=None, batch_size=128, num_epochs=1, shuffle=False):
  function get_multi_inputs_multi_outputs_data (line 109) | def get_multi_inputs_multi_outputs_data():
  function get_resource_for_simple_model (line 152) | def get_resource_for_simple_model(
  function randomize_io_type (line 203) | def randomize_io_type(array, name):
  function multi_inputs_multi_outputs_model (line 211) | def multi_inputs_multi_outputs_model():
  class MyHook (line 239) | class MyHook(tf.compat.v1.train.SessionRunHook):
    method begin (line 241) | def begin(self):
  class TestKerasEstimator (line 245) | class TestKerasEstimator(tf.test.TestCase, parameterized.TestCase):
    method setUp (line 247) | def setUp(self):
    method tearDown (line 254) | def tearDown(self):
    method test_train_keras_estimator (line 319) | def test_train_keras_estimator(self,
    method test_train_with_dense_features (line 357) | def test_train_with_dense_features(self):
    method DISABLED_test_train_with_dense_features_embedding (line 390) | def DISABLED_test_train_with_dense_features_embedding(self):
    method DISABLED_test_train_with_dense_features_v2 (line 431) | def DISABLED_test_train_with_dense_features_v2(self):
    method test_evaluate (line 471) | def test_evaluate(self):
    method test_evaluate_multi_io_model (line 512) | def test_evaluate_multi_io_model(self):
    method test_invalid_metric_names_map (line 593) | def test_invalid_metric_names_map(self):
    method test_predict (line 620) | def test_predict(self):
    method test_multi_inputs_multi_outputs_with_input_fn_as_dict (line 629) | def test_multi_inputs_multi_outputs_with_input_fn_as_dict(self):
    method test_multi_inputs_multi_outputs_with_input_fn_as_list (line 667) | def test_multi_inputs_multi_outputs_with_input_fn_as_list(self):
    method do_test_multi_inputs_multi_outputs_with_input_fn (line 696) | def do_test_multi_inputs_multi_outputs_with_input_fn(self, train_input...
    method test_init_from_file (line 708) | def test_init_from_file(self):
    method test_keras_model_init_error (line 733) | def test_keras_model_init_error(self):
    method test_invalid_ionames_error (line 747) | def test_invalid_ionames_error(self):
    method test_custom_objects (line 782) | def test_custom_objects(self):
    method test_tf_config (line 818) | def test_tf_config(self):
    method test_gpu_config (line 842) | def test_gpu_config(self):
    method test_with_empty_config (line 859) | def test_with_empty_config(self):
    method test_with_empty_config_and_empty_model_dir (line 887) | def test_with_empty_config_and_empty_model_dir(self):
    method test_with_conflicting_model_dir_and_config (line 901) | def test_with_conflicting_model_dir_and_config(self):
    method test_pretrained_weights (line 917) | def test_pretrained_weights(self):
    method assert_increasing_global_step (line 936) | def assert_increasing_global_step(self, optimizer):
    method test_model_fn_increments_global_step_tf_optimizer (line 957) | def test_model_fn_increments_global_step_tf_optimizer(self):
    method test_model_fn_increments_global_step_keras_optimizer (line 962) | def test_model_fn_increments_global_step_keras_optimizer(self):
    method test_export_keras_estimator (line 968) | def test_export_keras_estimator(self, checkpoint_format):
    method test_export_keras_estimator_custom_signatures (line 1015) | def test_export_keras_estimator_custom_signatures(self, checkpoint_for...
    method test_export_keras_estimator_unknown_signatures (line 1061) | def test_export_keras_estimator_unknown_signatures(self, checkpoint_fo...
    method test_export_subclassed_model_retains_model_state (line 1101) | def test_export_subclassed_model_retains_model_state(self):
    method test_warm_start_from_keras_ckpt (line 1125) | def test_warm_start_from_keras_ckpt(self):
    method test_sample_weights (line 1149) | def test_sample_weights(self):
    method test_model_to_estimator_with_rnn (line 1201) | def test_model_to_estimator_with_rnn(self, layer):
  function get_test_data (line 1223) | def get_test_data(train_samples,

FILE: tensorflow_estimator/python/estimator/model_fn.py
  class EstimatorSpec (line 37) | class EstimatorSpec(
    method __new__ (line 48) | def __new__(cls,
    method _replace (line 187) | def _replace(self, **kwds):
  class _TPUEstimatorSpec (line 196) | class _TPUEstimatorSpec(
    method __new__ (line 209) | def __new__(cls,
    method as_estimator_spec (line 244) | def as_estimator_spec(self):
  function _validate_estimator_spec_train_op (line 274) | def _validate_estimator_spec_train_op(train_op, mode):
  function _validate_estimator_spec_loss (line 307) | def _validate_estimator_spec_loss(loss, mode):
  function _validate_estimator_spec_predictions (line 343) | def _validate_estimator_spec_predictions(predictions, mode):
  function _validate_estimator_spec_export_outputs (line 389) | def _validate_estimator_spec_export_outputs(export_outputs, predictions,...
  function _validate_estimator_spec_hooks (line 420) | def _validate_estimator_spec_hooks(hooks):
  function _validate_eval_metric_ops (line 442) | def _validate_eval_metric_ops(eval_metric_ops):
  function _update_variable_collection (line 519) | def _update_variable_collection(collection_name, vars_to_add):
  function _validate_scaffold (line 528) | def _validate_scaffold(scaffold):
  function _check_is_tensor_or_operation (line 550) | def _check_is_tensor_or_operation(x, name):
  function _check_is_tensor (line 556) | def _check_is_tensor(x, tensor_name):
  function call_logit_fn (line 564) | def call_logit_fn(logit_fn, features, mode, params, config):
  function verify_model_fn_args (line 615) | def verify_model_fn_args(model_fn, params):

FILE: tensorflow_estimator/python/estimator/model_fn_test.py
  class _FakeHook (line 28) | class _FakeHook(tf.compat.v1.train.SessionRunHook):
  class _InvalidHook (line 32) | class _InvalidHook(object):
  class _InvalidScaffold (line 36) | class _InvalidScaffold(object):
  class EstimatorSpecTrainTest (line 40) | class EstimatorSpecTrainTest(tf.test.TestCase):
    method testRequiredArgumentsSet (line 43) | def testRequiredArgumentsSet(self):
    method testAllArgumentsSet (line 49) | def testAllArgumentsSet(self):
    method testLossNumber (line 75) | def testLossNumber(self):
    method testLoss1DTensor (line 82) | def testLoss1DTensor(self):
    method testLossMissing (line 88) | def testLossMissing(self):
    method testLossNotScalar (line 93) | def testLossNotScalar(self):
    method testLossSparseTensor (line 101) | def testLossSparseTensor(self):
    method testLossFromDifferentGraph (line 108) | def testLossFromDifferentGraph(self):
    method testTrainOpMissing (line 117) | def testTrainOpMissing(self):
    method testTrainOpNotOperationAndTensor (line 122) | def testTrainOpNotOperationAndTensor(self):
    method testTrainOpFromDifferentGraph (line 131) | def testTrainOpFromDifferentGraph(self):
    method testTrainingChiefHookInvalid (line 140) | def testTrainingChiefHookInvalid(self):
    method testTrainingHookInvalid (line 150) | def testTrainingHookInvalid(self):
    method testScaffoldInvalid (line 160) | def testScaffoldInvalid(self):
    method testReturnDefaultScaffold (line 170) | def testReturnDefaultScaffold(self):
  class EstimatorSpecEvalTest (line 177) | class EstimatorSpecEvalTest(tf.test.TestCase):
    method testRequiredArgumentsSet (line 180) | def testRequiredArgumentsSet(self):
    method testAllArgumentsSet (line 187) | def testAllArgumentsSet(self):
    method testEvaluationHookInvalid (line 212) | def testEvaluationHookInvalid(self):
    method testTupleMetric (line 221) | def testTupleMetric(self):
    method testLoss1DTensor (line 232) | def testLoss1DTensor(self):
    method testLossNumber (line 239) | def testLossNumber(self):
    method testLossMissing (line 246) | def testLossMissing(self):
    method testLossNotScalar (line 252) | def testLossNotScalar(self):
    method testLossSparseTensor (line 259) | def testLossSparseTensor(self):
    method testLossFromDifferentGraph (line 268) | def testLossFromDifferentGraph(self):
    method testReplaceRaisesConstructorChecks (line 279) | def testReplaceRaisesConstructorChecks(self):
    method testReplaceDoesReplace (line 287) | def testReplaceDoesReplace(self):
    method testReplaceNotAllowModeChange (line 295) | def testReplaceNotAllowModeChange(self):
    method testPredictionsMissingIsOkay (line 305) | def testPredictionsMissingIsOkay(self):
    method testPredictionsTensor (line 309) | def testPredictionsTensor(self):
    method testPredictionsNumber (line 315) | def testPredictionsNumber(self):
    method testPredictionsSparseTensor (line 324) | def testPredictionsSparseTensor(self):
    method testPredictionsFromDifferentGraph (line 336) | def testPredictionsFromDifferentGraph(self):
    method testEvalMetricOpsNoDict (line 345) | def testEvalMetricOpsNoDict(self):
    method testEvalMetricOpsNoTuple (line 355) | def testEvalMetricOpsNoTuple(self):
    method testEvalMetricOpsFromDifferentGraphWithMetricTuple (line 368) | def testEvalMetricOpsFromDifferentGraphWithMetricTuple(self):
    method testEvalMetricOpsFromDifferentGraphWithMetricObject (line 381) | def testEvalMetricOpsFromDifferentGraphWithMetricObject(self):
    method testEvalMetricOpsWithoutUpdates (line 396) | def testEvalMetricOpsWithoutUpdates(self):
    method testMetricVariablesAddedToCollections (line 408) | def testMetricVariablesAddedToCollections(self):
  class EstimatorSpecInferTest (line 437) | class EstimatorSpecInferTest(tf.test.TestCase):
    method testRequiredArgumentsSet (line 440) | def testRequiredArgumentsSet(self):
    method testAllArgumentsSet (line 446) | def testAllArgumentsSet(self):
    method testPredictionHookInvalid (line 472) | def testPredictionHookInvalid(self):
    method testPredictionsMissing (line 481) | def testPredictionsMissing(self):
    method testPredictionsTensor (line 486) | def testPredictionsTensor(self):
    method testPredictionsNumber (line 491) | def testPredictionsNumber(self):
    method testPredictionsSparseTensor (line 498) | def testPredictionsSparseTensor(self):
    method testExportOutputsNoDict (line 509) | def testExportOutputsNoDict(self):
    method testExportOutputsValueNotExportOutput (line 520) | def testExportOutputsValueNotExportOutput(self):
    method testExportOutputsSingleheadMissingDefault (line 531) | def testExportOutputsSingleheadMissingDefault(self):
    method testExportOutputsMultiheadWithDefault (line 549) | def testExportOutputsMultiheadWithDefault(self):
    method testExportOutputsMultiheadMissingDefault (line 569) | def testExportOutputsMultiheadMissingDefault(self):
    method testDefaultExportOutputCreated (line 590) | def testDefaultExportOutputCreated(self):
    method testDefaultExportOutputCreatedDict (line 596) | def testDefaultExportOutputCreatedDict(self):
    method _assertDefaultExportOutputForPredictions (line 602) | def _assertDefaultExportOutputForPredictions(self, predictions):
  class LogitFnTest (line 612) | class LogitFnTest(tf.test.TestCase):
    method test_simple_call_logit_fn (line 614) | def test_simple_call_logit_fn(self):
    method test_simple_call_multi_logit_fn (line 629) | def test_simple_call_multi_logit_fn(self):
    method test_invalid_logit_fn_results (line 642) | def test_invalid_logit_fn_results(self):
    method test_invalid_logit_fn_results_dict (line 658) | def test_invalid_logit_fn_results_dict(self):

FILE: tensorflow_estimator/python/estimator/object_checkpointing_test.py
  class SubclassedModel (line 37) | class SubclassedModel(tf_keras.models.Model):
    method __init__ (line 39) | def __init__(self):
    method call (line 44) | def call(self, inputs):
  function _serving_input_receiver_fn (line 48) | def _serving_input_receiver_fn():
  class ObjectCheckpointingTest (line 55) | class ObjectCheckpointingTest(tf.test.TestCase):
    method _make_estimator (line 57) | def _make_estimator(self, model_dir):
    method testTwoWayCompatibility (line 101) | def testTwoWayCompatibility(self):
    method testSavedModelExport (line 136) | def testSavedModelExport(self):

FILE: tensorflow_estimator/python/estimator/run_config.py
  function _get_session_master (line 63) | def _get_session_master(cluster_spec, task_type, task_id, tf_config):
  function _get_eval_session_master (line 106) | def _get_eval_session_master(task_type, tf_config):
  function _count_ps (line 114) | def _count_ps(cluster_spec):
  function _count_worker (line 123) | def _count_worker(cluster_spec, chief_task_type):
  function _validate_service (line 133) | def _validate_service(service):
  function _validate_task_type_and_task_id (line 142) | def _validate_task_type_and_task_id(cluster_spec, task_env, chief_task_t...
  function _get_global_id_in_cluster (line 192) | def _get_global_id_in_cluster(cluster_spec, task_type, task_id,
  function _validate_save_ckpt_with_replaced_keys (line 220) | def _validate_save_ckpt_with_replaced_keys(new_copy, replaced_keys):
  function _validate_properties (line 241) | def _validate_properties(run_config):
  function get_default_session_config (line 324) | def get_default_session_config():
  class TaskType (line 335) | class TaskType(object):
  class RunConfig (line 344) | class RunConfig(object):
    method __init__ (line 347) | def __init__(self,
    method _verify_strategy_compatibility (line 586) | def _verify_strategy_compatibility(self, train_distribute, eval_distri...
    method _maybe_overwrite_session_config_for_distributed_training (line 595) | def _maybe_overwrite_session_config_for_distributed_training(self):
    method _get_default_session_config_distributed (line 609) | def _get_default_session_config_distributed(self):
    method _init_distributed_setting_from_environment_var (line 639) | def _init_distributed_setting_from_environment_var(self, tf_config):
    method _init_distributed_setting_from_environment_var_with_master (line 698) | def _init_distributed_setting_from_environment_var_with_master(
    method cluster_spec (line 734) | def cluster_spec(self):
    method device_fn (line 738) | def device_fn(self):
    method evaluation_master (line 748) | def evaluation_master(self):
    method is_chief (line 752) | def is_chief(self):
    method master (line 756) | def master(self):
    method num_ps_replicas (line 760) | def num_ps_replicas(self):
    method num_worker_replicas (line 764) | def num_worker_replicas(self):
    method task_id (line 768) | def task_id(self):
    method global_id_in_cluster (line 772) | def global_id_in_cluster(self):
    method experimental_max_worker_delay_secs (line 812) | def experimental_max_worker_delay_secs(self):
    method task_type (line 816) | def task_type(self):
    method tf_random_seed (line 820) | def tf_random_seed(self):
    method save_summary_steps (line 824) | def save_summary_steps(self):
    method save_checkpoints_secs (line 828) | def save_checkpoints_secs(self):
    method session_config (line 832) | def session_config(self):
    method save_checkpoints_steps (line 836) | def save_checkpoints_steps(self):
    method checkpoint_save_graph_def (line 840) | def checkpoint_save_graph_def(self):
    method keep_checkpoint_max (line 844) | def keep_checkpoint_max(self):
    method session_creation_timeout_secs (line 848) | def session_creation_timeout_secs(self):
    method keep_checkpoint_every_n_hours (line 852) | def keep_checkpoint_every_n_hours(self):
    method log_step_count_steps (line 856) | def log_step_count_steps(self):
    method model_dir (line 860) | def model_dir(self):
    method service (line 864) | def service(self):
    method train_distribute (line 869) | def train_distribute(self):
    method eval_distribute (line 874) | def eval_distribute(self):
    method protocol (line 879) | def protocol(self):
    method replace (line 883) | def replace(self, **kwargs):
    method _replace (line 924) | def _replace(config, allowed_properties_list=None, **kwargs):
  function _get_model_dir (line 960) | def _get_model_dir(tf_config, model_dir):
  function path_to_str (line 989) | def path_to_str(path):

FILE: tensorflow_estimator/python/estimator/run_config_test.py
  function _create_run_config_with_cluster_spec (line 70) | def _create_run_config_with_cluster_spec(tf_config, **kwargs):
  class RunConfigTest (line 76) | class RunConfigTest(tf.test.TestCase):
    method test_default_property_values (line 78) | def test_default_property_values(self):
    method test_model_dir (line 94) | def test_model_dir(self):
    method test_replace_with_allowed_properties (line 101) | def test_replace_with_allowed_properties(self):
    method test_replace_none_value (line 125) | def test_replace_none_value(self):
    method test_replace_with_disallowallowed_properties (line 146) | def test_replace_with_disallowallowed_properties(self):
    method test_replace (line 154) | def test_replace(self):
    method test_replace_invalid_values (line 166) | def test_replace_invalid_values(self):
    method test_init_with_allowed_properties (line 194) | def test_init_with_allowed_properties(self):
    method test_init_none_value (line 216) | def test_init_none_value(self):
    method test_init_invalid_values (line 237) | def test_init_invalid_values(self):
    method test_incompatible_train_strategy (line 260) | def test_incompatible_train_strategy(self):
    method test_incompatible_eval_strategy (line 269) | def test_incompatible_eval_strategy(self):
  class RunConfigDistributedSettingTest (line 278) | class RunConfigDistributedSettingTest(tf.test.TestCase):
    method _assert_distributed_properties (line 280) | def _assert_distributed_properties(
    method test_default_values (line 295) | def test_default_values(self):
    method test_tf_config_for_local (line 307) | def test_tf_config_for_local(self):
    method test_session_master_for_local (line 323) | def test_session_master_for_local(self):
    method test_eval_session_master_for_local (line 336) | def test_eval_session_master_for_local(self):
    method test_invalid_task_type_for_local (line 349) | def test_invalid_task_type_for_local(self):
    method test_invalid_task_index_for_local (line 354) | def test_invalid_task_index_for_local(self):
    method test_chief_tf_config (line 359) | def test_chief_tf_config(self):
    method test_session_master_from_single_node_tf_config (line 382) | def test_session_master_from_single_node_tf_config(self):
    method test_session_master_from_multiple_nodes_tf_config (line 396) | def test_session_master_from_multiple_nodes_tf_config(self):
    method test_fail_with_multiple_chief_nodes (line 411) | def test_fail_with_multiple_chief_nodes(self):
    method test_fail_with_missing_chief_node (line 421) | def test_fail_with_missing_chief_node(self):
    method test_single_chief_node (line 430) | def test_single_chief_node(self):
    method test_fail_with_missing_task_type_for_distributed (line 451) | def test_fail_with_missing_task_type_for_distributed(self):
    method test_fail_with_missing_task_index_for_distributed (line 460) | def test_fail_with_missing_task_index_for_distributed(self):
    method test_fail_with_index_is_too_large (line 472) | def test_fail_with_index_is_too_large(self):
    method test_fail_with_invalid_task_index (line 485) | def test_fail_with_invalid_task_index(self):
    method test_fail_with_invalid_task_type (line 498) | def test_fail_with_invalid_task_type(self):
    method test_worker_tf_config (line 511) | def test_worker_tf_config(self):
    method test_ps_tf_config (line 534) | def test_ps_tf_config(self):
    method test_evaluator_tf_config (line 557) | def test_evaluator_tf_config(self):
    method test_eval_master_for_evaluator (line 582) | def test_eval_master_for_evaluator(self):
    method test_fail_with_invalid_task_index_for_evaluator (line 598) | def test_fail_with_invalid_task_index_for_evaluator(self):
    method test_global_id_in_cluster_for_chief (line 611) | def test_global_id_in_cluster_for_chief(self):
    method test_global_id_in_cluster_for_worker (line 626) | def test_global_id_in_cluster_for_worker(self):
    method test_global_id_in_cluster_for_ps (line 641) | def test_global_id_in_cluster_for_ps(self):
    method test_global_id_in_cluster_for_multipe_worker_types (line 656) | def test_global_id_in_cluster_for_multipe_worker_types(self):
  class RunConfigDistributedSettingWithMasterTest (line 677) | class RunConfigDistributedSettingWithMasterTest(tf.test.TestCase):
    method _assert_distributed_properties (line 679) | def _assert_distributed_properties(
    method test_invalid_task_type_for_local (line 694) | def test_invalid_task_type_for_local(self):
    method test_master_node (line 699) | def test_master_node(self):
    method test_session_master_in_single_node_tf_config (line 722) | def test_session_master_in_single_node_tf_config(self):
    method test_session_master_in_multiple_nodes_tf_config (line 736) | def test_session_master_in_multiple_nodes_tf_config(self):
    method test_fail_with_multiple_master_nodes (line 751) | def test_fail_with_multiple_master_nodes(self):
    method test_single_master_node (line 761) | def test_single_master_node(self):
    method test_fail_with_missing_task_type_for_distributed (line 782) | def test_fail_with_missing_task_type_for_distributed(self):
    method test_fail_with_missing_task_index_for_distributed (line 791) | def test_fail_with_missing_task_index_for_distributed(self):
    method test_fail_with_index_is_too_large (line 803) | def test_fail_with_index_is_too_large(self):
    method test_fail_with_invalid_task_index (line 816) | def test_fail_with_invalid_task_index(self):
    method test_fail_with_invalid_task_type (line 829) | def test_fail_with_invalid_task_type(self):
    method test_worker_tf_config (line 842) | def test_worker_tf_config(self):
    method test_ps_tf_config (line 865) | def test_ps_tf_config(self):
    method test_fail_with_evaluator (line 888) | def test_fail_with_evaluator(self):
    method test_fail_with_chief (line 904) | def test_fail_with_chief(self):
    method test_global_id_in_cluster_for_master (line 920) | def test_global_id_in_cluster_for_master(self):
    method test_global_id_in_cluster_for_worker (line 935) | def test_global_id_in_cluster_for_worker(self):
    method test_global_id_in_cluster_for_ps (line 950) | def test_global_id_in_cluster_for_ps(self):
    method test_global_id_in_cluster_for_multipe_worker_types (line 965) | def test_global_id_in_cluster_for_multipe_worker_types(self):
  class RunConfigSaveCheckpointsTest (line 986) | class RunConfigSaveCheckpointsTest(tf.test.TestCase):
    method test_save_checkpoint (line 988) | def test_save_checkpoint(self):
    method test_save_checkpoint_both_steps_and_secs_are_not_none (line 1003) | def test_save_checkpoint_both_steps_and_secs_are_not_none(self):
    method test_save_checkpoint_both_steps_and_secs_are_none (line 1013) | def test_save_checkpoint_both_steps_and_secs_are_none(self):
    method test_save_checkpoint_flip_secs_to_none (line 1020) | def test_save_checkpoint_flip_secs_to_none(self):
    method test_save_checkpoint_flip_steps_to_none (line 1026) | def test_save_checkpoint_flip_steps_to_none(self):
  class RunConfigServiceKeyTest (line 1034) | class RunConfigServiceKeyTest(tf.test.TestCase):
    method test_arbitrary_key_value_pairs (line 1036) | def test_arbitrary_key_value_pairs(self):
    method test_missing_service_key (line 1050) | def test_missing_service_key(self):
    method test_fail_with_non_dict (line 1057) | def test_fail_with_non_dict(self):
  class RunConfigModelDirTest (line 1065) | class RunConfigModelDirTest(tf.test.TestCase):
    method test_default (line 1067) | def test_default(self):
    method test_model_dir_in_constructor (line 1071) | def test_model_dir_in_constructor(self):
    method test_model_dir_in_tf_config (line 1075) | def test_model_dir_in_tf_config(self):
    method test_model_dir_both_set_in_both_constructor_and_tf_config (line 1082) | def test_model_dir_both_set_in_both_constructor_and_tf_config(self):
    method test_model_dir_different_in_both_constructor_and_tf_config (line 1089) | def test_model_dir_different_in_both_constructor_and_tf_config(self):
    method test_fail_with_empty_string_in_constructor (line 1095) | def test_fail_with_empty_string_in_constructor(self):
    method test_fail_with_empty_string_in_tf_config (line 1099) | def test_fail_with_empty_string_in_tf_config(self):
  class RunConfigSessionConfigTest (line 1105) | class RunConfigSessionConfigTest(tf.test.TestCase):
    method _assert_equal_session_config (line 1107) | def _assert_equal_session_config(self, session_config,
    method test_master_session_config (line 1119) | def test_master_session_config(self):
    method test_chief_session_config (line 1135) | def test_chief_session_config(self):
    method test_worker_session_config (line 1151) | def test_worker_session_config(self):
    method test_ps_session_config (line 1167) | def test_ps_session_config(self):
    method test_evaluator_session_config (line 1184) | def test_evaluator_session_config(self):
    method test_other_type_session_config (line 1199) | def test_other_type_session_config(self):

FILE: tensorflow_estimator/python/estimator/tf_estimator_doctest.py
  function find_modules (line 54) | def find_modules():
  function filter_on_submodules (line 69) | def filter_on_submodules(all_modules, submodule):
  function get_module_and_inject_docstring (line 90) | def get_module_and_inject_docstring(file_path):
  class TfTestCase (line 113) | class TfTestCase(tf.test.TestCase):
    method set_up (line 115) | def set_up(self, test):
    method tear_down (line 118) | def tear_down(self, test):
  class CustomOutputChecker (line 122) | class CustomOutputChecker(doctest.OutputChecker):
    method check_output (line 130) | def check_output(self, want, got, optionflags):
    method output_difference (line 144) | def output_difference(self, example, got, optionflags):
  function load_tests (line 150) | def load_tests(unused_loader, tests, unused_ignore):

FILE: tensorflow_estimator/python/estimator/tools/analytics.py
  function track_usage (line 21) | def track_usage(tool_id, tags):
  function track_numerical_issues (line 31) | def track_numerical_issues(exc_info):

FILE: tensorflow_estimator/python/estimator/tools/checkpoint_converter.py
  function _add_new_variable (line 119) | def _add_new_variable(initial_value, var_name_v2, var_name_v1, var_map,
  function _add_opt_variable (line 127) | def _add_opt_variable(opt_name_v2, var_name_v1, idx, suffix_v2, reader, ...
  function _convert_variables_in_ckpt (line 135) | def _convert_variables_in_ckpt(opt_name_v1, reader, variable_names, var_...
  function _convert_hyper_params_in_graph (line 198) | def _convert_hyper_params_in_graph(graph_from_path, opt_name_v1, var_map,
  function convert_checkpoint (line 247) | def convert_checkpoint(estimator_type, source_checkpoint, source_graph,
  function main (line 334) | def main(_):

FILE: tensorflow_estimator/python/estimator/tools/checkpoint_converter_test.py
  class DNNCheckpointConverterTest (line 36) | class DNNCheckpointConverterTest(tf.test.TestCase):
    method setUp (line 38) | def setUp(self):
    method tearDown (line 42) | def tearDown(self):
    method _test_ckpt_converter (line 50) | def _test_ckpt_converter(self, train_input_fn, eval_input_fn,
    method _create_input_fn (line 99) | def _create_input_fn(self, label_dimension, batch_size):
    method _test_ckpt_converter_with_an_optimizer (line 117) | def _test_ckpt_converter_with_an_optimizer(self, opt):
    method test_ckpt_converter_with_adagrad (line 133) | def test_ckpt_converter_with_adagrad(self):
    method test_ckpt_converter_with_rmsprop (line 137) | def test_ckpt_converter_with_rmsprop(self):
    method test_ckpt_converter_with_ftrl (line 141) | def test_ckpt_converter_with_ftrl(self):
    method test_ckpt_converter_with_adam (line 145) | def test_ckpt_converter_with_adam(self):
    method test_ckpt_converter_with_sgd (line 149) | def test_ckpt_converter_with_sgd(self):
  class LinearCheckpointConverterTest (line 154) | class LinearCheckpointConverterTest(tf.test.TestCase):
    method setUp (line 156) | def setUp(self):
    method tearDown (line 160) | def tearDown(self):
    method _test_ckpt_converter (line 168) | def _test_ckpt_converter(self, train_input_fn, eval_input_fn,
    method _create_input_fn (line 215) | def _create_input_fn(self, label_dimension, batch_size):
    method _test_ckpt_converter_with_an_optimizer (line 233) | def _test_ckpt_converter_with_an_optimizer(self, opt):
    method test_ckpt_converter_with_adagrad (line 249) | def test_ckpt_converter_with_adagrad(self):
    method test_ckpt_converter_with_rmsprop (line 253) | def test_ckpt_converter_with_rmsprop(self):
    method test_ckpt_converter_with_ftrl (line 257) | def test_ckpt_converter_with_ftrl(self):
    method test_ckpt_converter_with_adam (line 261) | def test_ckpt_converter_with_adam(self):
    method test_ckpt_converter_with_sgd (line 265) | def test_ckpt_converter_with_sgd(self):
  class DNNLinearCombinedCheckpointConverterTest (line 270) | class DNNLinearCombinedCheckpointConverterTest(tf.test.TestCase):
    method setUp (line 272) | def setUp(self):
    method tearDown (line 276) | def tearDown(self):
    method _test_ckpt_converter (line 284) | def _test_ckpt_converter(self, train_input_fn, eval_input_fn,
    method _create_input_fn (line 342) | def _create_input_fn(self, label_dimension, batch_size):
    method _test_ckpt_converter_with_an_optimizer (line 360) | def _test_ckpt_converter_with_an_optimizer(self, dnn_opt, linear_opt):
    method test_ckpt_converter_with_adagrad (line 377) | def test_ckpt_converter_with_adagrad(self):
    method test_ckpt_converter_with_rmsprop (line 381) | def test_ckpt_converter_with_rmsprop(self):
    method test_ckpt_converter_with_ftrl (line 385) | def test_ckpt_converter_with_ftrl(self):
    method test_ckpt_converter_with_adam (line 389) | def test_ckpt_converter_with_adam(self):
    method test_ckpt_converter_with_sgd (line 393) | def test_ckpt_converter_with_sgd(self):

FILE: tensorflow_estimator/python/estimator/tpu/_tpu_estimator_embedding.py
  function _get_table_name_from_embedding_var_name (line 70) | def _get_table_name_from_embedding_var_name(embedding_var_name):
  function _get_embedding_var_name_from_table_name (line 74) | def _get_embedding_var_name_from_table_name(table_name):
  function _get_embedding_variable_name (line 78) | def _get_embedding_variable_name(scope_name, var_name):
  function _get_slot_variable_names (line 84) | def _get_slot_variable_names(scope_name, var_name, optimization_paramete...
  function get_full_variable_names (line 123) | def get_full_variable_names(graph,
  function get_configs_from_feature_columns (line 149) | def get_configs_from_feature_columns(feature_columns):
  class EmbeddingConfigSpec (line 202) | class EmbeddingConfigSpec(
    method __new__ (line 258) | def __new__(cls,
  class EmbeddingConfig (line 389) | class EmbeddingConfig(object):
    method __init__ (line 397) | def __init__(self, embedding_config_spec, train_batch_size, eval_batch...
    method get_grad_multiplier (line 423) | def get_grad_multiplier(self):
    method has_embedding_tables (line 429) | def has_embedding_tables(self):
    method _create_tpu_embedding (line 432) | def _create_tpu_embedding(self, mode):
    method get_tpu_embedding (line 478) | def get_tpu_embedding(self, mode):
  function _maybe_dense_to_sparse (line 485) | def _maybe_dense_to_sparse(tensor):
  function split_inputs (line 496) | def split_inputs(ctx, features, labels, num_cores_per_batch=1):
  function _split_tensor (line 576) | def _split_tensor(tensor, num_splits):
  function _get_sparse_feature_from_feature (line 591) | def _get_sparse_feature_from_feature(feature_key, features):
  function _get_weights_from_features (line 605) | def _get_weights_from_features(weight_key_name, features):
  function get_tpu_embedding_columns (line 627) | def get_tpu_embedding_columns(feature_columns):

FILE: tensorflow_estimator/python/estimator/tpu/autotuning_iterations_per_loop_test.py
  class IterationsPerLoopParsingTest (line 29) | class IterationsPerLoopParsingTest(tf.test.TestCase):
    method _parse_and_validate_iterations_per_loop (line 31) | def _parse_and_validate_iterations_per_loop(self, value, expected_value,
    method _parse_and_validate_invalid_iterations_per_loop (line 38) | def _parse_and_validate_invalid_iterations_per_loop(self, value):
    method test_parsing_iterations_per_loop (line 45) | def test_parsing_iterations_per_loop(self):
  class IterationPredictorTest (line 81) | class IterationPredictorTest(tf.test.TestCase):
    method setUp (line 83) | def setUp(self):
    method test_empty (line 87) | def test_empty(self):
    method test_reset (line 92) | def test_reset(self):
    method test_invalid_update (line 105) | def test_invalid_update(self):
    method test_zero_mean (line 117) | def test_zero_mean(self):
    method test_diff_less_than_percentage (line 125) | def test_diff_less_than_percentage(self):
    method test_mean_runtime_secs (line 146) | def test_mean_runtime_secs(self):
    method test_mean_step_time_secs (line 161) | def test_mean_step_time_secs(self):
    method _test_std_step_time_secs (line 177) | def _test_std_step_time_secs(self):
    method test_buffer_capacity (line 188) | def test_buffer_capacity(self):
    method test_partial_wheel (line 204) | def test_partial_wheel(self):
    method test_update_convergence (line 223) | def test_update_convergence(self):
  class TPUStopAtStepHookTest (line 241) | class TPUStopAtStepHookTest(tf.test.TestCase):
    method test_invalid_parameters_on_construction (line 243) | def test_invalid_parameters_on_construction(self):
    method _validate_hook_life_cycle (line 271) | def _validate_hook_life_cycle(self, iterations_per_loop_counter, num_s...
    method test_hook_life_cycle (line 352) | def test_hook_life_cycle(self):
    method _validate_initialization (line 359) | def _validate_initialization(self, iterations_per_loop_counter, num_st...
    method test_initialization (line 384) | def test_initialization(self):

FILE: tensorflow_estimator/python/estimator/tpu/error_handling.py
  class ErrorRendezvous (line 39) | class ErrorRendezvous(object):
    method __init__ (line 54) | def __init__(self, num_sources):
    method record_error (line 60) | def record_error(self, source, exc_info, session=None):
    method record_done (line 107) | def record_done(self, source):
    method catch_errors (line 120) | def catch_errors(self, source, session=None):
    method raise_errors (line 127) | def raise_errors(self, timeout_sec=0):

FILE: tensorflow_estimator/python/estimator/tpu/error_handling_test.py
  class ErrorHandlingTest (line 25) | class ErrorHandlingTest(tf.test.TestCase):
    method catch_and_raise (line 27) | def catch_and_raise(self, error):
    method testInterestingError (line 33) | def testInterestingError(self):
    method testIgnoredError (line 37) | def testIgnoredError(self):

FILE: tensorflow_estimator/python/estimator/tpu/iteration_count_estimator.py
  class IterationCountEstimator (line 35) | class IterationCountEstimator(object):
    method __init__ (line 58) | def __init__(self, capacity=20):
    method _reset (line 70) | def _reset(self, capacity=20):
    method _mean_runtime_secs (line 83) | def _mean_runtime_secs(self):
    method _mean_step_time_secs (line 86) | def _mean_step_time_secs(self):
    method _std_step_time_secs (line 89) | def _std_step_time_secs(self):
    method _diff_less_than_percentage (line 92) | def _diff_less_than_percentage(self, actual, target, percentage):
    method _is_step_time_stable (line 113) | def _is_step_time_stable(self):
    method update (line 125) | def update(self, runtime_secs, count):
    method get (line 152) | def get(self, total_secs):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_config.py
  class InputPipelineConfig (line 37) | class InputPipelineConfig(object):
  class TPUConfig (line 55) | class TPUConfig(
    method __new__ (line 158) | def __new__(cls,
  class RunConfig (line 238) | class RunConfig(run_config_lib.RunConfig):
    method __init__ (line 241) | def __init__(self,
    method _maybe_overwrite_session_config_for_distributed_training (line 309) | def _maybe_overwrite_session_config_for_distributed_training(self):
    method evaluation_master (line 316) | def evaluation_master(self):
    method master (line 320) | def master(self):
    method tpu_config (line 324) | def tpu_config(self):
    method cluster (line 328) | def cluster(self):
    method replace (line 331) | def replace(self, **kwargs):
  function _get_tpu_job_name_from_tf_config (line 341) | def _get_tpu_job_name_from_tf_config():

FILE: tensorflow_estimator/python/estimator/tpu/tpu_config_test.py
  function _set_tf_config_env_variable (line 28) | def _set_tf_config_env_variable(tf_config):
  class TPURunConfigTest (line 33) | class TPURunConfigTest(tf.test.TestCase):
    method test_no_session_config_set_in_local_case (line 35) | def test_no_session_config_set_in_local_case(self):
    method test_no_session_config_overwrite_in_local_case (line 39) | def test_no_session_config_overwrite_in_local_case(self):
    method test_no_session_config_set_with_cluster_spec (line 44) | def test_no_session_config_set_with_cluster_spec(self):
    method test_no_session_config_overwrite_with_cluster_spec (line 59) | def test_no_session_config_overwrite_with_cluster_spec(self):
    method test_fail_with_invalid_num_shards (line 75) | def test_fail_with_invalid_num_shards(self):
    method _validate_invalid_iterations_per_loop (line 80) | def _validate_invalid_iterations_per_loop(self, iterations_per_loop):
    method test_fail_with_iterations_per_loop (line 86) | def test_fail_with_iterations_per_loop(self):
    method test_fail_with_invalid_num_cores_per_replica (line 93) | def test_fail_with_invalid_num_cores_per_replica(self):
    method _evaluate_iterations_per_loop_in_seconds (line 99) | def _evaluate_iterations_per_loop_in_seconds(self, value, expected_value,
    method test_valid_iterations_per_loop (line 109) | def test_valid_iterations_per_loop(self):
  class TPURunConfigMasterTest (line 117) | class TPURunConfigMasterTest(tf.test.TestCase):
    method test_default_values (line 119) | def test_default_values(self):
    method test_user_provided_master_and_evaluation_master (line 124) | def test_user_provided_master_and_evaluation_master(self):
    method test_evaluation_master_defaults_to_master (line 130) | def test_evaluation_master_defaults_to_master(self):
    method test_tf_config (line 135) | def test_tf_config(self):
    method test_evaluation_master_defaults_to_master_in_tf_config (line 145) | def test_evaluation_master_defaults_to_master_in_tf_config(self):
    method test_respect_evaluation_master_in_tf_config (line 154) | def test_respect_evaluation_master_in_tf_config(self):
    method test_user_overwrites_tf_config (line 168) | def test_user_overwrites_tf_config(self):
    method test_user_overwrites_master_in_tf_config (line 179) | def test_user_overwrites_master_in_tf_config(self):
  class TPUJobNameTest (line 190) | class TPUJobNameTest(tf.test.TestCase):
    method test_default_name (line 192) | def test_default_name(self):
    method test_with_tf_config (line 196) | def test_with_tf_config(self):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_context.py
  class TPUContext (line 48) | class TPUContext(object):
    method __init__ (line 87) | def __init__(self,
    method current_input_fn_deployment (line 99) | def current_input_fn_deployment(self):
    method num_replicas (line 151) | def num_replicas(self):
    method num_hosts (line 163) | def num_hosts(self):
    method current_host (line 168) | def current_host(self):
    method num_of_replicas_per_host (line 185) | def num_of_replicas_per_host(self):
    method device_assignment (line 193) | def device_assignment(self):
    method device_for_replica (line 204) | def device_for_replica(self, replica_id):
    method tpu_host_placement_function (line 221) | def tpu_host_placement_function(self):
  class _InternalTPUContext (line 235) | class _InternalTPUContext(object):
    method __init__ (line 258) | def __init__(self,
    method _assert_mode (line 291) | def _assert_mode(self):
    method with_mode (line 298) | def with_mode(self, mode):
    method mode (line 308) | def mode(self):
    method _get_master_address (line 311) | def _get_master_address(self):
    method _get_tpu_system_metadata (line 319) | def _get_tpu_system_metadata(self):
    method _get_device_assignment (line 341) | def _get_device_assignment(self):
    method tensor_core_embedding_columns (line 371) | def tensor_core_embedding_columns(self):
    method embedding_config (line 377) | def embedding_config(self):
    method allow_per_host_v2_parallel_get_next (line 401) | def allow_per_host_v2_parallel_get_next(self):
    method feed_hook (line 406) | def feed_hook(self):
    method model_parallelism_enabled (line 410) | def model_parallelism_enabled(self):
    method input_partition_dims (line 414) | def input_partition_dims(self):
    method device_assignment (line 418) | def device_assignment(self):
    method num_of_cores_per_host (line 423) | def num_of_cores_per_host(self):
    method num_cores (line 428) | def num_cores(self):
    method num_of_replicas_per_host (line 433) | def num_of_replicas_per_host(self):
    method num_replicas (line 442) | def num_replicas(self):
    method num_hosts (line 468) | def num_hosts(self):
    method config (line 473) | def config(self):
    method is_input_sharded_per_core (line 476) | def is_input_sharded_per_core(self):
    method is_input_per_host_with_iterators (line 483) | def is_input_per_host_with_iterators(self):
    method is_input_broadcast_with_iterators (line 488) | def is_input_broadcast_with_iterators(self):
    method is_input_slice_broadcast_to_all_cores (line 494) | def is_input_slice_broadcast_to_all_cores(self):
    method is_replica_across_hosts (line 501) | def is_replica_across_hosts(self):
    method is_running_on_cpu (line 509) | def is_running_on_cpu(self, is_export_mode=False):
    method _is_running_on_cpu (line 534) | def _is_running_on_cpu(self, is_export_mode):
    method global_batch_size (line 551) | def global_batch_size(self):
    method batch_size_for_input_fn (line 563) | def batch_size_for_input_fn(self):
    method batch_size_for_model_fn (line 578) | def batch_size_for_model_fn(self):
    method master_job (line 590) | def master_job(self):
    method tpu_host_placement_function (line 625) | def tpu_host_placement_function(self):
    method tpu_device_placement_function (line 652) | def tpu_device_placement_function(self):
    method tpu_ordinal_function (line 668) | def tpu_ordinal_function(self, host_id):
    method _validate_tpu_configuration (line 692) | def _validate_tpu_configuration(self):
    method device_for_replica (line 793) | def device_for_replica(self, replica_id):
  class _OneCoreTPUContext (line 823) | class _OneCoreTPUContext(_InternalTPUContext):
    method __init__ (line 826) | def __init__(self, config, train_batch_size, eval_batch_size,
    method _get_tpu_system_metadata (line 833) | def _get_tpu_system_metadata(self):
  class _TPUEstimatorReplicaContext (line 852) | class _TPUEstimatorReplicaContext(tf.distribute.ReplicaContext):
    method __init__ (line 859) | def __init__(self, replica_id_in_sync):
    method __enter__ (line 875) | def __enter__(self):
    method __exit__ (line 886) | def __exit__(self, exception_type, exception_value, traceback):
  function _get_tpu_context (line 893) | def _get_tpu_context(config, train_batch_size, eval_batch_size,

FILE: tensorflow_estimator/python/estimator/tpu/tpu_enqueue_sequence_test.py
  class TPUEnqueueSequenceTest (line 32) | class TPUEnqueueSequenceTest(tf.test.TestCase, parameterized.TestCase):
    method setUp (line 34) | def setUp(self):
    method get_activations_and_sequence_lengths (line 52) | def get_activations_and_sequence_lengths(
    method test_non_contiguous_sequence (line 220) | def test_non_contiguous_sequence(self):
    method test_non_contiguous_sequence_with_length_gt_max_sequence_length (line 294) | def test_non_contiguous_sequence_with_length_gt_max_sequence_length(se...
    method test_multivalent_sequence_features (line 378) | def test_multivalent_sequence_features(self, combiner: Text):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator.py
  function _is_iterable (line 123) | def _is_iterable(obj):
  class CatchInvalidHostcallFunctions (line 132) | class CatchInvalidHostcallFunctions(control_flow_ops.XLAControlFlowConte...
    method AddOp (line 134) | def AddOp(self, op):
  function _create_global_step (line 143) | def _create_global_step(graph):
  function _create_or_get_iterations_per_loop (line 162) | def _create_or_get_iterations_per_loop():
  function _sync_variables_ops (line 211) | def _sync_variables_ops(ctx):
  function _increase_eval_step_op (line 236) | def _increase_eval_step_op(iterations_per_loop):
  function _extract_key_names (line 254) | def _extract_key_names(tensor_or_dict):
  class PeriodicLogger (line 260) | class PeriodicLogger(object):
    method __init__ (line 262) | def __init__(self, seconds):
    method log (line 266) | def log(self, msg, *args, **kw):
  class _SIGNAL (line 272) | class _SIGNAL(object):
  class TPUEstimatorSpec (line 283) | class TPUEstimatorSpec(model_fn_lib._TPUEstimatorSpec):  # pylint: disab...
    method __new__ (line 326) | def __new__(cls,
    method as_estimator_spec (line 370) | def as_estimator_spec(self):
  class _OpQueueContext (line 400) | class _OpQueueContext(object):
    method __init__ (line 403) | def __init__(self, name, target, args):
    method stop (line 411) | def stop(self):
    method send_next_batch_signal (line 414) | def send_next_batch_signal(self, iterations):
    method read_iteration_counts (line 417) | def read_iteration_counts(self):
    method join (line 428) | def join(self):
  class _OpSignalOnceQueueContext (line 434) | class _OpSignalOnceQueueContext(_OpQueueContext):
    method __init__ (line 440) | def __init__(self, name, target, args):
    method send_next_batch_signal (line 444) | def send_next_batch_signal(self, iterations):
  class TPUInfeedOutfeedSessionHook (line 450) | class TPUInfeedOutfeedSessionHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 459) | def __init__(self,
    method begin (line 500) | def begin(self):
    method _run_infeed (line 518) | def _run_infeed(self, queue_ctx, session):
    method _run_outfeed (line 538) | def _run_outfeed(self, queue_ctx, session):
    method _create_infeed_controller (line 553) | def _create_infeed_controller(self, name, target, args):
    method _assertCompilationSucceeded (line 556) | def _assertCompilationSucceeded(self, result, coord):
    method after_create_session (line 566) | def after_create_session(self, session, coord):
    method before_run (line 601) | def before_run(self, run_context):
    method end (line 612) | def end(self, session):
  class TPUInfeedOutfeedSessionHookForPrediction (line 625) | class TPUInfeedOutfeedSessionHookForPrediction(TPUInfeedOutfeedSessionHo...
    method __init__ (line 627) | def __init__(self,
    method _create_infeed_controller (line 645) | def _create_infeed_controller(self, name, target, args):
  class _TPUStopAtStepHook (line 649) | class _TPUStopAtStepHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 671) | def __init__(self,
    method _next_iterations (line 706) | def _next_iterations(self, global_step, final_step):
    method begin (line 731) | def begin(self):
    method after_create_session (line 745) | def after_create_session(self, session, coord):
    method before_run (line 763) | def before_run(self, run_context):
    method after_run (line 768) | def after_run(self, run_context, run_values):
  class _SetEvalIterationsHook (line 797) | class _SetEvalIterationsHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 800) | def __init__(self, num_steps):
    method begin (line 808) | def begin(self):
    method after_create_session (line 811) | def after_create_session(self, session, coord):
  class _StoppingPredictHook (line 815) | class _StoppingPredictHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 818) | def __init__(self, scalar_stopping_signal):
    method begin (line 821) | def begin(self):
    method after_create_session (line 824) | def after_create_session(self, session, coord):
    method before_run (line 830) | def before_run(self, run_context):
    method after_run (line 833) | def after_run(self, run_context, run_values):
  function generate_per_core_enqueue_ops_fn_for_host (line 856) | def generate_per_core_enqueue_ops_fn_for_host(ctx, input_fn,
  function generate_per_host_enqueue_ops_fn_for_host (line 901) | def generate_per_host_enqueue_ops_fn_for_host(ctx, input_fn,
  function generate_per_host_v2_enqueue_ops_fn_for_host (line 993) | def generate_per_host_v2_enqueue_ops_fn_for_host(ctx, input_fn,
  function generate_broadcast_enqueue_ops_fn (line 1122) | def generate_broadcast_enqueue_ops_fn(ctx, input_fn, inputs_structure_re...
  class TensorPacker (line 1218) | class TensorPacker(object):
    method __init__ (line 1221) | def __init__(self, small_feature_dim_size,
    method maybe_concatenate_features (line 1227) | def maybe_concatenate_features(self, features):
    method maybe_split_features (line 1276) | def maybe_split_features(self, maybe_concatenated_features):
    method _get_small_feature_key (line 1285) | def _get_small_feature_key(self, dtype):
  class _InputPipeline (line 1289) | class _InputPipeline(object):
    class InputsStructureRecorder (line 1321) | class InputsStructureRecorder(object):
      method __init__ (line 1324) | def __init__(self, input_partition_dims=None):
      method flattened_input_dims (line 1348) | def flattened_input_dims(self):
      method has_labels (line 1352) | def has_labels(self):
      method _flatten_input_dims (line 1355) | def _flatten_input_dims(self, features, labels, feature_dims, label_...
      method validate_and_record_structure (line 1381) | def validate_and_record_structure(self, features, labels):
      method flatten_features_and_labels (line 1406) | def flatten_features_and_labels(self, features, labels, signals=None):
      method unflatten_features_and_labels (line 1419) | def unflatten_features_and_labels(self, flattened_inputs):
    method __init__ (line 1444) | def __init__(self, input_fn, batch_axis, ctx):
    method generate_infeed_enqueue_ops_and_dequeue_fn (line 1466) | def generate_infeed_enqueue_ops_and_dequeue_fn(self):
    method _invoke_input_fn_and_record_structure (line 1488) | def _invoke_input_fn_and_record_structure(self):
    method _validate_input_pipeline (line 1614) | def _validate_input_pipeline(self):
  function call_computation (line 1638) | def call_computation(computation_inputs, computation, batch_config=None):
  class _ModelFnWrapper (line 1700) | class _ModelFnWrapper(object):
    method __init__ (line 1710) | def __init__(self, model_fn, config, params, ctx):
    method call_without_tpu (line 1716) | def call_without_tpu(self, features, labels, is_export_mode):
    method _add_embedding_features (line 1719) | def _add_embedding_features(self, features, hook_dummy_table_variables):
    method convert_to_single_tpu_train_step (line 1733) | def convert_to_single_tpu_train_step(self, dequeue_fn):
    method convert_to_single_tpu_eval_step (line 1852) | def convert_to_single_tpu_eval_step(self, dequeue_fn):
    method convert_to_single_tpu_predict_step (line 1912) | def convert_to_single_tpu_predict_step(self, dequeue_fn):
    method _verify_tpu_spec_predictions (line 1961) | def _verify_tpu_spec_predictions(self, predictions):
    method _validate_model_features_and_labels (line 1975) | def _validate_model_features_and_labels(self, features, labels,
    method _call_model_fn (line 2014) | def _call_model_fn(self, features, labels, is_export_mode=False):
    method _verify_estimator_spec (line 2079) | def _verify_estimator_spec(self, estimator_spec):
  class _OutfeedHostCall (line 2097) | class _OutfeedHostCall(object):
    method __init__ (line 2100) | def __init__(self, ctx, outfeed_every_n_steps=1):
    method validate (line 2112) | def validate(host_calls):
    method create_cpu_hostcall (line 2136) | def create_cpu_hostcall(host_calls):
    method record (line 2157) | def record(self, host_calls):
    method create_enqueue_op (line 2179) | def create_enqueue_op(self, step=None):
    method create_tpu_hostcall (line 2209) | def create_tpu_hostcall(self):
  class _OutfeedHostCallHook (line 2322) | class _OutfeedHostCallHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 2325) | def __init__(self, tensors):
    method begin (line 2328) | def begin(self):
    method after_create_session (line 2341) | def after_create_session(self, session, coord):
    method before_run (line 2344) | def before_run(self, run_context):
    method end (line 2347) | def end(self, session):
  class _NotSaver (line 2351) | class _NotSaver(object):
    method __init__ (line 2354) | def __init__(self, message):
    method save (line 2357) | def save(self, *args, **kwargs):
  class ExamplesPerSecondHook (line 2362) | class ExamplesPerSecondHook(tf.compat.v1.train.StepCounterHook):
    method __init__ (line 2365) | def __init__(self,
    method _log_and_record (line 2378) | def _log_and_record(self, elapsed_steps, elapsed_time, global_step):
  class InstallSignalHandlerHook (line 2395) | class InstallSignalHandlerHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 2402) | def __init__(self):
    method before_run (line 2405) | def before_run(self, run_context):
    method end (line 2408) | def end(self, session):
  class ExportSavedModelApiVersion (line 2412) | class ExportSavedModelApiVersion(enum.Enum):
  class BatchConfig (line 2417) | class BatchConfig(
    method __new__ (line 2424) | def __new__(cls,
  class TPUEstimator (line 2458) | class TPUEstimator(estimator_lib.Estimator):
    method __init__ (line 2674) | def __init__(self,
    method _add_meta_graph_for_mode (line 2868) | def _add_meta_graph_for_mode(self,
    method _call_model_fn (line 2919) | def _call_model_fn(self, features, labels, mode, config):
    method _call_model_fn_for_inference (line 2940) | def _call_model_fn_for_inference(self, features, labels, mode, config):
    method _create_global_step (line 2953) | def _create_global_step(self, graph):
    method _convert_train_steps_to_hooks (line 2967) | def _convert_train_steps_to_hooks(self, steps, max_steps):
    method _convert_eval_steps_to_hooks (line 2989) | def _convert_eval_steps_to_hooks(self, steps):
    method _call_input_fn (line 3005) | def _call_input_fn(self, input_fn, mode, input_context=None):
    method _validate_features_in_predict_input (line 3076) | def _validate_features_in_predict_input(self, result):
    method train (line 3087) | def train(self,
    method evaluate (line 3108) | def evaluate(self,
    method predict (line 3129) | def predict(self,
    method _augment_model_fn (line 3154) | def _augment_model_fn(self, model_fn, batch_axis):
  function _check_add_preemption_hook (line 3549) | def _check_add_preemption_hook(cluster):
  function _export_output_to_tensors (line 3555) | def _export_output_to_tensors(export_output):
  function _clone_export_output_with_tensors (line 3581) | def _clone_export_output_with_tensors(export_output, tensors):
  function _eval_on_tpu_system (line 3615) | def _eval_on_tpu_system(ctx, model_fn_wrapper, dequeue_fn):
  function _train_on_tpu_system (line 3653) | def _train_on_tpu_system(ctx, model_fn_wrapper, dequeue_fn):
  function _predict_on_tpu_system (line 3692) | def _predict_on_tpu_system(ctx, model_fn_wrapper, dequeue_fn):
  function _wrap_computation_in_while_loop (line 3735) | def _wrap_computation_in_while_loop(device, op_fn):
  function _wrap_computation_in_while_loop_with_stopping_signals (line 3753) | def _wrap_computation_in_while_loop_with_stopping_signals(device, op_fn):
  function _validate_tpu_training_graph (line 3775) | def _validate_tpu_training_graph(ctx):
  class _CapturedObject (line 3799) | class _CapturedObject(object):
    method __init__ (line 3806) | def __init__(self):
    method capture (line 3810) | def capture(self, o):
    method get (line 3818) | def get(self):
  function _get_scaffold (line 3826) | def _get_scaffold(captured_scaffold_fn):
  class _CapturingContext (line 3849) | class _CapturingContext(control_flow_ops.ControlFlowContext):
    method __init__ (line 3852) | def __init__(self, message):
    method to_control_flow_context_def (line 3856) | def to_control_flow_context_def(self, context_def, export_scope=None):
    method AddOp (line 3862) | def AddOp(self, op):  # pylint: disable=invalid-name
    method AddValue (line 3868) | def AddValue(self, value):
    method __enter__ (line 3872) | def __enter__(self):
    method __exit__ (line 3879) | def __exit__(self, _, __, ___):  # pylint: disable=invalid-name
  class _Inputs (line 3883) | class _Inputs(object):
    method __init__ (line 3889) | def __init__(self, features=None, labels=None, dataset=None, signals=N...
    method from_input_fn (line 3904) | def from_input_fn(return_values):
    method _parse_inputs (line 3914) | def _parse_inputs(return_values):
    method is_dataset (line 3922) | def is_dataset(self):
    method dataset_initializer (line 3926) | def dataset_initializer(self):
    method features_and_labels (line 3935) | def features_and_labels(self):
    method signals (line 3946) | def signals(self):
    method dataset (line 3950) | def dataset(self):
  class _InputsWithStoppingSignals (line 3954) | class _InputsWithStoppingSignals(_Inputs):
    method __init__ (line 3957) | def __init__(self,
    method features_and_labels (line 4005) | def features_and_labels(self):
    method signals (line 4018) | def signals(self):
    method insert_stopping_signal (line 4029) | def insert_stopping_signal(stop, batch_size, add_padding=False):
  class _StopSignals (line 4079) | class _StopSignals(object):
    method __init__ (line 4085) | def __init__(self, stop, batch_size, padding_mask=None):
    method as_dict (line 4090) | def as_dict(self):
    method as_scalar_stopping_signal (line 4106) | def as_scalar_stopping_signal(signals):
    method should_stop (line 4110) | def should_stop(scalar_stopping_signal):
  class _PaddingSignals (line 4123) | class _PaddingSignals(object):
    method pad_features_and_labels (line 4127) | def pad_features_and_labels(features, labels, batch_size):
    method slice_tensor_or_dict (line 4166) | def slice_tensor_or_dict(tensor_or_dict, signals):
    method _find_any_tensor (line 4208) | def _find_any_tensor(batch_features):
    method _padding_mask (line 4217) | def _padding_mask(real_batch_size, missing_count, batch_size):
  function _verify_cross_hosts_transfer_size (line 4227) | def _verify_cross_hosts_transfer_size(tensor_dict, message):
  function _add_item_to_params (line 4247) | def _add_item_to_params(params, key, value):
  function export_estimator_savedmodel (line 4260) | def export_estimator_savedmodel(estimator,
  function model_fn_inference_on_tpu (line 4298) | def model_fn_inference_on_tpu(model_fn,
  function _build_computation_for_inference (line 4372) | def _build_computation_for_inference(model_fn, labels, config, params):
  function _build_tpu_computation_for_inference (line 4410) | def _build_tpu_computation_for_inference(model_fn, features, labels, con...
  function inference_on_tpu (line 4468) | def inference_on_tpu(computation,

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_embedding_test.py
  function dense_computation (line 51) | def dense_computation(features):
  function create_run_config (line 56) | def create_run_config(iterations_per_loop, **kwargs):
  class TPUEstimatorFeatureColumnTestBase (line 66) | class TPUEstimatorFeatureColumnTestBase(tf.test.TestCase):
    method setUp (line 68) | def setUp(self):
    method tearDown (line 80) | def tearDown(self):
    method _create_estimator_with_feature_columns (line 84) | def _create_estimator_with_feature_columns(self,
  class TPUEstimatorFeatureColumnTest (line 179) | class TPUEstimatorFeatureColumnTest(TPUEstimatorFeatureColumnTestBase,
    method test_get_tpu_embedding_config_from_feature_columns (line 182) | def test_get_tpu_embedding_config_from_feature_columns(self):
    method _create_estimator_with_config_dicts (line 297) | def _create_estimator_with_config_dicts(self,
    method _get_vocab_feature_columns (line 376) | def _get_vocab_feature_columns(self,
    method _get_vocab_input_fn_and_feature_columns (line 404) | def _get_vocab_input_fn_and_feature_columns(self,
    method test_feature_in_two_embeddings (line 467) | def test_feature_in_two_embeddings(self):
    method _test_two_features (line 482) | def _test_two_features(self, shared_embedding, sequence_column,
    method test_two_features_with_config_dicts (line 558) | def test_two_features_with_config_dicts(self,
    method test_non_tpu_embedding_column (line 607) | def test_non_tpu_embedding_column(self):
    method test_feature_in_embedding_and_shared_embedding (line 623) | def test_feature_in_embedding_and_shared_embedding(self):
    method test_sequence_column_with_no_max_length (line 642) | def test_sequence_column_with_no_max_length(self):
    method test_non_sequence_column_with_max_length (line 652) | def test_non_sequence_column_with_max_length(self):
    method test_sequence_column_shared_embedding_wrong_max_sequence_length (line 662) | def test_sequence_column_shared_embedding_wrong_max_sequence_length(se...
    method test_sequence_column_shared_embedding_non_sequence_with_max_length (line 673) | def test_sequence_column_shared_embedding_non_sequence_with_max_length...
    method test_sequence_column_shared_embedding_sequence_without_max_length (line 684) | def test_sequence_column_shared_embedding_sequence_without_max_length(...
    method test_sequence_column_length (line 697) | def test_sequence_column_length(self, input_method):
    method test_unknown_partition_strategy (line 736) | def test_unknown_partition_strategy(self):
    method test_mod_partition_strategy_on_cpu (line 748) | def test_mod_partition_strategy_on_cpu(self):
    method test_two_features (line 769) | def test_two_features(self, shared, sequence, input_method):
    method test_dynamic_learning_rate (line 786) | def test_dynamic_learning_rate(self, input_method):
  class TPUEstimatorWeightedFeatureColumnTest (line 931) | class TPUEstimatorWeightedFeatureColumnTest(TPUEstimatorFeatureColumnTes...
    method test_embedding_with_weighted_categorical_column (line 937) | def test_embedding_with_weighted_categorical_column(self, input_method):
    method test_shared_embedding_with_weighted_categorical_column_and_dataset (line 997) | def test_shared_embedding_with_weighted_categorical_column_and_dataset(
    method test_embedding_with_with_weighted_categorical_column_with_vocab_error (line 1064) | def test_embedding_with_with_weighted_categorical_column_with_vocab_er...
    method test_embedding_with_weighted_categorical_column_dense_weights_error (line 1130) | def test_embedding_with_weighted_categorical_column_dense_weights_erro...
    method test_embedding_with_weighted_categorical_column_share_weights_error (line 1173) | def test_embedding_with_weighted_categorical_column_share_weights_erro...
    method _test_tensor_core_embedding (line 1225) | def _test_tensor_core_embedding(self,
    method test_tensor_core_embedding (line 1303) | def test_tensor_core_embedding(self, shared, both_embeddings, input_me...

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_evaluation_test.py
  function dense_computation (line 45) | def dense_computation(features):
  function get_model_fn (line 50) | def get_model_fn(export_tpu_tensor=True, export_cpu_tensor=False,
  function dummy_input_fn_with_dataset (line 100) | def dummy_input_fn_with_dataset(batch_size, repeat=True, x=None):
  function dummy_input_fn (line 118) | def dummy_input_fn(batch_size, repeat=True):
  function create_run_config (line 124) | def create_run_config(iterations_per_loop, **kwargs):
  class TPUEstimatorEvaluationTest (line 134) | class TPUEstimatorEvaluationTest(tf.test.TestCase):
    method _create_input_fn (line 136) | def _create_input_fn(self):
    method _create_head (line 141) | def _create_head(self, mode, loss, eval_metrics):
    method _create_head_with_eval_metric_ops (line 154) | def _create_head_with_eval_metric_ops(self, mode, loss, eval_metric_ops):
    method _metric_fn_on_cpu (line 180) | def _metric_fn_on_cpu(self, labels, predictions):
    method _model_fn_without_eval_metrics (line 185) | def _model_fn_without_eval_metrics(self, features, labels, mode, params):
    method _model_fn_with_eval_tensor_list (line 194) | def _model_fn_with_eval_tensor_list(self, features, labels, mode, para...
    method _model_fn_with_eval_dict (line 205) | def _model_fn_with_eval_dict(self, features, labels, mode, params):
    method _model_fn_with_eval_metric_ops (line 218) | def _model_fn_with_eval_metric_ops(self, features, labels, mode, params):
    method _test_eval_steps (line 229) | def _test_eval_steps(self, model_fn, expected_eval_steps, iterations):
    method test_no_eval_metrics (line 271) | def test_no_eval_metrics(self):
    method test_eval_steps_not_effected_by_training_iterations (line 282) | def test_eval_steps_not_effected_by_training_iterations(self):
    method test_eval_steps_with_no_eval_metrics (line 292) | def test_eval_steps_with_no_eval_metrics(self):
    method test_eval_metrics_with_tensor_list (line 298) | def test_eval_metrics_with_tensor_list(self):
    method test_eval_batch_size_with_non_divisible_num_shards_broadcast_mode (line 309) | def test_eval_batch_size_with_non_divisible_num_shards_broadcast_mode(...
    method test_eval_metrics_with_tensor_list_on_cpu (line 322) | def test_eval_metrics_with_tensor_list_on_cpu(self):
    method test_eval_metrics_with_dict (line 334) | def test_eval_metrics_with_dict(self):
    method test_eval_metrics_with_dict_on_cpu (line 345) | def test_eval_metrics_with_dict_on_cpu(self):
    method test_eval_metrics_ops_cpu_training (line 357) | def test_eval_metrics_ops_cpu_training(self):
    method test_eval_metrics_ops_cpu_training_warning (line 370) | def test_eval_metrics_ops_cpu_training_warning(self):
    method test_eval_metrics_ops_tpu_training (line 384) | def test_eval_metrics_ops_tpu_training(self):
    method test_eval_metrics_ops_tpu_training_failure (line 397) | def test_eval_metrics_ops_tpu_training_failure(self):
    method test_error_out_if_steps_is_float (line 414) | def test_error_out_if_steps_is_float(self):
    method test_error_out_if_steps_is_invalid (line 425) | def test_error_out_if_steps_is_invalid(self):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_export_test.py
  function dense_computation (line 68) | def dense_computation(features):
  function get_model_fn (line 73) | def get_model_fn(export_tpu_tensor=True, export_cpu_tensor=False,
  function dummy_input_fn_with_dataset (line 123) | def dummy_input_fn_with_dataset(batch_size, repeat=True, x=None):
  function dummy_input_fn (line 141) | def dummy_input_fn(batch_size, repeat=True):
  function create_run_config (line 147) | def create_run_config(iterations_per_loop, **kwargs):
  class TPUEstimatorExportTest (line 157) | class TPUEstimatorExportTest(parameterized.TestCase):
    method setUp (line 159) | def setUp(self):
    method test_export_tpu_savedmodel_e2e (line 182) | def test_export_tpu_savedmodel_e2e(self, export_tpu_tensor, export_cpu...
    method _validate_export (line 231) | def _validate_export(self, export_dir_base, export_dir, export_tpu_ten...
    method test_export_tpu_savedmodel_export_to_cpu_false (line 316) | def test_export_tpu_savedmodel_export_to_cpu_false(self):
    method test_export_tpu_savedmodel_export_to_tpu_false (line 344) | def test_export_tpu_savedmodel_export_to_tpu_false(self):
    method test_export_tpu_savedmodel_export_to_tpu_false_eval (line 378) | def test_export_tpu_savedmodel_export_to_tpu_false_eval(self):
    method test_export_estimator_savedmodel (line 416) | def test_export_estimator_savedmodel(self):
    method test_regression_output_tensors_roundtrip (line 447) | def test_regression_output_tensors_roundtrip(self):
    method test_predict_output_tensors_roundtrip (line 462) | def test_predict_output_tensors_roundtrip(self):
    method test_classification_output_tensors_roundtrip_classes_only (line 485) | def test_classification_output_tensors_roundtrip_classes_only(self):
    method test_classification_output_tensors_roundtrip_scores_only (line 502) | def test_classification_output_tensors_roundtrip_scores_only(self):
    method test_classification_output_tensors_roundtrip_classify_both (line 518) | def test_classification_output_tensors_roundtrip_classify_both(self):
  function get_model_fn_v2 (line 539) | def get_model_fn_v2():
  class TPUEstimatorExportV2Test (line 590) | class TPUEstimatorExportV2Test(parameterized.TestCase):
    method setUp (line 592) | def setUp(self):
    method test_export_tpu_savedmodel_e2e (line 598) | def test_export_tpu_savedmodel_e2e(self):
    method _validate_export (line 626) | def _validate_export(self, export_dir_base, export_dir):
    method test_export_tpu_savedmodel_export_to_tpu_false (line 701) | def test_export_tpu_savedmodel_export_to_tpu_false(self):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_gradients_test.py
  function create_model_fn (line 50) | def create_model_fn(feature_columns, optimizer_type='adagrad'):
  function get_estimator (line 87) | def get_estimator(use_tpu,
  function get_feature_columns (line 127) | def get_feature_columns():
  class _EmbeddingVariableHook (line 142) | class _EmbeddingVariableHook(tf.train.SessionRunHook):
    method __init__ (line 145) | def __init__(self, use_tpu, include_slot_vars=True):
    method _set_bias_var (line 149) | def _set_bias_var(self):
    method begin (line 152) | def begin(self):
    method after_create_session (line 166) | def after_create_session(self, session, coord):
    method after_run (line 173) | def after_run(self, run_context, run_values):
  function get_activation_gradients (line 180) | def get_activation_gradients(label):
  function get_embedding_update (line 199) | def get_embedding_update(gradient, previous_accum_inc=0.0):
  function dense_to_sparse (line 215) | def dense_to_sparse(dense_tensor, out_type, ignore_value=-1):
  class TPUEstimatorGradientsSimpleTest (line 224) | class TPUEstimatorGradientsSimpleTest(tf.test.TestCase):
    method setUp (line 236) | def setUp(self):
    method _input_fn (line 239) | def _input_fn(self, params):
    method test_input_fn (line 259) | def test_input_fn(self):
    method assert_embedding_variables (line 285) | def assert_embedding_variables(self,
    method assert_embedding_slot_variables (line 312) | def assert_embedding_slot_variables(self, gradients_for_embedding,
    method test_one_sample_per_core (line 337) | def test_one_sample_per_core(self):
    method test_one_sample_per_core_tpu_vs_cpu (line 385) | def test_one_sample_per_core_tpu_vs_cpu(self):
    method test_multi_samples_per_core (line 415) | def test_multi_samples_per_core(self):
    method test_multi_samples_per_core_tpu_vs_cpu (line 469) | def test_multi_samples_per_core_tpu_vs_cpu(self):
  class TPUEstimatorGradientsWithIdCollisionTest (line 502) | class TPUEstimatorGradientsWithIdCollisionTest(tf.test.TestCase):
    method setUp (line 504) | def setUp(self):
    method _input_fn (line 507) | def _input_fn(self, params):
    method test_input_fn (line 548) | def test_input_fn(self):
    method test_adagrad_opt_embedding_variables_on_tpu (line 579) | def test_adagrad_opt_embedding_variables_on_tpu(self):
    method test_adagrad_opt_embedding_variables_on_cpu (line 738) | def test_adagrad_opt_embedding_variables_on_cpu(self):
    method test_sgd_opt_embedding_variables_on_cpu (line 781) | def test_sgd_opt_embedding_variables_on_cpu(self):
    method test_sgd_opt_embedding_variables_cpu_vs_tpu (line 821) | def test_sgd_opt_embedding_variables_cpu_vs_tpu(self):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_input_v2_test.py
  function create_run_config (line 35) | def create_run_config(iterations_per_loop, **kwargs):
  function dense_computation (line 44) | def dense_computation(features):
  function model_fn_global_step_incrementer (line 49) | def model_fn_global_step_incrementer(features, labels, mode, params):
  function dummy_input_fn_with_dataset (line 71) | def dummy_input_fn_with_dataset(batch_size, fea_len=1, repeat=True, x=No...
  class TpuEstimatorInputV2Test (line 89) | class TpuEstimatorInputV2Test(parameterized.TestCase, tf.test.TestCase):
    method test_batch_size (line 92) | def test_batch_size(self, num_cores_per_replica, num_shards):
    method test_run_spatial_partition (line 114) | def test_run_spatial_partition(self):
    method test_predict_mode (line 137) | def test_predict_mode(self):
    method test_evaluate_mode (line 175) | def test_evaluate_mode(self):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_integration_test.py
  function dense_computation (line 50) | def dense_computation(features):
  function model_fn_global_step_incrementer (line 55) | def model_fn_global_step_incrementer(features, labels, mode, params):
  function dummy_input_fn_with_dataset (line 77) | def dummy_input_fn_with_dataset(batch_size, repeat=True, x=None):
  function dummy_input_fn (line 95) | def dummy_input_fn(batch_size, repeat=True):
  function create_run_config (line 101) | def create_run_config(iterations_per_loop, **kwargs):
  class TPUEstimatorIntegrationTest (line 111) | class TPUEstimatorIntegrationTest(tf.test.TestCase):
    method setUp (line 113) | def setUp(self):
    method export_mode (line 123) | def export_mode(self):
    method assertInputFnCalledCountAndBatch (line 133) | def assertInputFnCalledCountAndBatch(self, expected_called_count,
    method _generate_expected_batch_size_and_called_count (line 142) | def _generate_expected_batch_size_and_called_count(
    method _wrap_input_fn_with_batch_size (line 195) | def _wrap_input_fn_with_batch_size(self, batch_size, input_fn):
    method _make_input_fn (line 202) | def _make_input_fn(self, mode, repeat=False, take=None):
    method _make_model_fn (line 235) | def _make_model_fn(self, batch_size_dict, use_tpu_estimator_spec=False):
    method _test_identity_savedmodel (line 305) | def _test_identity_savedmodel(self, export_dir):
    method test_complete_flow_with_per_core_input (line 323) | def test_complete_flow_with_per_core_input(self):
    method test_complete_flow_with_per_host_input (line 388) | def test_complete_flow_with_per_host_input(self):
    method test_complete_flow_with_eval_on_tpu (line 449) | def test_complete_flow_with_eval_on_tpu(self):
    method test_complete_flow_with_no_tpu (line 510) | def test_complete_flow_with_no_tpu(self):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_model_parallelism_test.py
  function dense_computation (line 44) | def dense_computation(features):
  function model_fn_global_step_incrementer (line 49) | def model_fn_global_step_incrementer(features, labels, mode, params):
  function dummy_input_fn_with_dataset (line 71) | def dummy_input_fn_with_dataset(batch_size, repeat=True, x=None):
  function dummy_input_fn (line 89) | def dummy_input_fn(batch_size, repeat=True):
  function create_run_config (line 95) | def create_run_config(iterations_per_loop, num_shards, num_cores_per_rep...
  class TPUEstimatorModelParallelismConstructorTest (line 106) | class TPUEstimatorModelParallelismConstructorTest(tf.test.TestCase):
    method test_fail_model_parallelism_for_per_core_input (line 108) | def test_fail_model_parallelism_for_per_core_input(self):
  class TPUEstimatorModelParallelismTrainingTest (line 121) | class TPUEstimatorModelParallelismTrainingTest(tf.test.TestCase):
    method _train_and_return_global_steps (line 123) | def _train_and_return_global_steps(self,
    method test_train_steps_with_model_parallelism (line 170) | def test_train_steps_with_model_parallelism(self):
  class TPUEstimatorModelParallelismEvaluationTest (line 182) | class TPUEstimatorModelParallelismEvaluationTest(tf.test.TestCase):
    method _create_input_fn (line 184) | def _create_input_fn(self):
    method _create_head (line 191) | def _create_head(self, mode, loss, eval_metrics):
    method _metric_fn_on_cpu (line 203) | def _metric_fn_on_cpu(self, labels, predictions):
    method _model_fn_with_eval_tensor_list (line 208) | def _model_fn_with_eval_tensor_list(self, features, labels, mode, para...
    method _model_fn_with_eval_dict (line 219) | def _model_fn_with_eval_dict(self, features, labels, mode, params):
    method _test_eval_steps (line 233) | def _test_eval_steps(self, expected_eval_steps, iterations):
    method test_eval_metrics_with_tensor_list (line 274) | def test_eval_metrics_with_tensor_list(self):
    method test_eval_metrics_with_dict (line 286) | def test_eval_metrics_with_dict(self):
    method test_fail_with_wrong_num_shards (line 298) | def test_fail_with_wrong_num_shards(self):
  class TPUEstimatorModelParallelismInFeedTest (line 311) | class TPUEstimatorModelParallelismInFeedTest(tf.test.TestCase):
    method setUp (line 313) | def setUp(self):
    method test_infeed_even_partition (line 321) | def test_infeed_even_partition(self):
    method test_infeed_uneven_partition (line 339) | def test_infeed_uneven_partition(self):
    method test_infeed_tailing_zero_partition (line 357) | def test_infeed_tailing_zero_partition(self):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_signals_test.py
  function make_input_fn (line 26) | def make_input_fn(num_samples):
  function make_input_fn_with_labels (line 43) | def make_input_fn_with_labels(num_samples):
  class TPUEstimatorStoppingSignalsTest (line 60) | class TPUEstimatorStoppingSignalsTest(tf.test.TestCase):
    method test_normal_output_without_signals (line 62) | def test_normal_output_without_signals(self):
    method test_output_with_stopping_signals (line 90) | def test_output_with_stopping_signals(self):
  class TPUEstimatorStoppingSignalsWithPaddingTest (line 129) | class TPUEstimatorStoppingSignalsWithPaddingTest(tf.test.TestCase):
    method test_num_samples_divisible_by_batch_size (line 131) | def test_num_samples_divisible_by_batch_size(self):
    method test_num_samples_not_divisible_by_batch_size (line 174) | def test_num_samples_not_divisible_by_batch_size(self):
    method test_slice (line 242) | def test_slice(self):
    method test_slice_with_multi_invocations_per_step (line 282) | def test_slice_with_multi_invocations_per_step(self):

FILE: tensorflow_estimator/python/estimator/tpu/tpu_estimator_test.py
  function events_from_file (line 101) | def events_from_file(filepath):
  function dense_computation (line 119) | def dense_computation(features):
  function get_model_fn (line 126) | def get_model_fn(export_tpu_tensor=True,
  function dummy_input_fn_with_dataset (line 178) | def dummy_input_fn_with_dataset(dataset_size,
  function dummy_input_fn (line 201) | def dummy_input_fn(batch_size, repeat=True):
  function create_run_config (line 207) | def create_run_config(iterations_per_loop, num_shards=None, **kwargs):
  class TPUEstimatorConstructorTest (line 217) | class TPUEstimatorConstructorTest(test.TestCase):
    method test_reserved_key (line 219) | def test_reserved_key(self):
    method test_missing_train_batch_size (line 226) | def test_missing_train_batch_size(self):
    method test_invalid_batch_size (line 233) | def test_invalid_batch_size(self):
    method test_batch_size_with_num_shards_for_per_core_input (line 239) | def test_batch_size_with_num_shards_for_per_core_input(self):
    method test_batch_size_with_num_shards_for_per_host_input (line 256) | def test_batch_size_with_num_shards_for_per_host_input(self):
    method test_train_batch_size_with_non_divisible_num_shards (line 272) | def test_train_batch_size_with_non_divisible_num_shards(self):
    method test_train_batch_size_with_non_divisible_num_shards_broadcast_mode (line 279) | def test_train_batch_size_with_non_divisible_num_shards_broadcast_mode...
    method test_eval_batch_size_with_non_divisible_num_shards (line 296) | def test_eval_batch_size_with_non_divisible_num_shards(self):
    method test_predict_batch_size_with_non_divisible_num_shards_broadcast_mode (line 306) | def test_predict_batch_size_with_non_divisible_num_shards_broadcast_mode(
    method test_predict_batch_size_with_non_divisible_num_shards (line 323) | def test_predict_batch_size_with_non_divisible_num_shards(self):
    method test_invalid_num_shards (line 338) | def test_invalid_num_shards(self):
  class TPUEstimatorTPUContextTest (line 348) | class TPUEstimatorTPUContextTest(test.TestCase):
    method test_context_replicas (line 350) | def test_context_replicas(self):
    method _query_system (line 368) | def _query_system(self, master_address, cluster_def, query_topology):
    method test_num_cores_per_replica_is_not_greater_than_num_cores_per_host (line 423) | def test_num_cores_per_replica_is_not_greater_than_num_cores_per_host(...
    method test_device_for_replica_fn (line 447) | def test_device_for_replica_fn(self):
    method test_input_deployment_for_per_host (line 472) | def test_input_deployment_for_per_host(self):
    method test_input_deployment_for_per_host_v2 (line 525) | def test_input_deployment_for_per_host_v2(self):
    method test_input_deployment_for_per_host_v2_with_model_parallelism (line 579) | def test_input_deployment_for_per_host_v2_with_model_parallelism(self):
    method test_input_deployment_model_parallelism_cross_host_replica (line 638) | def test_input_deployment_model_parallelism_cross_host_replica(self):
    method test_input_deployment_for_broadcast_mode (line 698) | def test_input_deployment_for_broadcast_mode(self):
    method test_input_deployment_for_eval_broadcast_mode (line 735) | def test_input_deployment_for_eval_broadcast_mode(self):
    method test_input_deployment_for_per_core (line 787) | def test_input_deployment_for_per_core(self):
    method test_hparams_as_params (line 840) | def test_hparams_as_params(self):
  class TPUEstimatorInputFnTest (line 859) | class TPUEstimatorInputFnTest(parameterized.TestCase):
    method setUp (line 861) | def setUp(self):
    method tearDown (line 868) | def tearDown(self):
    method test_succeed_with_dataset (line 874) | def test_succeed_with_dataset(self, num_features):
    method test_succeed_with_input_return_features_and_labels_with_dataset (line 906) | def test_succeed_with_input_return_features_and_labels_with_dataset(se...
    method test_fail_with_queue_based_input_fn_in_while_loop (line 919) | def test_fail_with_queue_based_input_fn_in_while_loop(self):
    method test_warning_with_queue_based_input_fn (line 940) | def test_warning_with_queue_based_input_fn(self):
    method test_nested_inputs_dict (line 961) | def test_nested_inputs_dict(self):
    method test_nested_inputs_tuple (line 964) | def test_nested_inputs_tuple(self):
    method test_nested_inputs_namedtuple (line 967) | def test_nested_inputs_namedtuple(self):
    method help_test_nested_inputs (line 970) | def help_test_nested_inputs(self, nest_type):
  class _DummyHook (line 1029) | class _DummyHook(session_run_hook.SessionRunHook):
    method __init__ (line 1032) | def __init__(self):
    method after_create_session (line 1036) | def after_create_session(self, sees, coord):
    method called (line 1041) | def called(self):
  class TPUEstimatorModelFnTest (line 1045) | class TPUEstimatorModelFnTest(test.TestCase):
    method test_succeed_with_missing_labels (line 1047) | def test_succeed_with_missing_labels(self):
    method test_succeed_with_log_step_count_steps_none (line 1063) | def test_succeed_with_log_step_count_steps_none(self):
    method test_missing_labels_in_model_fn_not_input_fn (line 1080) | def test_missing_labels_in_model_fn_not_input_fn(self):
    method test_missing_params (line 1098) | def test_missing_params(self):
    method test_invalid_arg (line 1115) | def test_invalid_arg(self):
    method test_valid_training_hook (line 1127) | def test_valid_training_hook(self):
    method test_valid_eval_hook (line 1150) | def test_valid_eval_hook(self):
    method test_valid_prediction_hook (line 1174) | def test_valid_prediction_hook(self):
    method test_invalid_training_chief_hook (line 1200) | def test_invalid_training_chief_hook(self):
    method test_access_device_assignment_in_model_fn (line 1226) | def test_access_device_assignment_in_model_fn(self):
    method test_fail_to_call_deployment_in_model_fn (line 1245) | def test_fail_to_call_deployment_in_model_fn(self):
  class TPUEstimatorPredictionTest (line 1264) | class TPUEstimatorPredictionTest(test.TestCase):
    method _test_train_and_predict (line 1266) | def _test_train_and_predict(self, run_config, dataset_size,
    method _construct_run_config (line 1309) | def _construct_run_config(self,
    method test_train_and_predict_per_host_v1 (line 1322) | def test_train_and_predict_per_host_v1(self):
    method test_train_and_predict_per_host_v2_evenly_distributed (line 1327) | def test_train_and_predict_per_host_v2_evenly_distributed(self):
    method test_train_and_predict_per_host_v2_not_evenly_distributed (line 1332) | def test_train_and_predict_per_host_v2_not_evenly_distributed(self):
    method test_train_and_predict_with_input_partition (line 1337) | def test_train_and_predict_with_input_partition(self):
    method test_train_and_predict_broadcast (line 1348) | def test_train_and_predict_broadcast(self):
    method test_non_static_shape (line 1353) | def test_non_static_shape(self):
    method test_predict_on_cpu (line 1376) | def test_predict_on_cpu(self):
    method test_train_and_export (line 1424) | def test_train_and_export(self):
  class TPUEstimatorTrainingTest (line 1454) | class TPUEstimatorTrainingTest(test.TestCase):
    method _train_and_return_global_steps (line 1456) | def _train_and_return_global_steps(self,
    method test_train_steps_not_divisible_by_iterations (line 1498) | def test_train_steps_not_divisible_by_iterations(self):
    method test_train_steps_divisible_by_iterations (line 1509) | def test_train_steps_divisible_by_iterations(self):
    method test_train_steps_with_large_iterations (line 1520) | def test_train_steps_with_large_iterations(self):
    method test_train_max_steps_not_divisible_by_iterations (line 1531) | def test_train_max_steps_not_divisible_by_iterations(self):
    method test_train_max_steps_divisible_by_iterations (line 1542) | def test_train_max_steps_divisible_by_iterations(self):
    method test_train_max_steps_with_large_iterations (line 1553) | def test_train_max_steps_with_large_iterations(self):
    method test_error_out_if_train_steps_is_float (line 1564) | def test_error_out_if_train_steps_is_float(self):
    method test_error_out_if_train_steps_is_invalid (line 1568) | def test_error_out_if_train_steps_is_invalid(self):
    method test_error_out_if_train_max_steps_is_float (line 1572) | def test_error_out_if_train_max_steps_is_float(self):
    method test_error_out_if_train_max_steps_is_invalid (line 1577) | def test_error_out_if_train_max_steps_is_invalid(self):
    method test_warm_starts (line 1581) | def test_warm_starts(self):
  class TPUEstimatorValidationTest (line 1627) | class TPUEstimatorValidationTest(parameterized.TestCase, test.TestCase):
    method _query_system (line 1629) | def _query_system(self, master_address, cluster_def, query_topology):
    method test_error_if_cross_replica_sum_missing (line 1638) | def test_error_if_cross_replica_sum_missing(self):
    method test_no_error_if_cross_replica_sum_present (line 1661) | def test_no_error_if_cross_replica_sum_present(self):
    method test_error_dynamic_shape_tensor_features_for_model (line 1674) | def test_error_dynamic_shape_tensor_features_for_model(self):
    method test_error_dynamic_shape_dict_tensor_features_for_model (line 1707) | def test_error_dynamic_shape_dict_tensor_features_for_model(self):
    method test_error_dynamic_shape_tensor_labels_for_model (line 1740) | def test_error_dynamic_shape_tensor_labels_for_model(self):
    method test_error_dynamic_shape_dict_tensor_labels_for_model (line 1775) | def test_error_dynamic_shape_dict_tensor_labels_for_model(self):
    method test_error_none_eval_batch_size_for_evaluation_mode (line 1811) | def test_error_none_eval_batch_size_for_evaluation_mode(self):
    method test_error_none_predict_batch_size_for_prediction_mode (line 1826) | def test_error_none_predict_batch_size_for_prediction_mode(self):
    method test_error_num_hosts_and_replicas_larger_than_1_in_eval_and_predict_mode (line 1845) | def test_error_num_hosts_and_replicas_larger_than_1_in_eval_and_predic...
    method test_evaluate_1host_and_replicas_larger_than_1_with_PER_HOST_V2 (line 1880) | def test_evaluate_1host_and_replicas_larger_than_1_with_PER_HOST_V2(
    method test_no_error_1host_1replica_in_eval_and_predict_mode (line 1921) | def test_no_error_1host_1replica_in_eval_and_predict_mode(
  class TPUConfigTest (line 1946) | class TPUConfigTest(test.TestCase):
    method _create_ctx (line 1948) | def _create_ctx(self, run_config, mode=_TRAIN):
    method test_no_cluster_spec (line 1955) | def test_no_cluster_spec(self):
    method test_cluster_spec_prop (line 1975) | def test_cluster_spec_prop(self):
    method test_cluster_spec_prop_multi_jobs (line 1987) | def test_cluster_spec_prop_multi_jobs(self):
    method test_cluster_spec_prop_cannot_infer (line 2002) | def test_cluster_spec_prop_cannot_infer(self):
    method test_session_config_none (line 2036) | def test_session_config_none(self):
    method test_override_name (line 2047) | def test_override_name(self):
    method test_evaluation_master (line 2053) | def test_evaluation_master(self):
    method test_input_partition_config (line 2061) | def test_input_partition_config(self):
  class TPUEstimatorInputPartitionValidationTest (line 2089) | class TPUEstimatorInputPartitionValidationTest(test.TestCase):
    method _train (line 2091) | def _train(self,
    method test_train_with_non_positive_dims (line 2121) | def test_train_with_non_positive_dims(self):
    method test_train_with_unmatched_partition_dims (line 2135) | def test_train_with_unmatched_partition_dims(self):
    method test_train_with_shape_unmatched_partition_dims (line 2150) | def test_train_with_shape_unmatched_partition_dims(self):
    method test_train_with_unmatched_feature_keys (line 2164) | def test_train_with_unmatched_feature_keys(self):
    method test_train_with_unmatched_label_keys (line 2179) | def test_train_with_unmatched_label_keys(self):
    method test_train_uneven_partitions_successful (line 2196) | def test_train_uneven_partitions_successful(self):
    method test_uneven_partitions_computation (line 2209) | def test_uneven_partitions_computation(self):
    method _test_input_partitions_with_nested_label (line 2242) | def _test_input_partitions_with_nested_label(self, input_partition_dims):
    method test_fully_specified_input_partitions_with_with_nested_label (line 2297) | def test_fully_specified_input_partitions_with_with_nested_label(self):
    method test_partial_specified_input_partitions_with_nested_label (line 2300) | def test_partial_specified_input_partitions_with_nested_label(self):
    method test_incorrect_input_partitions_with_nested_label (line 2311) | def test_incorrect_input_partitions_with_nested_label(self):
  class TPUEstimatorInputPipelinePlacementTest (line 2323) | class TPUEstimatorInputPipelinePlacementTest(test.TestCase):
    method _test_placement (line 2325) | def _test_placement(self, per_host):
    method _query_system (line 2388) | def _query_system(self, master_address, cluster_def, query_topology):
    method test_per_host_placement (line 2397) | def test_per_host_placement(self):
    method test_per_core_placement (line 2404) | def test_per_core_placement(self):
  class TPUEstimatorScaffoldTest (line 2412) | class TPUEstimatorScaffoldTest(test.TestCase):
    method _get_scaffold_fn (line 2414) | def _get_scaffold_fn(self, mode):
    method _input_fn (line 2430) | def _input_fn(self, params):
    method _predict_input_fn (line 2433) | def _predict_input_fn(self, params):
    method _model_fn (line 2437) | def _model_fn(self, features, labels, mode, config, params):
    method test_train (line 2471) | def test_train(self):
    method test_eval (line 2482) | def test_eval(self):
    method test_predict (line 2498) | def test_predict(self):
    method test_scaffold_fn_capture_tpu_tensor (line 2513) | def test_scaffold_fn_capture_tpu_tensor(self):
    method test_scaffold_capture_tpu_tensor (line 2542) | def test_scaffold_capture_tpu_tensor(self):
  class TPUEstimatorScaffoldWithEMATest (line 2574) | class TPUEstimatorScaffoldWithEMATest(test.TestCase):
    method _get_scaffold (line 2576) | def _get_scaffold(self, ema):
    method _input_fn (line 2580) | def _input_fn(self, params):
    method _model_fn (line 2583) | def _model_fn(self, features, labels, mode, config, params):
    method test_ema_with_train_and_evaluate (line 2624) | def test_ema_with_train_and_evaluate(self):
  class _ModelVariableValueHook (line 2656) | class _ModelVariableValueHook(session_run_hook.SessionRunHook):
    method __init__ (line 2659) | def __init__(self, scope_name, variable_name):
    method after_create_session (line 2665) | def after_create_session(self, sess, coord):
  class _RewriteVarHook (line 2672) | class _RewriteVarHook(session_run_hook.SessionRunHook):
    method __init__ (line 2675) | def __init__(self, scope_name, variable_name, value):
    method begin (line 2681) | def begin(self):
    method after_run (line 2685) | def after_run(self, run_context, run_values):
  class TPUEstimatorHostCallTest (line 2689) | class TPUEstimatorHostCallTest(test.TestCase):
    method _input_fn (line 2691) | def _input_fn(self, params):
    method _host_call (line 2694) | def _host_call(self, model_dir, mode):
    method _metric_fn_on_cpu (line 2725) | def _metric_fn_on_cpu(self, labels, predictions):
    method _model_fn (line 2730) | def _model_fn(self, model_dir):
    method _events_from_logdir (line 2755) | def _events_from_logdir(self, logdir):
    method _test_summaries (line 2769) | def _test_summaries(self, use_tpu, output_every_n_steps=False):
    method test_summaries (line 2837) | def test_summaries(self):
    method test_summaries_on_cpu (line 2840) | def test_summaries_on_cpu(self):
    method test_summaries_every_n_steps (line 2843) | def test_summaries_every_n_steps(self):
    method test_summaries_on_cpu_every_n_steps (line 2846) | def test_summaries_on_cpu_every_n_steps(self):
    method test_keras_tensorflow_op_layer (line 2849) | def test_keras_tensorflow_op_layer(self):

FILE: tensorflow_estimator/python/estimator/tpu/util.py
  function check_positive_integer (line 35) | def check_positive_integer(value, name):
  function parse_iterations_per_loop (line 44) | def parse_iterations_per_loop(iterations_per_loop):
  class MultiHostDatasetInitializerHook (line 85) | class MultiHostDatasetInitializerHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 88) | def __init__(self, dataset_initializers):
    method after_create_session (line 91) | def after_create_session(self, session, coord):

FILE: tensorflow_estimator/python/estimator/training.py
  function _validate_input_fn (line 45) | def _validate_input_fn(input_fn):
  function _validate_hooks (line 51) | def _validate_hooks(hooks):
  function _validate_saving_listeners (line 62) | def _validate_saving_listeners(saving_listeners):
  function _validate_exporters (line 74) | def _validate_exporters(exporters):
  function _is_google_env (line 118) | def _is_google_env():
  class TrainSpec (line 128) | class TrainSpec(
    method __new__ (line 151) | def __new__(cls, input_fn, max_steps=None, hooks=None, saving_listener...
  class EvalSpec (line 201) | class EvalSpec(
    method __new__ (line 214) | def __new__(cls,
  function train_and_evaluate (line 297) | def train_and_evaluate(estimator, train_spec, eval_spec):
  class _StopAtSecsHook (line 506) | class _StopAtSecsHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 509) | def __init__(self, stop_after_secs):
    method begin (line 513) | def begin(self):
    method after_run (line 516) | def after_run(self, run_context, run_values):
  class _NewCheckpointListenerForEvaluate (line 522) | class _NewCheckpointListenerForEvaluate(
    method __init__ (line 526) | def __init__(self, evaluator, eval_throttle_secs, continuous_eval_list...
    method begin (line 532) | def begin(self):
    method after_save (line 537) | def after_save(self, session, global_step_value):
    method end (line 561) | def end(self, session, global_step_value):
    method _evaluate (line 568) | def _evaluate(self, global_step_value):
  class _TrainingExecutor (line 579) | class _TrainingExecutor(object):
    method __init__ (line 586) | def __init__(self,
    method estimator (line 619) | def estimator(self):
    method run (line 622) | def run(self):
    method run_chief (line 672) | def run_chief(self):
    method run_worker (line 678) | def run_worker(self):
    method run_master (line 683) | def run_master(self):
    method run_evaluator (line 709) | def run_evaluator(self):
    method run_ps (line 714) | def run_ps(self):
    method run_local (line 720) | def run_local(self):
    method _start_std_server (line 751) | def _start_std_server(self, config):
    method _start_distributed_training (line 795) | def _start_distributed_training(self, saving_listeners=None):
    method _start_continuous_evaluation (line 830) | def _start_continuous_evaluation(self):
    method _execute_evaluator_once (line 862) | def _execute_evaluator_once(self, evaluator, continuous_eval_listener,
    class _Evaluator (line 909) | class _Evaluator(object):
      method __init__ (line 912) | def __init__(self, estimator, eval_spec, max_training_steps):
      method is_final_export_triggered (line 924) | def is_final_export_triggered(self):
      method evaluate_and_export (line 927) | def evaluate_and_export(self):
      method _log_err_msg (line 978) | def _log_err_msg(self, message):
      method _export_eval_result (line 985) | def _export_eval_result(self, eval_result, is_the_final_export):
  class _EvalStatus (line 1005) | class _EvalStatus(object):
  class _EvalResult (line 1027) | class _EvalResult(
    method __new__ (line 1032) | def __new__(cls, status, metrics=None, checkpoint_path=None):
  class _ContinuousEvalListener (line 1089) | class _ContinuousEvalListener(object):
    method before_eval (line 1092) | def before_eval(self):
    method after_eval (line 1101) | def after_eval(self, eval_result):
  function _assert_eval_spec (line 1114) | def _assert_eval_spec(eval_spec):

FILE: tensorflow_estimator/python/estimator/training_test.py
  class _FakeHook (line 150) | class _FakeHook(tf.compat.v1.train.SessionRunHook):
  class _InvalidHook (line 154) | class _InvalidHook(object):
  class _InvalidCheckpointSaverListener (line 158) | class _InvalidCheckpointSaverListener(object):
  function _create_exporter (line 162) | def _create_exporter(name):
  function _create_run_config_with_cluster_spec (line 179) | def _create_run_config_with_cluster_spec(tf_config):
  class TrainSpecTest (line 185) | class TrainSpecTest(tf.test.TestCase):
    method testRequiredArgumentsSet (line 188) | def testRequiredArgumentsSet(self):
    method testAllArgumentsSet (line 195) | def testAllArgumentsSet(self):
    method testInvalidInputFn (line 203) | def testInvalidInputFn(self):
    method testInvalidMaxStep (line 207) | def testInvalidMaxStep(self):
    method testInvalidHook (line 211) | def testInvalidHook(self):
    method testInvalidSavingListener (line 215) | def testInvalidSavingListener(self):
  class EvalSpecTest (line 221) | class EvalSpecTest(tf.test.TestCase):
    method testRequiredArgumentsSet (line 224) | def testRequiredArgumentsSet(self):
    method testAllArgumentsSet (line 235) | def testAllArgumentsSet(self):
    method testListOfExporters (line 256) | def testListOfExporters(self):
    method testInvalidInputFn (line 264) | def testInvalidInputFn(self):
    method testInvalidMaxStep (line 268) | def testInvalidMaxStep(self):
    method testInvalidName (line 272) | def testInvalidName(self):
    method testInvalidHook (line 276) | def testInvalidHook(self):
    method testInvalidDelaySecs (line 280) | def testInvalidDelaySecs(self):
    method testInvalidThrottleSecs (line 284) | def testInvalidThrottleSecs(self):
    method testInvalidTypeOfListOfExporters (line 288) | def testInvalidTypeOfListOfExporters(self):
    method testInvalidTypeOfIndividualExporter (line 294) | def testInvalidTypeOfIndividualExporter(self):
    method testInvalidTypeOfExporterName (line 298) | def testInvalidTypeOfExporterName(self):
    method testMultipleExportersWithTheSameName (line 303) | def testMultipleExportersWithTheSameName(self):
    method testMultipleExportersAndOneWithoutAName (line 310) | def testMultipleExportersAndOneWithoutAName(self):
    method testSingleExporterWithoutAName (line 317) | def testSingleExporterWithoutAName(self):
  class TrainAndEvaluateTest (line 322) | class TrainAndEvaluateTest(tf.test.TestCase):
    method test_run_task (line 324) | def test_run_task(self):
    method test_error_out_if_evaluator_task_id_is_non_zero (line 340) | def test_error_out_if_evaluator_task_id_is_non_zero(self):
    method test_invalid_estimator (line 359) | def test_invalid_estimator(self):
    method test_fail_fast_if_invalid_eval_spec (line 368) | def test_fail_fast_if_invalid_eval_spec(self):
  class TrainingExecutorConstructorTest (line 382) | class TrainingExecutorConstructorTest(tf.test.TestCase):
    method test_required_arguments_set (line 385) | def test_required_arguments_set(self):
    method test_invalid_estimator (line 393) | def test_invalid_estimator(self):
    method test_invalid_train_spec (line 401) | def test_invalid_train_spec(self):
    method test_invalid_eval_spec (line 409) | def test_invalid_eval_spec(self):
    method test_eval_spec_none (line 417) | def test_eval_spec_none(self):
    method test_invalid_train_hooks (line 425) | def test_invalid_train_hooks(self):
    method test_invalid_continuous_eval_listener (line 435) | def test_invalid_continuous_eval_listener(self):
  class _TrainingExecutorTrainingTest (line 449) | class _TrainingExecutorTrainingTest(object):
    method __init__ (line 452) | def __init__(self, run_config):
    method _run_task (line 455) | def _run_task(self, executor):
    method test_train_with_train_spec (line 462) | def test_train_with_train_spec(self, mock_server, unused_mock_sleep):
    method test_train_with_no_eval_spec (line 493) | def test_train_with_no_eval_spec(self, mock_server, unused_mock_sleep):
    method test_train_with_train_hooks (line 524) | def test_train_with_train_hooks(self, unused_mock_server, unused_mock_...
    method test_no_server_startup_in_google (line 544) | def test_no_server_startup_in_google(self, mock_server, unused_mock_sl...
    method test_fail_with_empty_cluster_spec (line 558) | def test_fail_with_empty_cluster_spec(self):
    method test_fail_with_empty_master (line 575) | def test_fail_with_empty_master(self):
    method test_single_worker_node_with_empty_tf_master (line 595) | def test_single_worker_node_with_empty_tf_master(self, mock_server,
    method test_fail_with_empty_task_type (line 615) | def test_fail_with_empty_task_type(self):
    method test_fail_with_none_task_id (line 632) | def test_fail_with_none_task_id(self):
  class TrainingExecutorRunWorkerTest (line 650) | class TrainingExecutorRunWorkerTest(_TrainingExecutorTrainingTest,
    method __init__ (line 654) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method test_delay_for_worker (line 661) | def test_delay_for_worker(self, _):
    method test_delay_disabled_for_worker (line 678) | def test_delay_disabled_for_worker(self, _):
  class TrainingExecutorRunChiefTest (line 694) | class TrainingExecutorRunChiefTest(_TrainingExecutorTrainingTest,
    method __init__ (line 698) | def __init__(self, methodName='runTest'):  # pylint: disable=invalid-name
    method test_no_delay_for_chief (line 705) | def test_no_delay_for_chief(self, _):
  class TrainingExecutorRunMasterTest (line 720) | class TrainingExecutorRunMasterTest(tf.test.TestCase):
    method setUp (line 723) | def setUp(self):
    method test_no_delay_for_master (line 728) | def test_no_delay_for_master(self, _):
    method test_train_with_train_spec (line 750) | def test_train_with_train_spec(self, mock_server, unused_mock_sleep):
    method test_train_with_no_eval_spec_fails (line 784) | def test_train_with_no_eval_spec_fails(self, mock_server, unused_mock_...
    method test_train_with_train_hooks (line 800) | def test_train_with_train_hooks(self, mock_server, unused_mock_sleep):
    method test_no_server_startup_in_google (line 824) | def test_no_server_startup_in_google(self, mock_server, unused_mock_sl...
    method test_fail_with_empty_cluster_spec (line 844) | def test_fail_with_empty_cluster_spec(self):
    method test_fail_with_empty_master (line 863) | def test_fail_with_empty_master(self):
    method test_single_master_node_with_empty_tf_master (line 887) | def test_single_master_node_with_empty_tf_master(self, mock_server,
    method test_fail_with_empty_task_type (line 915) | def test_fail_with_empty_task_type(self):
    method test_fail_with_none_task_id (line 934) | def test_fail_with_none_task_id(self):
    method test_run_master_triggers_evaluate_and_export (line 954) | def test_run_master_triggers_evaluate_and_export(self, _):
    method test_run_master_throttle_eval (line 998) | def test_run_master_throttle_eval(self, _, mock_timer_class):
    method test_run_master_throttle_eval_which_skips_final_ckpt (line 1054) | def test_run_master_throttle_eval_which_skips_final_ckpt(
  class TrainingExecutorRunEvaluatorTest (line 1110) | class TrainingExecutorRunEvaluatorTest(tf.test.TestCase):
    method _set_up_mock_est_to_train_and_evaluate_once (line 1113) | def _set_up_mock_est_to_train_and_evaluate_once(self, mock_est,
    method test_evaluate_with_evaluate_spec (line 1120) | def test_evaluate_with_evaluate_spec(self):
    method test_evaluate_with_no_eval_spec_fails (line 1145) | def test_evaluate_with_no_eval_spec_fails(self):
    method test_evaluate_with_train_hooks (line 1158) | def test_evaluate_with_train_hooks(self):
    method test_evaluate_multiple_times (line 1181) | def test_evaluate_multiple_times(self):
    method test_evaluate_listener_before_eval (line 1226) | def test_evaluate_listener_before_eval(self):
    method test_evaluate_listener_after_eval (line 1266) | def test_evaluate_listener_after_eval(self):
    method test_final_export_is_true_in_the_end (line 1309) | def test_final_export_is_true_in_the_end(self):
    method test_skip_evaluation_due_to_ckpt (line 1351) | def test_skip_evaluation_due_to_ckpt(self):
    method test_warning_if_throttle_secs_is_zero (line 1384) | def test_warning_if_throttle_secs_is_zero(self):
    method test_continuous_eval_listener_eval_result (line 1410) | def test_continuous_eval_listener_eval_result(self):
    method test_sleep_start_delay_secs (line 1474) | def test_sleep_start_delay_secs(self):
    method test_throttle_secs (line 1500) | def test_throttle_secs(self, mock_sleep, mock_time):
    method test_that_export_is_called (line 1520) | def test_that_export_is_called(self):
    method test_errors_out_if_evaluate_returns_empty_dict (line 1546) | def test_errors_out_if_evaluate_returns_empty_dict(self):
    method test_errors_out_if_evaluate_returns_non_dict (line 1557) | def test_errors_out_if_evaluate_returns_non_dict(self):
    method test_errors_out_if_evaluate_returns_dict_without_global_step (line 1568) | def test_errors_out_if_evaluate_returns_dict_without_global_step(self):
  class TrainingExecutorRunPsTest (line 1581) | class TrainingExecutorRunPsTest(tf.test.TestCase):
    method test_std_server (line 1585) | def test_std_server(self, mock_server):
    method test_fail_with_empty_cluster_spec (line 1609) | def test_fail_with_empty_cluster_spec(self):
    method test_fail_with_empty_master (line 1628) | def test_fail_with_empty_master(self):
    method test_fail_with_empty_task_type (line 1647) | def test_fail_with_empty_task_type(self):
    method test_fail_with_none_task_id (line 1664) | def test_fail_with_none_task_id(self):
  class StopAtSecsHookTest (line 1682) | class StopAtSecsHookTest(tf.test.TestCase):
    method test_stops_after_time (line 1686) | def test_stops_after_time(self, mock_time):
  class TrainingExecutorRunLocalTest (line 1708) | class TrainingExecutorRunLocalTest(tf.test.TestCase):
    method _model_fn (line 1711) | def _model_fn(self, features, labels, mode):
    method _input_fn (line 1725) | def _input_fn(self, repeat=True):
    method unique_checkpoint_every_time_fn (line 1731) | def unique_checkpoint_every_time_fn(self):
    method test_runs_evaluate_with_every_new_checkpoint (line 1734) | def test_runs_evaluate_with_every_new_checkpoint(self):
    method test_runs_with_eval_listener_before_eval (line 1771) | def test_runs_with_eval_listener_before_eval(self):
    method test_runs_with_eval_listener_after_eval (line 1801) | def test_runs_with_eval_listener_after_eval(self):
    method test_handles_no_new_checkpoint_found (line 1833) | def test_handles_no_new_checkpoint_found(self):
    method test_final_export_is_true_in_the_end (line 1851) | def test_final_export_is_true_in_the_end(self):
    method test_train_and_evaluate_args (line 1886) | def test_train_and_evaluate_args(self):
    method test_train_with_no_eval_spec_fails (line 1913) | def test_train_with_no_eval_spec_fails(self):
    method test_train_hooks (line 1924) | def test_train_hooks(self):
    method test_that_export_is_called_with_run_local (line 1945) | def test_that_export_is_called_with_run_local(self):
    method test_errors_out_if_evaluate_returns_empty_dict (line 1976) | def test_errors_out_if_evaluate_returns_empty_dict(self):
    method test_errors_out_if_evaluate_returns_non_dict (line 1991) | def test_errors_out_if_evaluate_returns_non_dict(self):
    method test_errors_out_if_evaluate_returns_dict_without_global_step (line 2005) | def test_errors_out_if_evaluate_returns_dict_without_global_step(self):
    method test_train_and_evaluate_return_metrics (line 2021) | def test_train_and_evaluate_return_metrics(self):
  class TrainAndEvaluateRunTest (line 2040) | class TrainAndEvaluateRunTest(tf.test.TestCase):
    method _test_run_task_and_executor (line 2042) | def _test_run_task_and_executor(self, run_config):
    method test_run_chief (line 2068) | def test_run_chief(self):
    method test_run_worker (line 2074) | def test_run_worker(self):
    method test_run_ps (line 2080) | def test_run_ps(self):
    method test_run_evaluator (line 2086) | def test_run_evaluator(self):
    method test_run_local (line 2093) | def test_run_local(self):
    method test_invalid_local_task (line 2099) | def test_invalid_local_task(self):
    method test_unsupported_task_due_to_missing_run_task (line 2120) | def test_unsupported_task_due_to_missing_run_task(self):
    method test_unsupported_task_due_to_not_callable (line 2142) | def test_unsupported_task_due_to_not_callable(self):
    method test_invalid_task_type (line 2165) | def test_invalid_task_type(self):
  class TrainAndEvaluateIntegrationTest (line 2181) | class TrainAndEvaluateIntegrationTest(tf.test.TestCase):
    method setUp (line 2183) | def setUp(self):
    method tearDown (line 2186) | def tearDown(self):
    method _as_label (line 2190) | def _as_label(self, data_in_float):
    method _get_exporter (line 2193) | def _get_exporter(self, name, fc):
    method _extract_loss_and_global_step (line 2200) | def _extract_loss_and_global_step(self, event_folder):
    method test_complete_flow_with_non_distributed_configuration (line 2224) | def test_complete_flow_with_non_distributed_configuration(self):

FILE: tensorflow_estimator/python/estimator/util.py
  function parse_input_fn_result (line 47) | def parse_input_fn_result(result):
  function parse_iterator_result (line 76) | def parse_iterator_result(result):
  class _DatasetInitializerHook (line 86) | class _DatasetInitializerHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 89) | def __init__(self, iterator):
    method begin (line 92) | def begin(self):
    method after_create_session (line 95) | def after_create_session(self, session, coord):
  class DistributedIteratorInitializerHook (line 100) | class DistributedIteratorInitializerHook(tf.compat.v1.train.SessionRunHo...
    method __init__ (line 103) | def __init__(self, iterator):
    method begin (line 106) | def begin(self):
    method after_create_session (line 109) | def after_create_session(self, session, coord):
  class MultiHostDatasetInitializerHook (line 114) | class MultiHostDatasetInitializerHook(tf.compat.v1.train.SessionRunHook):
    method __init__ (line 117) | def __init__(self, dataset_initializers):
    method after_create_session (line 120) | def after_create_session(self, session, coord):

FILE: tensorflow_estimator/python/estimator/util_test.py
  class UtilTest (line 29) | class UtilTest(tf.test.TestCase, parameterized.TestCase):
    method test_parse_input_fn_result_tuple (line 32) | def test_parse_input_fn_result_tuple(self):
    method test_parse_input_fn_result_dataset (line 50) | def test_parse_input_fn_result_dataset(self, dataset_class):
    method test_parse_input_fn_result_features_only (line 66) | def test_parse_input_fn_result_features_only(self):
    method test_parse_input_fn_result_features_only_dataset (line 82) | def test_parse_input_fn_result_features_only_dataset(self, dataset_cla...
    method test_parse_input_fn_result_invalid (line 99) | def test_parse_input_fn_result_invalid(self, dataset_class):

FILE: tensorflow_estimator/tools/pip_package/create_pip_helper.py
  class PipPackagingError (line 48) | class PipPackagingError(Exception):
  function create_init_files (line 52) | def create_init_files(pip_root):
  function verify_python_files_in_pip (line 80) | def verify_python_files_in_pip(pip_root, bazel_root, has_contrib):
  function main (line 114) | def main():