| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| sampling_ops.py | 32 resampling_temperature, partition_strategy):
68 partition_strategy: From `rank_sampled_softmax_loss`.
97 weights, sampled, partition_strategy, transform_fn=logsumexp_logit)
99 embedding_ops.embedding_lookup(biases, sampled, partition_strategy), [-1])
121 partition_strategy,
153 `partition_strategy="div"` for the two losses to be consistent, as in the
164 partition_strategy="div")
198 partition_strategy: A string specifying the partitioning strategy, relevant
215 if partition_strategy not in ("div", "mod"):
217 "unsupported partition_strategy ({})".format(partition_strategy)
[all...] |
| sampling_ops_test.py | 165 partition_strategy='div')
184 partition_strategy='div')
189 r'unsupported partition_strategy \(None\)'):
202 partition_strategy=None)
204 def _testCompareWithNN(self, weights, biases, partition_strategy):
218 partition_strategy=partition_strategy)
229 partition_strategy=partition_strategy)
290 partition_strategy='div'
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| embedding_ops.py | 86 partition_strategy="mod",
105 partition_strategy: See embedding_lookup.
151 if partition_strategy == "mod":
154 elif partition_strategy == "div":
189 partition_strategy)
254 partition_strategy="mod",
268 the elements of `params` according to the `partition_strategy`.
273 If `partition_strategy` is `"mod"`, we assign each id to partition
278 If `partition_strategy` is `"div"`, we assign ids to partitions in a
290 element must be appropriately sized for the given `partition_strategy`
[all...] |
| nn_impl.py | [all...] |
| nn_test.py | 620 partition_strategy="div",
656 partition_strategy="div",
693 partition_strategy="div",
737 partition_strategy="div",
775 partition_strategy="div",
[all...] |
| clustering_ops.py | 607 return embedding_lookup(self._inputs, indices, partition_strategy='div')
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| embedding_ops.py | 54 partition_strategy="div",
87 partition_strategy: A string specifying the partitioning strategy.
160 partition_strategy=partition_strategy,
374 params, ids, partition_strategy="div")
462 def embedding_lookup_unique(params, ids, partition_strategy="mod", name=None):
474 partition_strategy: A string specifying the partitioning strategy, relevant
493 partition_strategy)
588 partition_strategy="mod",
603 element must be appropriately sized for the given `partition_strategy`
[all...] |
| embedding_ops_test.py | 617 partition_strategy="mod",
633 if partition_strategy == "mod":
636 elif partition_strategy == "div":
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| wals_test.py | 56 partition_strategy='div')
59 partition_strategy='div')
62 partition_strategy='div')
65 partition_strategy='div')
|
| factorization_ops_test.py | 57 partition_strategy="div")
61 partition_strategy="div")
65 partition_strategy="div")
69 partition_strategy="div")
[all...] |
| factorization_ops.py | [all...] |
| gmm_ops.py | 88 clusters_init = embedding_lookup(data, indices, partition_strategy='div')
|
| clustering_ops.py | 606 return embedding_lookup(self._inputs, indices, partition_strategy='div')
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| embedding_ops_test.py | 191 partition_strategy="mod",
207 if partition_strategy == "mod":
210 elif partition_strategy == "div":
403 p, ids, partition_strategy="div")
406 params, id_vals, num_shards, vocab_size, partition_strategy="div")
429 p_variable, ids, partition_strategy="div")
432 params, id_vals, num_shards, vocab_size, partition_strategy="div")
454 p, ids, partition_strategy="div")
457 params, id_vals, num_shards, vocab_size, partition_strategy="div")
482 p, ids, partition_strategy="div"
[all...] |
