Searched refs:max_shards (Results 1 – 8 of 8) 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/distribute/ |
| D | sharded_variable.py | 137 max_shards=1, argument
150 if max_shards < 1:
151 raise ValueError('max_shards must be positive, got: %r' % max_shards)
156 self._max_shards = max_shards
196 def __init__(self, max_shard_bytes, max_shards=None, bytes_per_string=16): argument
209 if max_shards and max_shards < 1:
210 raise ValueError('max_shards must be positive, got: %r' % max_shards)
216 self._max_shards = max_shards
222 max_shards=self._max_shards,
|
| D | sharded_variable_test.py | 78 min_shard_bytes=4, max_shards=2)
83 min_shard_bytes=4, max_shards=10)
93 max_shard_bytes=4, max_shards=2)
|
| D | parameter_server_strategy_v2_test.py | 231 min_shard_bytes=64 << 20, max_shards=2)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.experimental.partitioners.-min-size-partitioner.pbtxt | 8 …argspec: "args=[\'self\', \'min_shard_bytes\', \'max_shards\', \'bytes_per_string\'], varargs=None…
|
| D | tensorflow.distribute.experimental.partitioners.-max-size-partitioner.pbtxt | 8 …argspec: "args=[\'self\', \'max_shard_bytes\', \'max_shards\', \'bytes_per_string\'], varargs=None…
|
| /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 | 2497 …argspec: "args=[\'max_shard_bytes\', \'axis\', \'bytes_per_string_element\', \'max_shards\'], vara…
|