Home
last modified time | relevance | path

Searched refs:sparse_ids (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py629 sparse_ids, argument
682 sparse_ids,
693 sparse_ids, argument
761 [sparse_ids, sparse_weights]) as scope:
763 original_shape = sparse_ids.dense_shape
765 sparse_ids.dense_shape.get_shape()[0])
769 sparse_ids = sparse_ops.sparse_reshape(sparse_ids, [
775 sparse_weights = sparse_tensor.SparseTensor(sparse_ids.indices,
777 sparse_ids.dense_shape)
780 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights)
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py822 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/
Dfeature_column_v2.py3198 sparse_ids = sparse_tensors.id_tensor
3210 sparse_ids.dense_shape.get_shape()[0])
3218 sparse_ids,
3499 sparse_ids = sparse_tensors.id_tensor
3505 sparse_ids.dense_shape.get_shape()[0])
3513 sparse_ids,
4313 def _prune_invalid_ids(sparse_ids, sparse_weights): argument
4315 is_id_valid = math_ops.greater_equal(sparse_ids.values, 0)
4320 sparse_ids = sparse_ops.sparse_retain(sparse_ids, is_id_valid)
4323 return sparse_ids, sparse_weights
[all …]
Dfeature_column.py2509 sparse_ids = sparse_tensors.id_tensor
2525 sparse_ids.dense_shape.get_shape()[0])
2533 sparse_ids,
2631 sparse_ids = sparse_tensors.id_tensor
2675 sparse_ids.dense_shape.get_shape()[0])
2683 sparse_ids,
/external/tensorflow/tensorflow/examples/saved_model/integration_tests/
Dexport_simple_text_embedding.py90 sparse_ids=tf.SparseTensor(token_ids, token_values, token_dense_shape),
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt253 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt329 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…