Searched refs:bytes_per_string_element (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | partitioned_variables.py | 69 max_shard_bytes, axis=0, bytes_per_string_element=16, max_shards=None): argument
100 if max_shard_bytes < 1 or bytes_per_string_element < 1:
131 element_size = bytes_per_string_element
158 bytes_per_string_element=16): argument
205 bytes_per_element = bytes_per_string_element
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | sharded_variable.py | 166 bytes_per_string_element=self._bytes_per_string)(shape, dtype)
226 bytes_per_string_element=self._bytes_per_string,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| H A D | partitioned_variables_test.py | 173 bytes_per_string_element=8)
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.pbtxt | 1701 …argspec: "args=[\'max_partitions\', \'axis\', \'min_slice_size\', \'bytes_per_string_element\'], v…
2509 …argspec: "args=[\'max_shard_bytes\', \'axis\', \'bytes_per_string_element\', \'max_shards\'], vara…
|