Searched defs:partition_strategy (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 88 partition_strategy="mod", argument
258 partition_strategy="mod", argument
402 partition_strategy="mod", argument
781 partition_strategy="div", argument
945 partition_strategy="mod", argument
|
| D | nn_impl.py | 1819 partition_strategy="mod", argument
2110 partition_strategy="mod", argument
2314 partition_strategy="mod", argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 664 partition_strategy='mod', argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 192 partition_strategy="mod", argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 1230 partition_strategy='div', argument
|