Home
last modified time | relevance | path

Searched refs:min_slice_size (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dpartitioned_variables_test.py200 def _testMinMaxVariablePartitioner(self, max_partitions, axis, min_slice_size, argument
204 max_partitions=max_partitions, axis=axis, min_slice_size=min_slice_size)
219 min_slice_size=2 << 10,
230 min_slice_size=256 << 10,
240 min_slice_size=256 << 10,
248 min_slice_size=256 << 10,
259 min_slice_size=128 << 10,
267 min_slice_size=512 << 10,
277 min_slice_size=256 << 10,
285 min_slice_size=256 << 10,
[all …]
Dembedding_ops_test.py183 max_partitions=num_shards, min_slice_size=1),
/external/tensorflow/tensorflow/python/ops/
Dpartitioned_variables.py159 min_slice_size=256 << 10, argument
210 partitions = total_size_bytes / min_slice_size
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils_test.py72 max_partitions=5, axis=0, min_slice_size=8 << 10))
279 max_partitions=5, axis=0, min_slice_size=8 << 10))
289 max_partitions=5, axis=0, min_slice_size=16 << 10))
317 max_partitions=5, axis=0, min_slice_size=8 << 10))
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py163 min_slice_size=self._min_shard_bytes,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1693 …argspec: "args=[\'max_partitions\', \'axis\', \'min_slice_size\', \'bytes_per_string_element\'], v…