Home
last modified time | relevance | path

Searched refs:weight_collections (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column_ops.py96 weight_collections, argument
112 if weight_collections:
113 weight_collections = list(set(list(weight_collections) +
135 weight_collections,
144 weight_collections,
157 weight_collections=None, argument
216 weight_collections,
227 weight_collections=None, argument
263 weight_collections,
275 weight_collections): argument
[all …]
Dfeature_column.py426 weight_collections=None, argument
491 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
493 del weight_collections
822 weight_collections=None, argument
867 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
869 del weight_collections
1020 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
1021 del weight_collections
1169 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
1173 weight_collections, trainable)
[all …]
Dfeature_column_ops_test.py872 features, [embedded_sparse], weight_collections=["my_collection"])
878 features, [embedded_sparse], weight_collections=["my_collection_core"])
1170 weight_collections=["my_collection"])
1192 weight_collections=["my_collection"],
1199 weight_collections=["my_collection"],
1517 weight_collections=["my_collection"])
2625 weight_collections=["my_collection"])
Dlayers.py2988 weight_collections=(ops.GraphKeys.WEIGHTS,), argument
3073 weight_collections = set(
3074 list(weight_collections or []) + [ops.GraphKeys.GLOBAL_VARIABLES])
3080 collections=weight_collections,
Dlayers_test.py3999 weight_collections=['unbiased'],
4012 weight_collections=['unbiased', 'all'],
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py166 weight_collections=None, argument
179 weight_collections = list(weight_collections or [])
180 if ops.GraphKeys.GLOBAL_VARIABLES not in weight_collections:
181 weight_collections.append(ops.GraphKeys.GLOBAL_VARIABLES)
182 if ops.GraphKeys.MODEL_VARIABLES not in weight_collections:
183 weight_collections.append(ops.GraphKeys.MODEL_VARIABLES)
198 weight_collections=weight_collections,
217 weight_collections=None, argument
273 return _internal_input_layer(features, feature_columns, weight_collections,
286 weight_collections=None, argument
[all …]
Dfeature_column_test.py736 }), weight_collections=('my_weights',))
1035 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
1190 def _get_dense_tensor(self, inputs, weight_collections=None, argument
1198 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
1416 fc.linear_model(features, [price], weight_collections=['my-vars'])
1430 features, [wire_cast], weight_collections=['my-vars'])
1493 weight_collections=['my-vars'])
1509 weight_collections=['my-vars'])
2538 }), weight_collections=('my_weights',))
2933 }), weight_collections=('my_weights',))
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Ddynamic_rnn_estimator.py169 weight_collections=None, argument
196 weight_collections=weight_collections,
202 weight_collections=weight_collections,
Dcomposable_model.py230 weight_collections=[self._scope],
238 weight_collections=[self._scope],
354 weight_collections=[self._scope],
Ddnn_linear_combined.py241 weight_collections=[dnn_parent_scope],
247 weight_collections=[dnn_parent_scope])
297 weight_collections=[linear_parent_scope],
304 weight_collections=[linear_parent_scope],
311 weight_collections=[linear_parent_scope])
Ddnn.py155 weight_collections=[parent_scope],
161 weight_collections=[parent_scope])
Dlinear.py163 weight_collections=[parent_scope],
170 weight_collections=[parent_scope])
Dstate_saving_rnn_estimator.py364 weight_collections=None,
/external/tensorflow/tensorflow/tools/api/golden/
Dtensorflow.feature_column.pbtxt37 …argspec: "args=[\'features\', \'feature_columns\', \'weight_collections\', \'trainable\', \'cols_t…
41 …features\', \'feature_columns\', \'units\', \'sparse_combiner\', \'weight_collections\', \'trainab…
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Ddnn_tree_combined_estimator.py141 weight_collections=[dnn_parent_scope],
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py174 weight_collections=[scope])