diff options
Diffstat (limited to 'tensorflow/contrib/estimator')
| -rw-r--r-- | tensorflow/contrib/estimator/python/estimator/dnn_with_layer_annotations.py | 2 |
1 files changed, 1 insertions, 1 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 5faf0aacfe..6ca7aaf989 100644 --- a/tensorflow/contrib/estimator/python/estimator/dnn_with_layer_annotations.py +++ b/tensorflow/contrib/estimator/python/estimator/dnn_with_layer_annotations.py @@ -151,7 +151,7 @@ def make_input_layer_with_layer_annotations(original_input_layer): # spec and looking at the keys. spec = feature_column_lib.make_parse_example_spec(feature_columns) for key in spec.keys(): - tensor = ops.convert_to_tensor(features[key]) + tensor = ops.convert_to_tensor_or_indexed_slices(features[key]) ops.add_to_collection( LayerAnnotationsCollectionNames.keys( LayerAnnotationsCollectionNames.UNPROCESSED_FEATURES), key) |