diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-11-21 23:55:59 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-22 00:01:14 -0800 |
commit | d0a3b2d3983b970b750329088013dc5cb67d96f9 (patch) | |
tree | 17cc584c4568e4e64a4bdd6bbee0be0b9d96f62c /tensorflow/contrib/linear_optimizer | |
parent | c6d603f02e1a98f871912cda6716cdcbed6b439e (diff) |
Merged commit includes the following changes:
176617057 by yifeif:
Internal change.
--
176615737 by yifeif:
Fix internal tests.
--
PiperOrigin-RevId: 176617057
Diffstat (limited to 'tensorflow/contrib/linear_optimizer')
-rw-r--r-- | tensorflow/contrib/linear_optimizer/python/ops/sdca_ops.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/linear_optimizer/python/ops/sdca_ops.py b/tensorflow/contrib/linear_optimizer/python/ops/sdca_ops.py index 86d8484391..7526f3ae0d 100644 --- a/tensorflow/contrib/linear_optimizer/python/ops/sdca_ops.py +++ b/tensorflow/contrib/linear_optimizer/python/ops/sdca_ops.py @@ -251,8 +251,9 @@ class SdcaModel(object): result_dense = 0.0 for i in range(len(dense_variables)): - result_dense += math_ops.matmul( - dense_features[i], array_ops.expand_dims(dense_variables[i], -1)) + result_dense += math_ops.matmul(dense_features[i], + array_ops.expand_dims( + dense_variables[i], -1)) # Reshaping to allow shape inference at graph construction time. return array_ops.reshape(result_dense, [-1]) + result_sparse |