| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| sdca_estimator.py | 119 partitioner = params["partitioner"]
141 partitioner=partitioner)
147 partitioner=partitioner) as scope:
220 partitioner=None):
248 partitioner: Variable partitioner for the primal weights (`div`
276 "partitioner": partitioner
[all...] |
| sdca_optimizer.py | 67 data sets. You can also specify a `partitioner` object to partition the primal
78 partitioner=None):
85 self._partitioner = partitioner
115 def partitioner(self): member in class:SDCAOptimizer
185 # If a partitioner was used during variable creation, we will have a
238 # If a partitioner was used during variable creation, we will have a
|
| /external/tensorflow/tensorflow/python/training/ |
| warm_starting_util_test.py | 61 partitioner=None):
68 partitioner=partitioner)
70 if partitioner:
100 def _create_linear_model(self, feature_cols, partitioner):
102 with variable_scope.variable_scope("", partitioner=partitioner):
136 partitioner=lambda shape, dtype: [2, 1])
159 partitioner=lambda shape, dtype: [2, 1])
176 partitioner=lambda shape, dtype: [2, 1]
[all...] |
| checkpoint_ops_test.py | 154 partitioner=partitioned_variables.fixed_size_partitioner(2))
188 partitioner=partitioned_variables.fixed_size_partitioner(2))
226 partitioner=partitioned_variables.fixed_size_partitioner(2))
262 partitioner=partitioned_variables.fixed_size_partitioner(2))
296 partitioner=partitioned_variables.fixed_size_partitioner(2))
342 partitioner=partitioned_variables.fixed_size_partitioner(2))
380 partitioner=partitioned_variables.fixed_size_partitioner(2))
|
| saver_large_partitioned_variable_test.py | 51 partitioner=partitioned_variables.fixed_size_partitioner(4),
|
| checkpoint_utils_test.py | 64 partitioner=partitioned_variables.min_max_variable_partitioner(
271 partitioner=partitioned_variables.min_max_variable_partitioner(
281 partitioner=partitioned_variables.min_max_variable_partitioner(
309 partitioner=partitioned_variables.min_max_variable_partitioner(
|
| localhost_cluster_performance_test.py | 101 partitioner=partitioned_variables.variable_axis_size_partitioner(
|
| slot_creator.py | 56 current_partitioner = variable_scope.get_variable_scope().partitioner
|
| /external/tensorflow/tensorflow/python/ops/ |
| variable_scope.py | 293 partitioner=None,
315 If a partitioner is provided, a `PartitionedVariable` is returned.
344 partitioner: Optional callable that accepts a fully defined `TensorShape`
386 partitioner was used).
445 partitioner=None,
455 if partitioner is not None and not is_scalar:
456 if not callable(partitioner):
458 "Partitioner must be callable, but received: %s" % partitioner)
470 partitioner=partitioner
1107 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.
161 """Partitioner to allocate minimum size per slice.
163 Returns a partitioner that partitions the variable of given shape and dtype
177 A partition function usable as the `partitioner` argument to
182 """Partitioner that partitions list for a variable of given shape and type.
223 """Partitioner to specify a fixed number of shards along given axis
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| partitioned_variables_test.py | 44 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
45 with variable_scope.variable_scope("root", partitioner=partitioner):
55 partitioner = partitioned_variables.fixed_size_partitioner(4, axis=0)
56 with variable_scope.variable_scope("root", partitioner=partitioner):
63 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
65 "root", partitioner=partitioner, use_resource=True):
80 partitioner = partitioned_variables.variable_axis_size_partitioner
[all...] |
| checkpoint_ops_test.py | 303 def _test_loading_variable_with_max_rows(self, np_value, partitioner,
312 partitioner=partitioner)
380 partitioner=None,
389 partitioner=None,
401 partitioner=None,
410 partitioner=None,
418 partitioner=partitioned_variables.fixed_size_partitioner(3),
428 partitioner=partitioned_variables.fixed_size_partitioner(3),
441 partitioner=partitioned_variables.fixed_size_partitioner(3)
[all...] |
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| moving_average_optimizer_test.py | 83 partitioner = partitioned_variables.fixed_size_partitioner(
91 partitioner = None
96 partitioner=partitioner)
101 partitioner=partitioner)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| composable_model.py | 231 partitioner = partitioned_variables.min_max_variable_partitioner(
235 partitioner=partitioner) as scope:
365 partitioner=input_layer_partitioner) as scope:
380 partitioner=hidden_layer_partitioner) as scope:
395 partitioner=hidden_layer_partitioner) as scope:
|
| linear.py | 150 partitioner = partitioned_variables.min_max_variable_partitioner(
157 partitioner=partitioner) as scope:
249 partitioner=optimizer.partitioner) as scope:
[all...] |
| dnn.py | 139 partitioner = partitioned_variables.min_max_variable_partitioner(
144 partitioner=partitioner):
152 partitioner=input_layer_partitioner) as input_layer_scope:
[all...] |
| /external/tensorflow/tensorflow/python/layers/ |
| base.py | 310 partitioner=None,
338 partitioner: (optional) partitioner instance (callable). If
340 into multiple partitions according to `partitioner`. In this case,
350 instance. If `partitioner` is not `None`, a `PartitionedVariable`
374 partitioner=partitioner,
438 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| variables.py | 212 partitioner=None,
236 partitioner: Optional callable that accepts a fully defined `TensorShape`
274 partitioner=partitioner,
290 partitioner=None,
315 partitioner: Optional callable that accepts a fully defined `TensorShape`
346 partitioner=partitioner,
|
| /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 | 169 partitioner = partitioned_variables.variable_axis_size_partitioner(40)
171 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/contrib/quantize/python/ |
| quant_ops_test.py | 90 # scalar. There would be a tensorflow error thrown if the partitioner was
94 'part', partitioner=partitioned_variables.fixed_size_partitioner(2)):
105 # scalar. There would be a tensorflow error thrown if the partitioner was
109 'part', partitioner=partitioned_variables.fixed_size_partitioner(2)):
|
| quantize_test.py | 422 partitioner = partitioned_variables.fixed_size_partitioner(1)
425 with variable_scope.variable_scope('part', partitioner=partitioner):
452 partitioner = partitioned_variables.fixed_size_partitioner(2)
455 with variable_scope.variable_scope('part', partitioner=partitioner):
|
| /external/tensorflow/tensorflow/python/tools/ |
| freeze_graph_test.py | 278 partitioner = partitioned_variables.fixed_size_partitioner(1)
280 with variable_scope.variable_scope("part", partitioner=partitioner):
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| rnn_cell.py | 220 variable_scope partitioner instead.
222 variable_scope partitioner instead.
251 "Use a variable scope with a partitioner instead.", self)
310 def add_variable_wrapped(name, shape, initializer, index, partitioner):
312 name, shape=shape, initializer=initializer, partitioner=partitioner)
381 partitioner=maybe_proj_partitioner)
|
