Searched refs:max_shard_bytes (Results 1 – 4 of 4) 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/training/ |
| D | localhost_cluster_performance_test.py | 102 max_shard_bytes=4 * partition_size))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 2397 …argspec: "args=[\'max_shard_bytes\', \'axis\', \'bytes_per_string_element\', \'max_shards\'], vara…
|