Searched refs:min_slice_size (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | partitioned_variables_test.py | 200 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 …]
|
D | embedding_ops_test.py | 182 max_partitions=num_shards, min_slice_size=1),
|
/external/tensorflow/tensorflow/contrib/framework/python/framework/ |
D | checkpoint_utils_test.py | 63 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/ |
D | partitioned_variables.py | 159 min_slice_size=256 << 10, argument 210 partitions = total_size_bytes / min_slice_size
|
/external/tensorflow/tensorflow/python/training/ |
D | checkpoint_utils_test.py | 65 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/ |
D | dnn_linear_combined.py | 202 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))
|
D | composable_model.py | 232 max_partitions=self._num_ps_replicas, min_slice_size=64 << 20) 361 max_partitions=self._num_ps_replicas, min_slice_size=64 << 20))
|
D | dnn.py | 148 min_slice_size=input_layer_min_slice_size))
|
D | linear.py | 152 min_slice_size=64 << 20)
|
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
D | dnn_tree_combined_estimator.py | 167 max_partitions=config.num_ps_replicas, min_slice_size=64 << 20))
|
/external/tensorflow/tensorflow/contrib/layers/python/layers/ |
D | feature_column_ops_test.py | 2121 max_partitions=10, min_slice_size=((64 << 20) - 1))) as scope:
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.pbtxt | 1629 …argspec: "args=[\'max_partitions\', \'axis\', \'min_slice_size\', \'bytes_per_string_element\'], v…
|