Searched refs:bytes_per_string (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | sharded_variable.py | 139 bytes_per_string=16): argument
154 if bytes_per_string < 1:
159 self._bytes_per_string = bytes_per_string
198 def __init__(self, max_shard_bytes, max_shards=None, bytes_per_string=16): argument
214 if bytes_per_string < 1:
220 self._bytes_per_string = bytes_per_string
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.distribute.experimental.partitioners.-min-size-partitioner.pbtxt | 8 …argspec: "args=[\'self\', \'min_shard_bytes\', \'max_shards\', \'bytes_per_string\'], varargs=None…
|
| H A D | tensorflow.distribute.experimental.partitioners.-max-size-partitioner.pbtxt | 8 …argspec: "args=[\'self\', \'max_shard_bytes\', \'max_shards\', \'bytes_per_string\'], varargs=None…
|