| /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 | 104 partition_strategy="mod",
123 partition_strategy: See embedding_lookup.
165 if partition_strategy == "mod":
168 elif partition_strategy == "div":
200 partition_strategy)
265 partition_strategy="mod",
279 the elements of `params` according to the `partition_strategy`.
284 If `partition_strategy` is `"mod"`, we assign each id to partition
289 If `partition_strategy` is `"div"`, we assign ids to partitions in a
301 element must be appropriately sized for the given `partition_strategy`
[all...] |
| nn_impl.py | [all...] |
| nn_test.py | 576 partition_strategy="div",
612 partition_strategy="div",
649 partition_strategy="div",
693 partition_strategy="div",
731 partition_strategy="div",
774 partition_strategy="div")
790 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.
156 partition_strategy=partition_strategy,
360 params, ids, partition_strategy="div")
570 partition_strategy="mod",
585 element must be appropriately sized for the given `partition_strategy`.
591 partition_strategy: A string specifying the partitioning strategy, relevant
656 partition_strategy=partition_strategy,
[all...] |
| embedding_ops_test.py | 604 partition_strategy="mod",
620 if partition_strategy == "mod":
623 elif partition_strategy == "div":
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| embedding_ops_test.py | 183 partition_strategy="mod",
199 if partition_strategy == "mod":
202 elif partition_strategy == "div":
387 p, ids, partition_strategy="div")
390 params, id_vals, num_shards, vocab_size, partition_strategy="div")
412 p_variable, ids, partition_strategy="div")
415 params, id_vals, num_shards, vocab_size, partition_strategy="div")
436 p, ids, partition_strategy="div")
439 params, id_vals, num_shards, vocab_size, partition_strategy="div")
463 p, ids, partition_strategy="div"
[all...] |
| /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 | 610 return embedding_lookup(self._inputs, indices, partition_strategy='div')
[all...] |
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column.py | [all...] |
