Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py674 sparse_ids, argument
764 sparse_ids,
776 sparse_ids, argument
878 [sparse_ids, sparse_weights]) as scope:
880 original_shape = sparse_ids.dense_shape
882 sparse_ids.dense_shape.get_shape()[0])
886 sparse_ids = sparse_ops.sparse_reshape(sparse_ids, [
892 sparse_weights = sparse_tensor.SparseTensor(sparse_ids.indices,
894 sparse_ids.dense_shape)
897 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights)
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py827 sparse_ids = sparse_tensor.SparseTensor(
837 return sparse_ids, sparse_weights
853 sparse_ids = sparse_tensor.SparseTensor(
863 return sparse_ids, sparse_weights
869 sparse_ids, sparse_weights = self._ids_and_weights_2d()
873 sparse_ids,
885 sparse_ids, sparse_weights = self._ids_and_weights_2d()
889 embedding_weights, sparse_ids, sparse_weights, default_id=3))
901 sparse_ids, _ = self._ids_and_weights_2d()
905 sparse_ids, None))
[all …]
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py2911 sparse_ids = sparse_tensors.id_tensor
2923 sparse_ids.dense_shape.get_shape()[0])
2931 sparse_ids,
3216 sparse_ids = sparse_tensors.id_tensor
3222 sparse_ids.dense_shape.get_shape()[0])
3230 sparse_ids,
4040 def _prune_invalid_ids(sparse_ids, sparse_weights): argument
4042 is_id_valid = math_ops.greater_equal(sparse_ids.values, 0)
4047 sparse_ids = sparse_ops.sparse_retain(sparse_ids, is_id_valid)
4050 return sparse_ids, sparse_weights
[all …]
Dfeature_column.py2534 sparse_ids = sparse_tensors.id_tensor
2550 sparse_ids.dense_shape.get_shape()[0])
2558 sparse_ids,
2656 sparse_ids = sparse_tensors.id_tensor
2700 sparse_ids.dense_shape.get_shape()[0])
2708 sparse_ids,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt261 …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…