| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 372 sp_ids, argument
461 if not isinstance(sp_ids, sparse_tensor.SparseTensor):
467 sp_ids.values.get_shape().assert_is_compatible_with(
469 sp_ids.indices.get_shape().assert_is_compatible_with(
471 sp_ids.dense_shape.get_shape().assert_is_compatible_with(
477 params + [sp_ids]) as name:
478 segment_ids = sp_ids.indices[:, 0]
482 ids = sp_ids.values
547 sp_ids, argument
623 params, sp_ids, sp_weights, "div", name, combiner, max_norm)
|
| D | sparse_ops.py | 1517 def sparse_merge(sp_ids, sp_values, vocab_size, name=None, argument
1610 if isinstance(sp_ids, sparse_tensor.SparseTensorValue) or isinstance(
1611 sp_ids, sparse_tensor.SparseTensor):
1612 sp_ids = [sp_ids]
1619 if not isinstance(sp_ids, collections.Iterable):
1621 "Found %s" % type(sp_ids))
1630 if len(sp_ids) != len(vocab_size):
1633 with ops.name_scope(name, "SparseMerge", [sp_ids, sp_values]):
1634 sp_ids = [_convert_to_sparse_tensor(sp_ids_dim) for sp_ids_dim in sp_ids]
1637 for sp_ids_dim in sp_ids:
[all …]
|
| D | parsing_ops.py | 332 sp_ids = tensor_dict[feature.index_key]
334 sp_ids = [tensor_dict[index_key] for index_key in feature.index_key]
337 sp_ids,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 586 sp_ids, argument
643 if not isinstance(sp_ids, sparse_tensor.SparseTensor):
649 sp_ids.values.get_shape().assert_is_compatible_with(
651 sp_ids.indices.get_shape().assert_is_compatible_with(
653 sp_ids.dense_shape.get_shape().assert_is_compatible_with(
659 params + [sp_ids]) as name:
660 segment_ids = sp_ids.indices[:, 0]
664 ids = sp_ids.values
|
| D | embedding_ops_test.py | 688 sp_ids = sparse_tensor_lib.SparseTensor(
697 return sp_ids, sp_weights, ids, weights, vals_per_batch_entry
713 sp_ids, sp_weights, ids, weights, vals_per_batch_entry = (
731 sp_ids,
758 sp_ids, sp_weights, _, _, _ = (self._RandomIdsAndWeights(
770 sp_ids,
784 sp_ids = sparse_tensor_lib.SparseTensor(
795 x, sp_ids, sp_weights, combiner="mean")
|
| D | feature_column.py | 994 weighted_column = sparse_ops.sparse_merge(sp_ids=sparse_id_column,
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| D | lookup_ops_test.py | 1371 sp_ids = table.lookup(sp_features)
1373 self.assertAllEqual([5], sp_ids.values._shape_as_list())
1376 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
1400 sp_ids = table.lookup(sp_features)
1402 self.assertAllEqual([5], sp_ids.values._shape_as_list())
1405 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
1429 sp_ids = table.lookup(sp_features)
1431 self.assertAllEqual([5], sp_ids.values._shape_as_list())
1434 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | lookup_ops_test.py | 869 sp_ids = table.lookup(sp_features)
871 self.assertAllEqual([5], sp_ids.values._shape_as_list())
874 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
895 sp_ids = table.lookup(sp_features)
897 self.assertAllEqual([5], sp_ids.values._shape_as_list())
900 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
918 sp_ids = table.lookup(sp_features)
920 self.assertAllEqual([5], sp_ids.values._shape_as_list())
923 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
2411 sp_ids = table.lookup(sp_features)
[all …]
|
| D | embedding_ops_test.py | 666 sp_ids = sparse_tensor.SparseTensor(
675 return sp_ids, sp_weights, ids, weights, vals_per_batch_entry
692 sp_ids, sp_weights, ids, weights, vals_per_batch_entry = (
710 sp_ids,
749 sp_ids, sp_weights, _, _, _ = (self._RandomIdsAndWeights(
761 sp_ids,
776 sp_ids = sparse_tensor.SparseTensor(
787 x, sp_ids, sp_weights, combiner="mean")
|
| D | sparse_ops_test.py | 223 sp_ids=ids, sp_values=values, vocab_size=5)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 9 belong. It corresponds to sp_ids.indices[:,0] in embedding_lookup_sparse().
16 It corresponds to sp_ids.values in embedding_lookup_sparse().
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.pbtxt | 53 …argspec: "args=[\'sp_ids\', \'sp_values\', \'vocab_size\', \'name\', \'already_sorted\'], varargs=…
|
| D | tensorflow.nn.pbtxt | 193 …argspec: "args=[\'params\', \'sp_ids\', \'sp_weights\', \'partition_strategy\', \'name\', \'combin…
|
| D | tensorflow.pbtxt | 2089 …argspec: "args=[\'sp_ids\', \'sp_values\', \'vocab_size\', \'name\', \'already_sorted\'], varargs=…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 153 …argspec: "args=[\'params\', \'sp_ids\', \'sp_weights\', \'combiner\', \'max_norm\', \'name\'], var…
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column.py | 3023 sp_ids=id_tensor,
|
| D | feature_column_v2.py | 4189 sp_ids=id_tensor,
|