| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 675 sparse_weights=None, argument
765 sparse_weights=sparse_weights,
777 sparse_weights=None, argument
869 dtype = sparse_weights.dtype if sparse_weights is not None else None
878 [sparse_ids, sparse_weights]) as scope:
891 if sparse_weights is not None:
892 sparse_weights = sparse_tensor.SparseTensor(sparse_ids.indices,
893 sparse_weights.values,
897 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights)
899 sparse_ids, sparse_weights = _prune_invalid_weights(
[all …]
|
| D | bincount_ops_test.py | 554 sparse_weights = sparse_tensor.SparseTensor(inp_indices, weight_vals,
561 sparse_inp, sparse_weights, axis=0)))
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 832 sparse_weights = sparse_tensor.SparseTensor(
837 return sparse_ids, sparse_weights
858 sparse_weights = sparse_tensor.SparseTensor(
863 return sparse_ids, sparse_weights
869 sparse_ids, sparse_weights = self._ids_and_weights_2d()
874 sparse_weights))
885 sparse_ids, sparse_weights = self._ids_and_weights_2d()
889 embedding_weights, sparse_ids, sparse_weights, default_id=3))
933 sparse_ids, sparse_weights = self._ids_and_weights_2d()
943 embedding_weights, sparse_ids, sparse_weights)
[all …]
|
| /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/python/feature_column/ |
| D | feature_column_v2.py | 2272 sparse_weights=weight_tensor,
2912 sparse_weights = sparse_tensors.weight_tensor
2932 sparse_weights,
3217 sparse_weights = sparse_tensors.weight_tensor
3231 sparse_weights,
4040 def _prune_invalid_ids(sparse_ids, sparse_weights): argument
4043 if sparse_weights is not None:
4046 array_ops.ones_like(sparse_weights.values, dtype=dtypes.bool))
4048 if sparse_weights is not None:
4049 sparse_weights = sparse_ops.sparse_retain(sparse_weights, is_id_valid)
[all …]
|
| D | feature_column.py | 2095 sparse_weights=weight_tensor,
2535 sparse_weights = sparse_tensors.weight_tensor
2559 sparse_weights,
2657 sparse_weights = sparse_tensors.weight_tensor
2709 sparse_weights,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SdcaOptimizer.pbtxt | 37 name: "sparse_weights"
155 name: "sparse_weights"
|
| D | SdcaOptimizerV2.pbtxt | 37 name: "sparse_weights"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SdcaOptimizer.pbtxt | 37 name: "sparse_weights"
155 name: "sparse_weights"
|
| D | SdcaOptimizerV2.pbtxt | 37 name: "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/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 437 …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 | 3977 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
3981 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 261 …argspec: "args=[\'embedding_weights\', \'sparse_ids\', \'sparse_weights\', \'combiner\', \'default…
|
| D | tensorflow.raw_ops.pbtxt | 3977 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
3981 …es\', \'example_weights\', \'example_labels\', \'sparse_indices\', \'sparse_weights\', \'dense_wei…
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2.py | 1623 table, inp, sparse_weights=weight, combiner=feature.table.combiner))
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 33368 …hts tf.Output, example_labels tf.Output, sparse_indices []tf.Output, sparse_weights []tf.Output, d…
33379 …eights, example_labels, tf.OutputList(sparse_indices), tf.OutputList(sparse_weights), tf.OutputLis…
38557 …hts tf.Output, example_labels tf.Output, sparse_indices []tf.Output, sparse_weights []tf.Output, d…
38568 …eights, example_labels, tf.OutputList(sparse_indices), tf.OutputList(sparse_weights), tf.OutputLis…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 44867 name: "sparse_weights"
44986 name: "sparse_weights"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 66532 name: "sparse_weights"
66650 name: "sparse_weights"
66769 name: "sparse_weights"
|
| D | ops_history.v0.pbtxt | 29568 name: "sparse_weights"
|