| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 629 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/ |
| D | embedding_ops.py | 50 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 …]
|
| D | embedding_ops_test.py | 82 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 …]
|
| D | feature_column_ops.py | 306 sparse_weights=embedding_lookup_arguments.weight_tensor,
354 sparse_weights=None,
|
| D | feature_column.py | 1287 sparse_weights=weight_tensor,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 827 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/ |
| D | feature_column_v2.py | 2458 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 …]
|
| D | feature_column.py | 2012 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/ |
| D | sdca_ops_test.cc | 161 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()
|
| D | sdca_internal.cc | 166 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()
|
| D | sdca_internal.h | 298 const std::vector<FeatureWeightsSparseStorage>& sparse_weights() const { in sparse_weights() function
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 416 sparse_weights = []
501 sparse_weights.append(batch_gathered_weights)
513 sparse_weights,
532 sparse_weights,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SdcaOptimizerV2.pbtxt | 47 corresponding weights in sparse_weights. This field maybe omitted for the
52 name: "sparse_weights"
|
| D | api_def_SdcaOptimizer.pbtxt | 46 corresponding weights in sparse_weights. This field maybe omitted for the
51 name: "sparse_weights"
|
| /external/tensorflow/tensorflow/examples/saved_model/integration_tests/ |
| D | export_simple_text_embedding.py | 92 sparse_weights=None,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.train.pbtxt | 101 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
|
| D | tensorflow.nn.pbtxt | 253 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
|
| D | tensorflow.raw_ops.pbtxt | 3229 …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/ |
| D | tensorflow.train.pbtxt | 429 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
|
| D | tensorflow.nn.pbtxt | 329 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
|
| D | tensorflow.raw_ops.pbtxt | 3229 …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/ |
| D | wrappers.go | 8490 …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/ |
| D | ops.pbtxt | 33507 name: "sparse_weights"
33626 name: "sparse_weights"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 67822 name: "sparse_weights"
67940 name: "sparse_weights"
68059 name: "sparse_weights"
|
| D | ops_history.v2.pbtxt | 67195 name: "sparse_weights"
67313 name: "sparse_weights"
67432 name: "sparse_weights"
|