Searched refs:max_shard_bytes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | partitioned_variables_test.py | 76 max_shard_bytes, argument
81 axis=axis, max_shard_bytes=max_shard_bytes, max_shards=max_shards)
107 max_shard_bytes=131072,
115 max_shard_bytes=65536,
126 max_shard_bytes=32768,
138 max_shard_bytes=2048,
152 max_shard_bytes=1024,
160 max_shard_bytes=1024,
169 max_shard_bytes=1024,
177 max_shard_bytes=32768,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | partitioned_variables.py | 73 max_shard_bytes, axis=0, bytes_per_string_element=16, max_shards=None): argument
104 if max_shard_bytes < 1 or bytes_per_string_element < 1:
142 slices_per_shard = max(1, math.floor(max_shard_bytes / bytes_per_slice))
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 196 def __init__(self, max_shard_bytes, max_shards=None, bytes_per_string=16): argument
206 if max_shard_bytes < 1:
208 max_shard_bytes)
215 self._max_shard_bytes = max_shard_bytes
221 max_shard_bytes=self._max_shard_bytes,
|
| D | sharded_variable_test.py | 88 partitioner = sharded_variable.MaxSizePartitioner(max_shard_bytes=4)
93 max_shard_bytes=4, max_shards=2)
97 partitioner = sharded_variable.MaxSizePartitioner(max_shard_bytes=1024)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| 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/training/ |
| D | localhost_cluster_performance_test.py | 102 max_shard_bytes=4 * partition_size))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 2497 …argspec: "args=[\'max_shard_bytes\', \'axis\', \'bytes_per_string_element\', \'max_shards\'], vara…
|