Searched refs:sp_weights (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 401 sp_weights, argument
491 ignore_weights = sp_weights is None
493 if not isinstance(sp_weights, sparse_tensor.SparseTensor):
496 sp_weights.values.get_shape())
498 sp_weights.indices.get_shape())
500 sp_weights.dense_shape.get_shape())
519 weights = sp_weights.values
584 sp_weights, argument
667 return embedding_lookup_sparse(params, sp_ids, sp_weights, "div", name,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 675 sp_weights = sparse_tensor.SparseTensor(
680 return sp_ids, sp_weights, ids, weights, vals_per_batch_entry
697 sp_ids, sp_weights, ids, weights, vals_per_batch_entry = (
716 None if ignore_weights else sp_weights,
754 sp_ids, sp_weights, _, _, _ = (self._RandomIdsAndWeights(
767 None if ignore_weights else sp_weights,
785 sp_weights = sparse_tensor.SparseTensor(
792 x, sp_ids, sp_weights, combiner="mean")
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 23 aggregation weights. It corresponds to sp_weights.values in
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable_test.py | 429 sp_weights = sparse_tensor.SparseTensor(
434 sv, sp_ids, sp_weights)
|
| /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/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 193 …argspec: "args=[\'params\', \'sp_ids\', \'sp_weights\', \'partition_strategy\', \'name\', \'combin…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3851 aggregation weights. It corresponds to sp_weights.values in
|