| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 628 sparse_ids, argument
680 sparse_ids,
691 sparse_ids, argument
759 embedding_weights + [sparse_ids,
762 original_shape = sparse_ids.dense_shape
764 sparse_ids.dense_shape.get_shape()[0])
769 sparse_ids = sparse_ops.sparse_reshape(sparse_ids, [
775 sparse_ids.indices,
776 sparse_weights.values, sparse_ids.dense_shape)
779 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights)
[all …]
|
| /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.py | 49 sparse_ids, argument
123 embedding_weights + [sparse_ids,
126 original_shape = sparse_ids.dense_shape
128 sparse_ids.dense_shape.get_shape()[0]))
133 sparse_ids = sparse_ops.sparse_reshape(sparse_ids, [
139 sparse_ids.indices,
140 sparse_weights.values, sparse_ids.dense_shape)
143 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights)
145 sparse_ids, sparse_weights = _prune_invalid_weights(
146 sparse_ids, sparse_weights)
[all …]
|
| D | embedding_ops_test.py | 77 sparse_ids = sparse_tensor_lib.SparseTensor(
87 return sparse_ids, sparse_weights
103 sparse_ids = sparse_tensor_lib.SparseTensor(
113 return sparse_ids, sparse_weights
118 sparse_ids, sparse_weights = self._ids_and_weights_2d()
121 embedding_weights, sparse_ids, sparse_weights).eval())
131 sparse_ids, sparse_weights = self._ids_and_weights_2d()
134 embedding_weights, sparse_ids, sparse_weights, default_id=3).eval())
145 sparse_ids, _ = self._ids_and_weights_2d()
148 embedding_weights, sparse_ids, None).eval())
[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/kernel_tests/ |
| D | embedding_ops_test.py | 822 sparse_ids = sparse_tensor.SparseTensor(
832 return sparse_ids, sparse_weights
848 sparse_ids = sparse_tensor.SparseTensor(
858 return sparse_ids, sparse_weights
864 sparse_ids, sparse_weights = self._ids_and_weights_2d()
868 embedding_weights, sparse_ids, sparse_weights).eval())
879 sparse_ids, sparse_weights = self._ids_and_weights_2d()
883 embedding_weights, sparse_ids, sparse_weights,
896 sparse_ids, _ = self._ids_and_weights_2d()
900 embedding_weights, sparse_ids, None).eval())
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 3051 sparse_ids = sparse_tensors.id_tensor
3065 sparse_ids=sparse_ids,
3324 sparse_ids = sparse_tensors.id_tensor
3332 sparse_ids=sparse_ids,
4137 def _prune_invalid_ids(sparse_ids, sparse_weights): argument
4139 is_id_valid = math_ops.greater_equal(sparse_ids.values, 0)
4144 sparse_ids = sparse_ops.sparse_retain(sparse_ids, is_id_valid)
4147 return sparse_ids, sparse_weights
4150 def _prune_invalid_weights(sparse_ids, sparse_weights): argument
4154 sparse_ids = sparse_ops.sparse_retain(sparse_ids, is_weights_valid)
[all …]
|
| D | feature_column.py | 2428 sparse_ids = sparse_tensors.id_tensor
2446 sparse_ids=sparse_ids,
2543 sparse_ids = sparse_tensors.id_tensor
2589 sparse_ids=sparse_ids,
|
| /external/tensorflow/tensorflow/examples/saved_model/integration_tests/ |
| D | export_simple_text_embedding.py | 91 sparse_ids=tf.SparseTensor(token_ids, token_values, token_dense_shape),
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 253 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 329 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
|