| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 293 partitioner=None, argument
445 partitioner=None, argument
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,
479 if (reuse is True and partitioner is None
491 partitioner=None,
539 "partitioner": partitioner,
562 partitioner=partitioner,
[all …]
|
| D | partitioned_variables.py | 298 partitioner = lambda **unused_kwargs: slicing function
309 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 61 partitioner=None): argument
68 partitioner=partitioner)
70 if partitioner:
100 def _create_linear_model(self, feature_cols, partitioner): argument
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])
185 partitioner=lambda shape, dtype: [2, 1])
276 partitioner=lambda shape, dtype: [2, 1])
[all …]
|
| D | 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))
|
| D | 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(
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_estimator.py | 119 partitioner = params["partitioner"]
141 partitioner=partitioner)
147 partitioner=partitioner) as scope:
220 partitioner=None): argument
276 "partitioner": partitioner,
347 partitioner=None): argument
390 partitioner=partitioner)
478 partitioner=None): argument
521 partitioner=partitioner)
|
| D | sdca_optimizer.py | 78 partitioner=None): argument
85 self._partitioner = partitioner
115 def partitioner(self): member in SDCAOptimizer
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | 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(
83 with variable_scope.variable_scope("root", partitioner=partitioner):
181 "root", partitioner=partitioner_axis3_str):
203 partitioner = partitioned_variables.min_max_variable_partitioner(
[all …]
|
| D | checkpoint_ops_test.py | 303 def _test_loading_variable_with_max_rows(self, np_value, partitioner, argument
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),
449 partitioner=partitioned_variables.fixed_size_partitioner(3),
|
| D | variable_scope_test.py | 1387 "scope0", partitioner=axis0_into2_partitioner):
1401 "scope0", partitioner=axis0_into2_partitioner):
1405 "scope0", partitioner=axis0_into3_partitioner, reuse=True):
1413 "scope0", partitioner=axis0_into1_partitioner, reuse=True):
1424 "scope0", partitioner=axis0_into2_partitioner):
1434 "scope0", partitioner=axis0_into2_partitioner):
1443 "scope0", partitioner=axis0_into2_partitioner):
1452 "scope0", partitioner=axis0_into2_partitioner) as vs:
1453 self.assertEqual(axis0_into2_partitioner, vs.partitioner)
1455 self.assertEqual(axis0_into2_partitioner, vs1.partitioner)
[all …]
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | 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:
|
| D | linear.py | 150 partitioner = partitioned_variables.min_max_variable_partitioner(
157 partitioner=partitioner) as scope:
249 partitioner=optimizer.partitioner) as scope:
|
| D | dnn.py | 139 partitioner = partitioned_variables.min_max_variable_partitioner(
144 partitioner=partitioner):
152 partitioner=input_layer_partitioner) as input_layer_scope:
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable-scope.pbtxt | 34 name: "partitioner"
51 …e\', \'name\', \'initializer\', \'regularizer\', \'caching_device\', \'partitioner\', \'custom_get…
59 …er\', \'reuse\', \'trainable\', \'collections\', \'caching_device\', \'partitioner\', \'validate_s…
91 argspec: "args=[\'self\', \'partitioner\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.variable_scope.pbtxt | 7 …', \'values\', \'initializer\', \'regularizer\', \'caching_device\', \'partitioner\', \'custom_get…
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | 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/framework/python/ops/ |
| D | variables.py | 212 partitioner=None, argument
274 partitioner=partitioner,
290 partitioner=None, argument
346 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | base.py | 310 partitioner=None, argument
374 partitioner=partitioner,
438 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | layers.py | 47 partitioner=None, argument
66 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | 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/ |
| D | 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):
|
| D | quant_ops_test.py | 94 'part', partitioner=partitioned_variables.fixed_size_partitioner(2)):
109 'part', partitioner=partitioned_variables.fixed_size_partitioner(2)):
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | pruning_test.py | 169 partitioner = partitioned_variables.variable_axis_size_partitioner(40)
171 with variable_scope.variable_scope("", partitioner=partitioner):
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 310 def add_variable_wrapped(name, shape, initializer, index, partitioner): argument
312 name, shape=shape, initializer=initializer, partitioner=partitioner)
381 partitioner=maybe_proj_partitioner)
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | freeze_graph_test.py | 278 partitioner = partitioned_variables.fixed_size_partitioner(1)
280 with variable_scope.variable_scope("part", partitioner=partitioner):
|