| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| embedding_ops.py | 50 sparse_weights=None,
79 sparse_weights: `SparseTensor` of same shape as `sparse_ids`, containing
112 dtype = sparse_weights.dtype if sparse_weights is not None else None
120 [sparse_weights])
124 sparse_weights]) as scope:
136 if sparse_weights is not None:
137 sparse_weights = sparse_tensor.SparseTensor(
139 sparse_weights.values, sparse_ids.dense_shape)
142 sparse_ids, sparse_weights = _prune_invalid_ids(sparse_ids, sparse_weights
[all...] |
| embedding_ops_test.py | 78 sparse_weights = sparse_tensor_lib.SparseTensor(
83 return sparse_ids, sparse_weights
104 sparse_weights = sparse_tensor_lib.SparseTensor(
109 return sparse_ids, sparse_weights
114 sparse_ids, sparse_weights = self._ids_and_weights_2d()
117 embedding_weights, sparse_ids, sparse_weights).eval())
127 sparse_ids, sparse_weights = self._ids_and_weights_2d()
130 embedding_weights, sparse_ids, sparse_weights, default_id=3).eval())
169 sparse_ids, sparse_weights = self._ids_and_weights_2d()
179 embedding_weights, sparse_ids, sparse_weights)
[all...] |
| feature_column_ops.py | 306 sparse_weights=embedding_lookup_arguments.weight_tensor,
354 sparse_weights=None,
[all...] |
| feature_column.py | [all...] |
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column.py | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| sdca_ops_test.cc | 161 std::vector<NodeBuilder::NodeOut> sparse_weights; local
165 sparse_weights.push_back(NodeBuilder::NodeOut(n));
221 .Input(sparse_weights)
|
| sdca_internal.cc | 95 context->input_list("sparse_weights", &sparse_weights_inputs));
160 const FeatureWeightsSparseStorage& sparse_weights = local
161 model_weights.sparse_weights()[j];
169 const float sparse_weight = sparse_weights.nominals(l, feature_index);
172 sparse_weights.deltas(l, feature_index) * num_loss_partitions;
|
| sdca_internal.h | 300 const std::vector<FeatureWeightsSparseStorage>& sparse_weights() const { function in class:tensorflow::sdca::ModelWeights
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| sdca_ops.py | 319 sparse_weights = []
328 sparse_weights.append(array_ops.gather(w, sparse_indices[-1]))
339 sparse_weights,
|
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
