OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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