Home
last modified time | relevance | path

Searched refs:sparse_weights (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py629 sparse_weights=None, argument
681 sparse_weights=sparse_weights,
692 sparse_weights=None, argument
750 dtype = sparse_weights.dtype if sparse_weights is not None else None
760 sparse_weights]) as scope:
773 if sparse_weights is not None:
774 sparse_weights = sparse_tensor.SparseTensor(
776 sparse_weights.values, sparse_ids.dense_shape)
779 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights)
781 sparse_ids, sparse_weights = _prune_invalid_weights(
[all …]
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dembedding_ops.py50 sparse_weights=None, argument
112 dtype = sparse_weights.dtype if sparse_weights is not None else None
120 [sparse_weights])
124 sparse_weights]) as scope:
137 if sparse_weights is not None:
138 sparse_weights = sparse_tensor.SparseTensor(
140 sparse_weights.values, sparse_ids.dense_shape)
143 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights)
145 sparse_ids, sparse_weights = _prune_invalid_weights(
146 sparse_ids, sparse_weights)
[all …]
Dembedding_ops_test.py82 sparse_weights = sparse_tensor_lib.SparseTensor(
87 return sparse_ids, sparse_weights
108 sparse_weights = sparse_tensor_lib.SparseTensor(
113 return sparse_ids, sparse_weights
118 sparse_ids, sparse_weights = self._ids_and_weights_2d()
121 embedding_weights, sparse_ids, sparse_weights).eval())
131 sparse_ids, sparse_weights = self._ids_and_weights_2d()
134 embedding_weights, sparse_ids, sparse_weights, default_id=3).eval())
173 sparse_ids, sparse_weights = self._ids_and_weights_2d()
183 embedding_weights, sparse_ids, sparse_weights)
[all …]
Dfeature_column_ops.py306 sparse_weights=embedding_lookup_arguments.weight_tensor,
354 sparse_weights=None,
Dfeature_column.py1287 sparse_weights=weight_tensor,
/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py827 sparse_weights = sparse_tensor.SparseTensor(
832 return sparse_ids, sparse_weights
853 sparse_weights = 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,
928 sparse_ids, sparse_weights = self._ids_and_weights_2d()
938 embedding_weights, sparse_ids, sparse_weights)
[all …]
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py2458 sparse_weights=weight_tensor,
3052 sparse_weights = sparse_tensors.weight_tensor
3066 sparse_weights=sparse_weights,
3325 sparse_weights = sparse_tensors.weight_tensor
3333 sparse_weights=sparse_weights,
4137 def _prune_invalid_ids(sparse_ids, sparse_weights): argument
4140 if sparse_weights is not None:
4143 array_ops.ones_like(sparse_weights.values, dtype=dtypes.bool))
4145 if sparse_weights is not None:
4146 sparse_weights = sparse_ops.sparse_retain(sparse_weights, is_id_valid)
[all …]
Dfeature_column.py2012 sparse_weights=weight_tensor,
2429 sparse_weights = sparse_tensors.weight_tensor
2447 sparse_weights=sparse_weights,
2544 sparse_weights = sparse_tensors.weight_tensor
2590 sparse_weights=sparse_weights,
/external/tensorflow/tensorflow/core/kernels/
Dsdca_ops_test.cc161 std::vector<NodeBuilder::NodeOut> sparse_weights; in GetGraphs() local
165 sparse_weights.push_back(NodeBuilder::NodeOut(n)); in GetGraphs()
221 .Input(sparse_weights) in GetGraphs()
Dsdca_internal.cc166 const FeatureWeightsSparseStorage& sparse_weights = in ComputeWxAndWeightedExampleNorm() local
167 model_weights.sparse_weights()[j]; in ComputeWxAndWeightedExampleNorm()
175 const float sparse_weight = sparse_weights.nominals(l, feature_index); in ComputeWxAndWeightedExampleNorm()
178 sparse_weights.deltas(l, feature_index) * num_loss_partitions; in ComputeWxAndWeightedExampleNorm()
Dsdca_internal.h298 const std::vector<FeatureWeightsSparseStorage>& sparse_weights() const { in sparse_weights() function
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/
Dsdca_ops.py416 sparse_weights = []
501 sparse_weights.append(batch_gathered_weights)
513 sparse_weights,
532 sparse_weights,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SdcaOptimizerV2.pbtxt47 corresponding weights in sparse_weights. This field maybe omitted for the
52 name: "sparse_weights"
Dapi_def_SdcaOptimizer.pbtxt46 corresponding weights in sparse_weights. This field maybe omitted for the
51 name: "sparse_weights"
/external/tensorflow/tensorflow/examples/saved_model/integration_tests/
Dexport_simple_text_embedding.py92 sparse_weights=None,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.train.pbtxt101 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
Dtensorflow.nn.pbtxt253 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
Dtensorflow.raw_ops.pbtxt3229 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
3233 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.pbtxt429 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
Dtensorflow.nn.pbtxt329 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
Dtensorflow.raw_ops.pbtxt3229 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
3233 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go8490 …hts tf.Output, example_labels tf.Output, sparse_indices []tf.Output, sparse_weights []tf.Output, d…
8501 …eights, example_labels, tf.OutputList(sparse_indices), tf.OutputList(sparse_weights), tf.OutputLis…
16274 …hts tf.Output, example_labels tf.Output, sparse_indices []tf.Output, sparse_weights []tf.Output, d…
16285 …eights, example_labels, tf.OutputList(sparse_indices), tf.OutputList(sparse_weights), tf.OutputLis…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt33507 name: "sparse_weights"
33626 name: "sparse_weights"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt67195 name: "sparse_weights"
67313 name: "sparse_weights"
67432 name: "sparse_weights"
Dops_history.v1.pbtxt67822 name: "sparse_weights"
67940 name: "sparse_weights"
68059 name: "sparse_weights"

12