| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 88 partition_strategy="mod", argument
155 if partition_strategy == "mod":
158 elif partition_strategy == "div":
193 partition_strategy)
258 partition_strategy="mod", argument
318 partition_strategy=partition_strategy,
325 partition_strategy=partition_strategy,
402 partition_strategy="mod", argument
512 params, ids, partition_strategy=partition_strategy, max_norm=max_norm)
769 partition_strategy="div",
[all …]
|
| D | nn_impl.py | 1819 partition_strategy="mod", argument
1907 weights, all_ids, partition_strategy=partition_strategy)
1926 biases, all_ids, partition_strategy=partition_strategy)
2095 partition_strategy="div",
2110 partition_strategy="mod", argument
2202 partition_strategy=partition_strategy,
2298 partition_strategy="div",
2314 partition_strategy="mod", argument
2398 partition_strategy=partition_strategy,
|
| D | nn_test.py | 665 partition_strategy="div",
701 partition_strategy="div",
738 partition_strategy="div",
782 partition_strategy="div",
820 partition_strategy="div",
|
| D | clustering_ops.py | 607 return embedding_lookup(self._inputs, indices, partition_strategy='div')
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 192 partition_strategy="mod", argument
208 if partition_strategy == "mod":
211 elif partition_strategy == "div":
404 p, ids, partition_strategy="div")
407 params, id_vals, num_shards, vocab_size, partition_strategy="div")
430 p_variable, ids, partition_strategy="div")
433 params, id_vals, num_shards, vocab_size, partition_strategy="div")
455 p, ids, partition_strategy="div")
458 params, id_vals, num_shards, vocab_size, partition_strategy="div")
483 p, ids, partition_strategy="div")
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.estimator.tpu.experimental.-embedding-config-spec.pbtxt | 27 name: "partition_strategy"
|
| 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/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 664 partition_strategy='mod', argument
671 partition_strategy, name,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 1230 partition_strategy='div', argument
1281 if partition_strategy not in ('div', 'mod'):
1283 'Invalid partition_strategy {}'.format(partition_strategy))
1284 self._partition_strategy = partition_strategy
|
| /external/libaom/libaom/av1/ |
| D | av1.cmake | 189 "${AOM_ROOT}/av1/encoder/partition_strategy.h"
190 "${AOM_ROOT}/av1/encoder/partition_strategy.c"
|
| /external/libaom/ |
| D | Android.bp | 259 "libaom/av1/encoder/partition_strategy.c",
|
| /external/tensorflow/ |
| D | RELEASE.md | 1337 * Update `tf.embedding_lookup` to use `partition_strategy` and `max_norm`
2656 * We now use "div" as the default partition_strategy in
|