| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 63 partitioner=None): argument
70 partitioner=partitioner)
72 if partitioner:
102 def _create_linear_model(self, feature_cols, partitioner): argument
104 with variable_scope.variable_scope("", partitioner=partitioner):
138 partitioner=lambda shape, dtype: [2, 1])
161 partitioner=lambda shape, dtype: [2, 1])
178 partitioner=lambda shape, dtype: [2, 1])
187 partitioner=lambda shape, dtype: [2, 1])
278 partitioner=lambda shape, dtype: [2, 1])
[all …]
|
| D | checkpoint_ops_test.py | 155 partitioner=partitioned_variables.fixed_size_partitioner(2))
189 partitioner=partitioned_variables.fixed_size_partitioner(2))
227 partitioner=partitioned_variables.fixed_size_partitioner(2))
263 partitioner=partitioned_variables.fixed_size_partitioner(2))
297 partitioner=partitioned_variables.fixed_size_partitioner(2))
343 partitioner=partitioned_variables.fixed_size_partitioner(2))
381 partitioner=partitioned_variables.fixed_size_partitioner(2))
|
| D | checkpoint_utils_test.py | 71 partitioner=partitioned_variables.min_max_variable_partitioner(
278 partitioner=partitioned_variables.min_max_variable_partitioner(
288 partitioner=partitioned_variables.min_max_variable_partitioner(
316 partitioner=partitioned_variables.min_max_variable_partitioner(
|
| D | saver_large_partitioned_variable_test.py | 51 partitioner=partitioned_variables.fixed_size_partitioner(4),
|
| D | slot_creator_test.py | 145 partitioner=partitioned_variables.fixed_size_partitioner(2))
167 partitioner=partitioned_variables.fixed_size_partitioner(2))
|
| D | localhost_cluster_performance_test.py | 101 partitioner=partitioned_variables.variable_axis_size_partitioner(
|
| D | saver_test.py | 177 partitioner = partitioned_variables.fixed_size_partitioner(num_shards=2)
181 partitioner=partitioner,
442 partitioner=partitioned_variables.fixed_size_partitioner(
447 partitioner=partitioned_variables.fixed_size_partitioner(
1012 def _save(partitioner=None): argument
1017 if partitioner:
1023 partitioner=partitioner,
1042 def _restore(partitioner=None): argument
1044 if partitioner:
1050 partitioner=partitioner)
[all …]
|
| D | slot_creator.py | 55 current_partitioner = variable_scope.get_variable_scope().partitioner
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 297 partitioner=None, argument
446 partitioner=None, argument
456 if partitioner is not None and not is_scalar:
457 if not callable(partitioner):
459 partitioner)
471 partitioner=partitioner,
480 if (reuse is True and partitioner is None
492 partitioner=None,
540 "partitioner": partitioner,
563 partitioner=partitioner,
[all …]
|
| D | partitioned_variables.py | 298 partitioner = lambda **unused_kwargs: slicing function
309 partitioner=partitioner,
|
| /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 | 304 def _test_loading_variable_with_max_rows(self, np_value, partitioner, argument
313 partitioner=partitioner)
381 partitioner=None,
390 partitioner=None,
402 partitioner=None,
411 partitioner=None,
419 partitioner=partitioned_variables.fixed_size_partitioner(3),
429 partitioner=partitioned_variables.fixed_size_partitioner(3),
442 partitioner=partitioned_variables.fixed_size_partitioner(3),
450 partitioner=partitioned_variables.fixed_size_partitioner(3),
|
| D | variable_scope_test.py | 1399 "scope0", partitioner=axis0_into2_partitioner):
1413 "scope0", partitioner=axis0_into2_partitioner):
1417 "scope0", partitioner=axis0_into3_partitioner, reuse=True):
1425 "scope0", partitioner=axis0_into1_partitioner, reuse=True):
1436 "scope0", partitioner=axis0_into2_partitioner):
1446 "scope0", partitioner=axis0_into2_partitioner):
1455 "scope0", partitioner=axis0_into2_partitioner):
1464 "scope0", partitioner=axis0_into2_partitioner) as vs:
1465 self.assertEqual(axis0_into2_partitioner, vs.partitioner)
1467 self.assertEqual(axis0_into2_partitioner, vs1.partitioner)
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | dense_features.py | 72 partitioner=None, argument
96 partitioner=partitioner,
|
| /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…
|
| D | tensorflow.keras.layers.-dense-features.pbtxt | 116 …argspec: "args=[\'self\', \'feature_columns\', \'trainable\', \'name\', \'partitioner\'], varargs=…
136 …\', \'initializer\', \'regularizer\', \'trainable\', \'constraint\', \'partitioner\', \'use_resour…
|
| D | tensorflow.estimator.-linear-estimator.pbtxt | 24 …ad\', \'feature_columns\', \'model_dir\', \'optimizer\', \'config\', \'partitioner\', \'sparse_com…
|
| D | tensorflow.estimator.-linear-regressor.pbtxt | 24 …, \'label_dimension\', \'weight_column\', \'optimizer\', \'config\', \'partitioner\', \'warm_start…
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_preprocessing_layer.py | 80 partitioner=None, argument
106 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | base.py | 327 partitioner=None, argument
391 partitioner=partitioner,
455 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 312 def add_variable_wrapped(name, shape, initializer, index, partitioner): argument
314 name, shape=shape, initializer=initializer, partitioner=partitioner)
383 partitioner=maybe_proj_partitioner)
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | freeze_graph_test.py | 282 partitioner = partitioned_variables.fixed_size_partitioner(1)
284 with variable_scope.variable_scope("part", partitioner=partitioner):
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 221 partitioner = partitioned_variables.fixed_size_partitioner(num_shards)
231 partitioner=partitioner)
244 partitioner=partitioner)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 425 partitioner = self._scope.partitioner
428 partitioner = None
499 if partitioner:
500 self._scope.set_partitioner(partitioner)
|