| /external/tensorflow/tensorflow/contrib/estimator/python/estimator/ |
| linear.py | 90 partitioner=None):
105 partitioner: Optional. Partitioner for input layer.
115 partitioner=partitioner,
|
| dnn_linear_combined_test.py | 100 partitioner=None):
107 input_layer_partitioner=partitioner,
|
| /external/tensorflow/tensorflow/python/estimator/canned/ |
| linear.py | 115 partitioner, config):
128 partitioner: Partitioner for variables.
146 partitioner = partitioner or (
154 partitioner=partitioner):
249 partitioner=None,
282 partitioner: Optional. Partitioner for input layer
[all...] |
| dnn.py | 61 input_layer_partitioner: Partitioner for input layer.
90 partitioner=input_layer_partitioner):
148 input_layer_partitioner: Partitioner for input layer. Defaults
166 partitioner = partitioned_variables.min_max_variable_partitioner(
171 partitioner=partitioner):
319 input_layer_partitioner: Optional. Partitioner for input layer. Defaults
473 input_layer_partitioner: Optional. Partitioner for input layer. Defaults
|
| dnn_linear_combined.py | 116 input_layer_partitioner: Partitioner for input layer.
159 partitioner=dnn_partitioner):
182 partitioner=input_layer_partitioner) as scope:
354 input_layer_partitioner: Partitioner for input layer. Defaults to
528 input_layer_partitioner: Partitioner for input layer. Defaults to
|
| /external/tensorflow/tensorflow/python/estimator/ |
| warm_starting_util_test.py | 59 partitioner=None):
66 partitioner=partitioner)
68 if partitioner:
82 def _create_linear_model(self, feature_cols, partitioner):
84 with variable_scope.variable_scope("", partitioner=partitioner):
116 partitioner=lambda shape, dtype: [2, 1])
137 partitioner=lambda shape, dtype: [2, 1])
152 partitioner=lambda shape, dtype: [2, 1]
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| variable_scope.py | 234 partitioner=None, validate_shape=True, use_resource=None,
251 If a partitioner is provided, a `PartitionedVariable` is returned.
278 partitioner: Optional callable that accepts a fully defined `TensorShape`
311 partitioner was used).
353 partitioner=None, validate_shape=True, use_resource=None,
359 if partitioner is not None and not is_scalar:
360 if not callable(partitioner):
362 "Partitioner must be callable, but received: %s" % partitioner)
373 partitioner=partitioner
966 def partitioner(self): member in class:VariableScope
[all...] |
| partitioned_variables.py | 74 """Get a partitioner for VariableScope to keep shards below `max_shard_bytes`.
76 This partitioner will shard a Variable along one axis, attempting to keep
82 If the partitioner hits the `max_shards` limit, then each shard may end up
98 A partition function usable as the `partitioner` argument to
112 """Partitioner that partitions shards to have max_shard_bytes total size.
160 """Partitioner to allocate minimum size per slice.
162 Returns a partitioner that partitions the variable of given shape and dtype
176 A partition function usable as the `partitioner` argument to
181 """Partitioner that partitions list for a variable of given shape and type.
222 """Partitioner to specify a fixed number of shards along given axis
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| partitioned_variables_test.py | 40 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
41 with variable_scope.variable_scope("root", partitioner=partitioner):
51 partitioner = partitioned_variables.fixed_size_partitioner(4, axis=0)
52 with variable_scope.variable_scope("root", partitioner=partitioner):
59 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
61 "root", partitioner=partitioner, use_resource=True):
76 partitioner = partitioned_variables.variable_axis_size_partitioner
[all...] |
| checkpoint_ops_test.py | 298 def _test_loading_variable_with_max_rows(self, np_value, partitioner,
307 partitioner=partitioner)
374 partitioner=None,
382 partitioner=None,
393 partitioner=None,
401 partitioner=None,
408 partitioner=partitioned_variables.fixed_size_partitioner(3),
417 partitioner=partitioned_variables.fixed_size_partitioner(3),
429 partitioner=partitioned_variables.fixed_size_partitioner(3)
[all...] |
| variable_scope_test.py | [all...] |
| /external/tensorflow/tensorflow/python/training/ |
| checkpoint_ops_test.py | 151 partitioner=partitioned_variables.fixed_size_partitioner(2))
185 partitioner=partitioned_variables.fixed_size_partitioner(2))
223 partitioner=partitioned_variables.fixed_size_partitioner(2))
259 partitioner=partitioned_variables.fixed_size_partitioner(2))
293 partitioner=partitioned_variables.fixed_size_partitioner(2))
339 partitioner=partitioned_variables.fixed_size_partitioner(2))
377 partitioner=partitioned_variables.fixed_size_partitioner(2))
|
| checkpoint_utils_test.py | 63 partitioner=partitioned_variables.min_max_variable_partitioner(
241 partitioner=partitioned_variables.min_max_variable_partitioner(
251 partitioner=partitioned_variables.min_max_variable_partitioner(
279 partitioner=partitioned_variables.min_max_variable_partitioner(
|
| localhost_cluster_performance_test.py | 99 partitioner=partitioned_variables.variable_axis_size_partitioner(
|
| slot_creator.py | 56 current_partitioner = variable_scope.get_variable_scope().partitioner
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| composable_model.py | 220 partitioner = partitioned_variables.min_max_variable_partitioner(
224 partitioner=partitioner) as scope:
350 partitioner=input_layer_partitioner) as scope:
365 partitioner=hidden_layer_partitioner) as scope:
380 partitioner=hidden_layer_partitioner) as scope:
|
| dnn.py | 134 partitioner = partitioned_variables.min_max_variable_partitioner(
139 partitioner=partitioner):
147 partitioner=input_layer_partitioner) as input_layer_scope:
[all...] |
| linear.py | 145 partitioner = partitioned_variables.min_max_variable_partitioner(
152 partitioner=partitioner) as scope:
[all...] |
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| variables.py | 199 partitioner=None, custom_getter=None, use_resource=None):
219 partitioner: Optional callable that accepts a fully defined `TensorShape`
245 partitioner=partitioner,
252 caching_device=None, device=None, partitioner=None,
274 partitioner: Optional callable that accepts a fully defined `TensorShape`
290 partitioner=partitioner, custom_getter=custom_getter,
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| layers.py | 47 partitioner=None,
66 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| pruning_test.py | 149 partitioner = partitioned_variables.variable_axis_size_partitioner(40)
151 with variable_scope.variable_scope("", partitioner=partitioner):
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| checkpoint_utils_test.py | 62 partitioner=partitioned_variables.min_max_variable_partitioner(
226 partitioner=partitioned_variables.min_max_variable_partitioner(
234 partitioner=partitioned_variables.min_max_variable_partitioner(
256 partitioner=partitioned_variables.min_max_variable_partitioner(
|
| /external/tensorflow/tensorflow/python/eager/ |
| graph_callable.py | 111 # TODO(apassos) ignoring the regularizer and partitioner here; figure out
116 partitioner=None, validate_shape=True,
118 del getter, regularizer, partitioner, validate_shape, use_resource, dtype
137 # TODO(apassos) ignoring the regularizer and partitioner here; figure out
142 partitioner=None, validate_shape=True,
144 del getter, regularizer, collections, caching_device, partitioner
|
| /external/tensorflow/tensorflow/python/layers/ |
| normalization.py | 304 partitioner = self._scope.partitioner
307 partitioner = None
357 if partitioner:
358 self._scope.set_partitioner(partitioner)
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| dnn_tree_combined_estimator.py | 98 dnn_input_layer_partitioner: Partitioner for input layer of the DNN.
132 partitioner=dnn_partitioner):
137 partitioner=dnn_partitioner) as input_layer_scope:
313 dnn_input_layer_partitioner: Partitioner for input layer of the DNN.
402 dnn_input_layer_partitioner: Partitioner for input layer of the DNN.
491 dnn_input_layer_partitioner: Partitioner for input layer of the DNN.
|
