Home
last modified time | relevance | path

Searched defs:max_shards (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dpartitioned_variables.py73 max_shard_bytes, axis=0, bytes_per_string_element=16, max_shards=None): argument
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py137 max_shards=1, argument
196 def __init__(self, max_shard_bytes, max_shards=None, bytes_per_string=16): argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dpartitioned_variables_test.py79 max_shards=None): argument