diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-05-10 11:25:02 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-10 17:39:04 -0700 |
| commit | 9472ba05620332fb274e643e08228b5c530e3e52 (patch) | |
| tree | 068c24a7b40cc2570e09f123490c7af44d000bbd /tensorflow/contrib/tensor_forest/python | |
| parent | 64698dd7bb17dc0796f9635963918059e8a1ebc2 (diff) | |
Automated g4 rollback of changelist 152936284
PiperOrigin-RevId: 155652925
Diffstat (limited to 'tensorflow/contrib/tensor_forest/python')
| -rw-r--r-- | tensorflow/contrib/tensor_forest/python/ops/data_ops.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/tensor_forest/python/ops/data_ops.py b/tensorflow/contrib/tensor_forest/python/ops/data_ops.py index 92f7faf7ac..2e54f620d5 100644 --- a/tensorflow/contrib/tensor_forest/python/ops/data_ops.py +++ b/tensorflow/contrib/tensor_forest/python/ops/data_ops.py @@ -145,8 +145,7 @@ def ParseDataTensorOrDict(data): col_spec.size = shape[1].value dense_features_size += col_spec.size - x = array_ops.reshape(data[k], [-1, 1]) - dense_features.append(CastToFloat(x)) + dense_features.append(CastToFloat(data[k])) processed_dense_features = None processed_sparse_features = None |