Searched defs:partition_strategy (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 92 partition_strategy="mod", argument
266 partition_strategy="mod", argument
410 partition_strategy="mod", argument
792 partition_strategy="div", argument
956 partition_strategy="mod", argument
|
| D | nn_impl.py | 1835 partition_strategy="mod", argument
2126 partition_strategy="mod", argument
2330 partition_strategy="mod", argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | embedding_ops_test.py | 187 partition_strategy="mod", argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 900 partition_strategy='mod', argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 1310 partition_strategy='div', argument
|