Home
last modified time | relevance | path

Searched refs:min_slice_size (Results 1 – 12 of 12) 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.py182 max_partitions=num_shards, min_slice_size=1),
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dcheckpoint_utils_test.py63 max_partitions=5, axis=0, min_slice_size=8 << 10))
227 max_partitions=5, axis=0, min_slice_size=8 << 10))
235 max_partitions=5, axis=0, min_slice_size=8 << 10))
257 max_partitions=5, axis=0, min_slice_size=8 << 10))
/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.py65 max_partitions=5, axis=0, min_slice_size=8 << 10))
272 max_partitions=5, axis=0, min_slice_size=8 << 10))
282 max_partitions=5, axis=0, min_slice_size=16 << 10))
310 max_partitions=5, axis=0, min_slice_size=8 << 10))
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Ddnn_linear_combined.py202 min_slice_size=64 << 20))
290 min_slice_size=64 << 20)
690 min_slice_size=input_layer_min_slice_size))
1006 min_slice_size=input_layer_min_slice_size))
Dcomposable_model.py232 max_partitions=self._num_ps_replicas, min_slice_size=64 << 20)
361 max_partitions=self._num_ps_replicas, min_slice_size=64 << 20))
Ddnn.py148 min_slice_size=input_layer_min_slice_size))
Dlinear.py152 min_slice_size=64 << 20)
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Ddnn_tree_combined_estimator.py167 max_partitions=config.num_ps_replicas, min_slice_size=64 << 20))
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column_ops_test.py2121 max_partitions=10, min_slice_size=((64 << 20) - 1))) as scope:
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1629 …argspec: "args=[\'max_partitions\', \'axis\', \'min_slice_size\', \'bytes_per_string_element\'], v…