Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py373 sp_weights, argument
463 ignore_weights = sp_weights is None
465 if not isinstance(sp_weights, sparse_tensor.SparseTensor):
468 sp_weights.values.get_shape())
470 sp_weights.indices.get_shape())
472 sp_weights.dense_shape.get_shape())
490 weights = sp_weights.values
548 sp_weights, argument
623 params, sp_ids, sp_weights, "div", name, combiner, max_norm)
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dembedding_ops.py587 sp_weights, argument
645 ignore_weights = sp_weights is None
647 if not isinstance(sp_weights, sparse_tensor.SparseTensor):
650 sp_weights.values.get_shape())
652 sp_weights.indices.get_shape())
654 sp_weights.dense_shape.get_shape())
670 weights = None if ignore_weights else sp_weights.values
Dembedding_ops_test.py692 sp_weights = 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 = (
732 None if ignore_weights else sp_weights,
758 sp_ids, sp_weights, _, _, _ = (self._RandomIdsAndWeights(
771 None if ignore_weights else sp_weights,
788 sp_weights = sparse_tensor_lib.SparseTensor(
795 x, sp_ids, sp_weights, combiner="mean")
/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py670 sp_weights = 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 = (
711 None if ignore_weights else sp_weights,
749 sp_ids, sp_weights, _, _, _ = (self._RandomIdsAndWeights(
762 None if ignore_weights else sp_weights,
780 sp_weights = sparse_tensor.SparseTensor(
787 x, sp_ids, sp_weights, combiner="mean")
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt23 aggregation weights. It corresponds to sp_weights.values in
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt153 …argspec: "args=[\'params\', \'sp_ids\', \'sp_weights\', \'combiner\', \'max_norm\', \'name\'], var…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt193 …argspec: "args=[\'params\', \'sp_ids\', \'sp_weights\', \'partition_strategy\', \'name\', \'combin…