Searched refs:max_shards (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | partitioned_variables.py | 73 max_shard_bytes, axis=0, bytes_per_string_element=16, max_shards=None): argument
107 if max_shards and max_shards < 1:
147 if max_shards:
148 axis_shards = min(max_shards, axis_shards)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | partitioned_variables_test.py | 79 max_shards=None): argument
81 axis=axis, max_shard_bytes=max_shard_bytes, max_shards=max_shards)
163 max_shards=33)
172 max_shards=2)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 2397 …argspec: "args=[\'max_shard_bytes\', \'axis\', \'bytes_per_string_element\', \'max_shards\'], vara…
|