| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| D | sampling_ops.py | 32 resampling_temperature, partition_strategy): argument
97 weights, sampled, partition_strategy, transform_fn=logsumexp_logit)
99 embedding_ops.embedding_lookup(biases, sampled, partition_strategy), [-1])
121 partition_strategy, argument
215 if partition_strategy not in ("div", "mod"):
217 "unsupported partition_strategy ({})".format(partition_strategy))
232 partition_strategy)
243 partition_strategy=partition_strategy,
255 partition_strategy="mod", argument
332 partition_strategy=partition_strategy,
|
| D | sampling_ops_test.py | 165 partition_strategy='div')
184 partition_strategy='div')
202 partition_strategy=None)
204 def _testCompareWithNN(self, weights, biases, partition_strategy): argument
218 partition_strategy=partition_strategy)
229 partition_strategy=partition_strategy)
290 partition_strategy='div')
301 partition_strategy='div')
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 86 partition_strategy="mod", argument
151 if partition_strategy == "mod":
154 elif partition_strategy == "div":
189 partition_strategy)
254 partition_strategy="mod", argument
313 partition_strategy=partition_strategy,
374 partition_strategy="mod", argument
486 params, ids, partition_strategy=partition_strategy, max_norm=max_norm)
685 partition_strategy="div",
696 partition_strategy='div', argument
[all …]
|
| D | nn_impl.py | 1378 partition_strategy="mod", argument
1466 weights, all_ids, partition_strategy=partition_strategy)
1485 biases, all_ids, partition_strategy=partition_strategy)
1653 partition_strategy="div",
1667 partition_strategy="mod", argument
1759 partition_strategy=partition_strategy,
1854 partition_strategy="div",
1869 partition_strategy="mod", argument
1950 partition_strategy=partition_strategy,
|
| D | nn_test.py | 620 partition_strategy="div",
656 partition_strategy="div",
693 partition_strategy="div",
737 partition_strategy="div",
775 partition_strategy="div",
|
| D | clustering_ops.py | 607 return embedding_lookup(self._inputs, indices, partition_strategy='div')
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 54 partition_strategy="div", argument
160 partition_strategy=partition_strategy,
374 params, ids, partition_strategy="div")
462 def embedding_lookup_unique(params, ids, partition_strategy="mod", name=None): argument
493 partition_strategy)
588 partition_strategy="mod", argument
674 partition_strategy=partition_strategy,
717 partition_strategy="mod", argument
776 if partition_strategy == "mod":
779 elif partition_strategy == "div":
[all …]
|
| D | embedding_ops_test.py | 617 partition_strategy="mod", argument
633 if partition_strategy == "mod":
636 elif partition_strategy == "div":
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 191 partition_strategy="mod", argument
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 …]
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals_test.py | 56 partition_strategy='div')
59 partition_strategy='div')
62 partition_strategy='div')
65 partition_strategy='div')
|
| D | factorization_ops_test.py | 57 partition_strategy="div")
61 partition_strategy="div")
65 partition_strategy="div")
69 partition_strategy="div")
|
| D | factorization_ops.py | 896 right_factors, gather_indices, partition_strategy="div")
926 row_wt, update_indices, partition_strategy="div")
937 col_wt, gather_indices, partition_strategy="div")
|
| D | gmm_ops.py | 88 clusters_init = embedding_lookup(data, indices, partition_strategy='div')
|
| D | clustering_ops.py | 606 return embedding_lookup(self._inputs, indices, partition_strategy='div')
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 189 …argspec: "args=[\'params\', \'ids\', \'partition_strategy\', \'name\', \'validate_indices\', \'max…
193 …argspec: "args=[\'params\', \'sp_ids\', \'sp_weights\', \'partition_strategy\', \'name\', \'combin…
285 …', \'num_true\', \'sampled_values\', \'remove_accidental_hits\', \'partition_strategy\', \'name\']…
329 …s\', \'sparse_weights\', \'combiner\', \'default_id\', \'name\', \'partition_strategy\', \'max_nor…
333 …', \'num_true\', \'sampled_values\', \'remove_accidental_hits\', \'partition_strategy\', \'name\',…
|
| /external/libaom/libaom/av1/ |
| D | av1.cmake | 175 "${AOM_ROOT}/av1/encoder/partition_strategy.h"
176 "${AOM_ROOT}/av1/encoder/partition_strategy.c"
|
| /external/libaom/ |
| D | Android.bp | 218 "libaom/av1/encoder/partition_strategy.c",
|
| /external/tensorflow/ |
| D | RELEASE.md | 47 …* We now use "div" as the default partition_strategy in `tf.nn.safe_embedding_lookup_sparse`, `tf.…
|