diff options
| author |  Rohan Jain <rohanj@google.com> | 2018-09-19 08:19:07 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-19 08:23:13 -0700 |
| commit | e00d7744dbab5c73e4d8ffa8a7d361f7b2dcefff (patch) | |
| tree | f2bf2e8fd776e11de859b6f1303db0d3426c7a8c /tensorflow/contrib/estimator | |
| parent | f8655c08cfe3bd99ec1703211e1c9154a14a6150 (diff) | |
Automated rollback of commit 9fe177881224571aff0c267593f747f5fd7a2967
PiperOrigin-RevId: 213630404
Diffstat (limited to 'tensorflow/contrib/estimator')
| -rw-r--r-- | tensorflow/contrib/estimator/python/estimator/dnn_with_layer_annotations.py | 15 |
1 files changed, 6 insertions, 9 deletions
diff --git a/tensorflow/contrib/estimator/python/estimator/dnn_with_layer_annotations.py b/tensorflow/contrib/estimator/python/estimator/dnn_with_layer_annotations.py index a8eeff6f6d..152431d1b2 100644 --- a/tensorflow/contrib/estimator/python/estimator/dnn_with_layer_annotations.py +++ b/tensorflow/contrib/estimator/python/estimator/dnn_with_layer_annotations.py @@ -76,7 +76,6 @@ def make_input_layer_with_layer_annotations(original_input_layer, mode): weight_collections=None, trainable=True, cols_to_vars=None, - scope=None, cols_to_output_tensors=None): """Returns a dense `Tensor` as input layer based on given `feature_columns`. @@ -113,7 +112,6 @@ def make_input_layer_with_layer_annotations(original_input_layer, mode): 'some_variable:0' shape=(5, 10), <tf.Variable 'some_variable:1' shape=(5, 10)]} If a column creates no variables, its value will be an empty list. - scope: A name or variable scope to use cols_to_output_tensors: If not `None`, must be a dictionary that will be filled with a mapping from '_FeatureColumn' to the associated output `Tensor`s. @@ -134,7 +132,6 @@ def make_input_layer_with_layer_annotations(original_input_layer, mode): weight_collections=weight_collections, trainable=trainable, cols_to_vars=cols_to_vars, - scope=scope, cols_to_output_tensors=local_cols_to_output_tensors) if cols_to_output_tensors is not None: @@ -304,9 +301,9 @@ def DNNClassifierWithLayerAnnotations( # pylint: disable=invalid-name def _model_fn(features, labels, mode, config): with _monkey_patch( - feature_column_lib, '_internal_input_layer', - make_input_layer_with_layer_annotations( - feature_column_lib._internal_input_layer, mode)): # pylint: disable=protected-access + feature_column_lib, 'input_layer', + make_input_layer_with_layer_annotations(feature_column_lib.input_layer, + mode)): return original.model_fn(features, labels, mode, config) return estimator.Estimator( @@ -425,9 +422,9 @@ def DNNRegressorWithLayerAnnotations( # pylint: disable=invalid-name def _model_fn(features, labels, mode, config): with _monkey_patch( - feature_column_lib, '_internal_input_layer', - make_input_layer_with_layer_annotations( - feature_column_lib._internal_input_layer, mode)): # pylint: disable=protected-access + feature_column_lib, 'input_layer', + make_input_layer_with_layer_annotations(feature_column_lib.input_layer, + mode)): return original.model_fn(features, labels, mode, config) return estimator.Estimator( |