diff options
author | 2016-07-14 12:42:51 -0800 | |
---|---|---|
committer | 2016-07-14 13:48:24 -0700 | |
commit | c9811b947737c7d634d0f486fc6e435434f1957d (patch) | |
tree | 773ab0e123ca971d359c7b942fd5dbf4e52f16aa /tensorflow | |
parent | 4f63d69d562053a39b4ea25253a2e571dcec2554 (diff) |
Set 'partitioner' correctly while creating var scope for wide layer weights.
Change: 127468219
Diffstat (limited to 'tensorflow')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/composable_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/composable_model.py b/tensorflow/contrib/learn/python/learn/estimators/composable_model.py index d9739d09d7..20cafc4cb4 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/composable_model.py +++ b/tensorflow/contrib/learn/python/learn/estimators/composable_model.py @@ -168,7 +168,7 @@ class LinearComposableModel(_ComposableModel): max_partitions=self._num_ps_replicas, min_slice_size=64 << 20) with variable_scope.variable_op_scope( - features.values(), "linear", partitioner) as scope: + features.values(), "linear", partitioner=partitioner) as scope: logits, _, _ = layers.weighted_sum_from_feature_columns( columns_to_tensors=features, feature_columns=self._get_feature_columns(), |