| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column.py | 173 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 …]
|
| D | feature_column_v2.py | 2630 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
2631 del weight_collections
2734 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
2735 del weight_collections
2786 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
2789 del weight_collections
2944 def _old_get_dense_tensor_internal(self, sparse_tensors, weight_collections, argument
2948 if (weight_collections and
2949 ops.GraphKeys.GLOBAL_VARIABLES not in weight_collections):
2950 weight_collections.append(ops.GraphKeys.GLOBAL_VARIABLES)
[all …]
|
| D | feature_column_test.py | 836 }), weight_collections=('my_weights',))
1157 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
1244 weight_collections=None, argument
1290 weight_collections=None, argument
1297 weight_collections,
1422 def _get_dense_tensor(self, inputs, weight_collections=None, argument
1430 def _get_sparse_tensors(self, inputs, weight_collections=None, argument
1698 fc.linear_model(features, [price], weight_collections=['my-vars'])
1712 features, [wire_cast], weight_collections=['my-vars'])
1775 weight_collections=['my-vars'])
[all …]
|
| D | feature_column_v2_test.py | 1303 weight_collections=None, argument
1571 weight_collections=None, argument
1841 fc_old.linear_model(features, [price], weight_collections=['my-vars'])
1854 fc_old.linear_model(features, [wire_cast], weight_collections=['my-vars'])
1917 weight_collections=['my-vars'])
1933 weight_collections=['my-vars'])
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column.py | 132 def _creator(weight_collections, scope): argument
136 weight_collections=weight_collections,
453 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
457 self, inputs, weight_collections, trainable)
462 self, inputs, weight_collections, trainable)
477 self, inputs, weight_collections=None, trainable=None): argument
481 self, inputs, weight_collections, trainable)
486 self, inputs, weight_collections, trainable)
592 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
596 self, inputs, weight_collections, trainable)
[all …]
|
| D | feature_column_v2.py | 509 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
513 self, inputs, weight_collections, trainable)
518 self, inputs, weight_collections, trainable)
564 self, inputs, weight_collections=None, trainable=None): argument
568 self, inputs, weight_collections, trainable)
573 self, inputs, weight_collections, trainable)
978 def _get_dense_tensor(self, inputs, weight_collections=None, trainable=None): argument
985 self)._get_dense_tensor(inputs, weight_collections,
990 self)._get_dense_tensor(inputs, weight_collections,
1010 if (weight_collections and
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.feature_column.pbtxt | 37 …argspec: "args=[\'features\', \'feature_columns\', \'weight_collections\', \'trainable\', \'cols_t…
41 …features\', \'feature_columns\', \'units\', \'sparse_combiner\', \'weight_collections\', \'trainab…
|