HomeSort by relevance Sort by last modified time
    Searched refs:max_shard_bytes (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/python/ops/
partitioned_variables.py 73 max_shard_bytes, axis=0, bytes_per_string_element=16, max_shards=None):
74 """Get a partitioner for VariableScope to keep shards below `max_shard_bytes`.
77 the maximum shard size below `max_shard_bytes`. In practice, this is not
83 larger than `max_shard_bytes`. By default `max_shards` equals `None` and no
86 One reasonable value for `max_shard_bytes` is `(64 << 20) - 1`, or almost
90 max_shard_bytes: The maximum size any given shard is allowed to be.
95 over `max_shard_bytes`.
104 if max_shard_bytes < 1 or bytes_per_string_element < 1:
106 "Both max_shard_bytes and bytes_per_string_element must be positive.")
112 """Partitioner that partitions shards to have max_shard_bytes total size
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
partitioned_variables_test.py 72 max_shard_bytes,
77 axis=axis, max_shard_bytes=max_shard_bytes, max_shards=max_shards)
99 # No need to slice: bytes_per_slice * dim0 = 65536 < max_shard_bytes
103 max_shard_bytes=131072,
107 # Slice exactly once: bytes_per_slice * dim1 = 65536 = max_shard_bytes
111 max_shard_bytes=65536,
122 max_shard_bytes=32768,
134 max_shard_bytes=2048,
144 # Slice into max of 32 parts because: max_shard_bytes < bytes_per_slic
    [all...]
  /external/tensorflow/tensorflow/python/training/
localhost_cluster_performance_test.py 89 # max_shard_bytes is 4, shape is 1000*partition_size float32s which should
100 max_shard_bytes=4 * partition_size))

Completed in 465 milliseconds