Searched refs:max_partitions (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) 217 max_partitions=100, 228 max_partitions=100, 238 max_partitions=16, 246 max_partitions=1, 257 max_partitions=100, 265 max_partitions=100, 275 max_partitions=100, 283 max_partitions=100, [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 | 158 def min_max_variable_partitioner(max_partitions=1, axis=0, argument 215 max_partitions,
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
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)) 375 max_partitions=self._num_ps_replicas))
|
D | dnn_linear_combined.py | 201 max_partitions=num_ps_replicas, 230 max_partitions=num_ps_replicas)) 289 max_partitions=num_ps_replicas, 689 max_partitions=config.num_ps_replicas if config else 0, 1005 max_partitions=config.num_ps_replicas if config else 0,
|
D | dnn.py | 140 max_partitions=num_ps_replicas) 147 max_partitions=num_ps_replicas,
|
D | linear.py | 151 max_partitions=num_ps_replicas,
|
/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/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…
|