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…
|