Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py173 weight_collections=None, argument
188 weight_collections = list(weight_collections or [])
189 if ops.GraphKeys.GLOBAL_VARIABLES not in weight_collections:
190 weight_collections.append(ops.GraphKeys.GLOBAL_VARIABLES)
191 if ops.GraphKeys.MODEL_VARIABLES not in weight_collections:
192 weight_collections.append(ops.GraphKeys.MODEL_VARIABLES)
204 weight_collections=weight_collections,
236 weight_collections=None, argument
299 weight_collections=weight_collections,
314 weight_collections=None, argument
[all …]
Dfeature_column_v2.py2803 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
2804 del weight_collections
2903 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
2904 del weight_collections
2955 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
2958 del weight_collections
3078 def _old_get_dense_tensor_internal(self, sparse_tensors, weight_collections, argument
3082 if (weight_collections and
3083 ops.GraphKeys.GLOBAL_VARIABLES not in weight_collections):
3084 weight_collections.append(ops.GraphKeys.GLOBAL_VARIABLES)
[all …]
Dfeature_column_test.py853 }), weight_collections=('my_weights',))
1184 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
1272 weight_collections=None, argument
1318 weight_collections=None, argument
1325 weight_collections,
1450 def _get_dense_tensor(self, inputs, weight_collections=None, argument
1458 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
1726 fc.linear_model(features, [price], weight_collections=['my-vars'])
1740 features, [wire_cast], weight_collections=['my-vars'])
1803 weight_collections=['my-vars'])
[all …]
Dfeature_column_v2_test.py1490 weight_collections=None, argument
2460 weight_collections=None, argument
2730 fc_old.linear_model(features, [price], weight_collections=['my-vars'])
2743 fc_old.linear_model(features, [wire_cast], weight_collections=['my-vars'])
2806 weight_collections=['my-vars'])
2822 weight_collections=['my-vars'])
/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
1024 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
1025 del weight_collections
1173 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
1177 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.py3192 weight_collections=(ops.GraphKeys.WEIGHTS,), argument
3277 weight_collections = set(
3278 list(weight_collections or []) + [ops.GraphKeys.GLOBAL_VARIABLES])
3284 collections=weight_collections,
Dlayers_test.py4053 weight_collections=['unbiased'],
4066 weight_collections=['unbiased', 'all'],
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py93 def _creator(weight_collections, scope): argument
97 weight_collections=weight_collections,
279 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
283 self, inputs, weight_collections, trainable)
288 self, inputs, weight_collections, trainable)
377 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
381 self, inputs, weight_collections, trainable)
386 self, inputs, weight_collections, trainable)
/external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/
Dsequence_feature_column.py42 weight_collections=None, argument
120 weight_collections=weight_collections,
490 self, inputs, weight_collections=None, trainable=None): argument
493 del weight_collections
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Ddynamic_rnn_estimator.py174 weight_collections=None, argument
201 weight_collections=weight_collections,
207 weight_collections=weight_collections,
Dcomposable_model.py241 weight_collections=[self._scope],
249 weight_collections=[self._scope],
369 weight_collections=[self._scope],
Ddnn_linear_combined.py246 weight_collections=[dnn_parent_scope],
252 weight_collections=[dnn_parent_scope])
302 weight_collections=[linear_parent_scope],
309 weight_collections=[linear_parent_scope],
316 weight_collections=[linear_parent_scope])
Ddnn.py160 weight_collections=[parent_scope],
166 weight_collections=[parent_scope])
Dlinear.py168 weight_collections=[parent_scope],
175 weight_collections=[parent_scope])
Dstate_saving_rnn_estimator.py369 weight_collections=None,
/external/tensorflow/tensorflow/tools/api/golden/v1/
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.py187 weight_collections=[dnn_parent_scope])
192 weight_collections=[dnn_parent_scope],
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py201 features, [fc], weight_collections=[scope])