Searched refs:sparse_ids (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| D | sparse_ops_test.py | 209 sparse_ids = sparse_ops.indicators_to_sparse_ids(indicators)
215 ), sparse_ids.eval())
222 sparse_ids = sparse_ops.indicators_to_sparse_ids(indicators)
228 ), sparse_ids.eval())
237 sparse_ids = sparse_ops.indicators_to_sparse_ids(indicators)
249 ), sparse_ids.eval())
256 sparse_ids = sparse_ops.indicators_to_sparse_ids(
263 ), sparse_ids.eval())
270 sparse_ids = sparse_ops.indicators_to_sparse_ids(
277 ), sparse_ids.eval())
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops_test.py | 73 sparse_ids = sparse_tensor_lib.SparseTensor(
83 return sparse_ids, sparse_weights
99 sparse_ids = sparse_tensor_lib.SparseTensor(
109 return sparse_ids, sparse_weights
114 sparse_ids, sparse_weights = self._ids_and_weights_2d()
117 embedding_weights, sparse_ids, sparse_weights).eval())
127 sparse_ids, sparse_weights = self._ids_and_weights_2d()
130 embedding_weights, sparse_ids, sparse_weights, default_id=3).eval())
141 sparse_ids, _ = self._ids_and_weights_2d()
144 embedding_weights, sparse_ids, None).eval())
[all …]
|
| D | embedding_ops.py | 49 sparse_ids, argument
123 embedding_weights + [sparse_ids,
126 original_shape = sparse_ids.dense_shape
127 original_rank_dim = sparse_ids.dense_shape.get_shape()[0]
132 sparse_ids = sparse_ops.sparse_reshape(sparse_ids, [
138 sparse_ids.indices,
139 sparse_weights.values, sparse_ids.dense_shape)
142 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights)
145 sparse_ids, is_row_empty = sparse_ops.sparse_fill_empty_rows(sparse_ids,
153 sparse_ids,
[all …]
|
| D | encoders.py | 78 sparse_ids = ids
80 sparse_ids = sparse_ops.dense_to_sparse_tensor(ids)
82 [embeddings], sparse_ids, combiner='mean', default_id=0)
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column.py | 2185 sparse_ids = sparse_tensors.id_tensor
2207 sparse_ids=sparse_ids,
2254 sparse_ids = sparse_tensors.id_tensor
2300 sparse_ids=sparse_ids,
2753 sparse_ids, argument
2818 embedding_weights + [sparse_ids,
2821 original_shape = sparse_ids.dense_shape
2822 original_rank_dim = sparse_ids.dense_shape.get_shape()[0]
2827 sparse_ids = sparse_ops.sparse_reshape(sparse_ids, [
2833 sparse_ids.indices,
[all …]
|