| /external/tensorflow/tensorflow/python/distribute/ |
| D | vars_test.py | 100 variables_lib.VariableAggregation.NONE,
101 variables_lib.VariableAggregation.SUM,
102 variables_lib.VariableAggregation.MEAN,
103 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,
112 variables_lib.VariableAggregation.SUM):
130 2., aggregation=variables_lib.VariableAggregation.MEAN)
132 -2., aggregation=variables_lib.VariableAggregation.MEAN)
147 variables_lib.VariableAggregation.NONE,
148 variables_lib.VariableAggregation.SUM,
149 variables_lib.VariableAggregation.MEAN,
[all …]
|
| D | mirrored_variable_test.py | 222 aggregation=variable_scope.VariableAggregation.SUM)
227 aggregation=variable_scope.VariableAggregation.MEAN)
244 self.assertEqual(variable_scope.VariableAggregation.SUM, v2.aggregation)
247 self.assertEqual(variable_scope.VariableAggregation.MEAN, v3.aggregation)
260 aggregation=variable_scope.VariableAggregation.SUM)
264 aggregation=variable_scope.VariableAggregation.MEAN)
282 self.assertEqual(variable_scope.VariableAggregation.SUM, v2.aggregation)
285 self.assertEqual(variable_scope.VariableAggregation.MEAN,
291 aggregation = variable_scope.VariableAggregation.ONLY_FIRST_REPLICA
452 aggregation=variable_scope.VariableAggregation.SUM)
[all …]
|
| D | reduce_util.py | 43 variable_scope.VariableAggregation.SUM: ReduceOp.SUM,
44 variable_scope.VariableAggregation.MEAN: ReduceOp.MEAN,
|
| D | values.py | 48 if var.aggregation == vs.VariableAggregation.NONE:
60 if var.aggregation == vs.VariableAggregation.MEAN and (
446 if (aggregation == variables_lib.VariableAggregation.MEAN and
1042 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and
1043 self._aggregation != vs.VariableAggregation.NONE):
1051 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and
1052 self._aggregation != vs.VariableAggregation.NONE):
1060 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and
1061 self._aggregation != vs.VariableAggregation.NONE):
1238 if self._aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:
[all …]
|
| D | distribute_utils.py | 271 aggregation = kwargs.get("aggregation", vs.VariableAggregation.NONE)
273 if aggregation not in (vs.VariableAggregation.NONE,
274 vs.VariableAggregation.SUM,
275 vs.VariableAggregation.MEAN,
276 vs.VariableAggregation.ONLY_FIRST_REPLICA):
|
| D | values_test.py | 101 mirrored = var_cls(distribution, v, variable_scope.VariableAggregation.SUM)
414 variables_lib.VariableAggregation.MEAN,
415 variables_lib.VariableAggregation.SUM,
416 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,
524 aggregation != variables_lib.VariableAggregation.SUM):
588 and aggregation == variables_lib.VariableAggregation.SUM):
600 aggregation != variables_lib.VariableAggregation.NONE):
684 if aggregation != variables_lib.VariableAggregation.MEAN:
804 if aggregation != variables_lib.VariableAggregation.MEAN:
872 (aggregation == variables_lib.VariableAggregation.SUM or
[all …]
|
| D | values_util.py | 122 if aggregation == vs.VariableAggregation.SUM:
188 if var.aggregation == vs.VariableAggregation.SUM:
200 if var.aggregation == vs.VariableAggregation.SUM:
217 if var.aggregation == vs.VariableAggregation.SUM:
326 if aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:
|
| D | parameter_server_strategy_test.py | 153 aggregation=variable_scope.VariableAggregation.SUM)
167 aggregation=variable_scope.VariableAggregation.SUM)
177 aggregation=variable_scope.VariableAggregation.SUM)
235 aggregation=variable_scope.VariableAggregation.SUM,
248 aggregation=variable_scope.VariableAggregation.SUM,
308 aggregation=variable_scope.VariableAggregation.SUM)
320 aggregation=variable_scope.VariableAggregation.SUM)
331 aggregation=variable_scope.VariableAggregation.SUM)
389 aggregation=variable_scope.VariableAggregation.SUM)
392 aggregation=variable_scope.VariableAggregation.SUM)
[all …]
|
| D | tpu_values.py | 260 self.aggregation == variable_scope.VariableAggregation.NONE):
273 self.aggregation == variable_scope.VariableAggregation.NONE):
286 self.aggregation == variable_scope.VariableAggregation.NONE):
412 var.aggregation == variable_scope.VariableAggregation.NONE):
430 var.aggregation == variable_scope.VariableAggregation.NONE):
443 var.aggregation == variable_scope.VariableAggregation.NONE):
|
| D | parameter_server_strategy.py | 429 aggregation = kwargs.pop("aggregation", vs.VariableAggregation.NONE)
431 vs.VariableAggregation.NONE,
432 vs.VariableAggregation.SUM,
433 vs.VariableAggregation.MEAN,
434 vs.VariableAggregation.ONLY_FIRST_REPLICA
|
| D | moving_averages_test.py | 203 aggregation=variables.VariableAggregation.MEAN)
228 aggregation=variables.VariableAggregation.MEAN)
247 aggregation=variables.VariableAggregation.MEAN)
|
| D | distribute_lib_test.py | 170 variable_scope.VariableAggregation.NONE)
190 variable_scope.VariableAggregation.NONE)
248 variable_scope.VariableAggregation.MEAN)
255 aggregation=variable_scope.VariableAggregation.MEAN))
270 variable_scope.VariableAggregation.NONE)
|
| D | mirrored_strategy_test.py | 699 1.0, name="foo", aggregation=variable_scope.VariableAggregation.SUM)
731 1.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
752 1.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
812 1.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
833 1.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
875 5.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
896 5.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
943 aggregation=variable_scope.VariableAggregation.SUM)
969 aggregation=variable_scope.VariableAggregation.SUM)
996 aggregation=variable_scope.VariableAggregation.MEAN)
|
| D | ps_values_test.py | 51 1, aggregation=variables_lib.VariableAggregation.MEAN)
|
| D | checkpoint_utils_test.py | 85 aggregation=variable_scope.VariableAggregation.MEAN)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable-aggregation.pbtxt | 1 path: "tensorflow.VariableAggregation"
3 is_instance: "<enum \'VariableAggregation\'>"
6 mtype: "<enum \'VariableAggregation\'>"
10 mtype: "<enum \'VariableAggregation\'>"
14 mtype: "<enum \'VariableAggregation\'>"
18 mtype: "<enum \'VariableAggregation\'>"
22 mtype: "<enum \'VariableAggregation\'>"
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | saved_model_test.py | 88 aggregation=tf.VariableAggregation.SUM)
145 aggregation=tf.VariableAggregation.SUM)
171 aggregation=tf.VariableAggregation.MEAN)
290 aggregation=tf.VariableAggregation.SUM)
343 aggregation=tf.VariableAggregation.SUM)
420 aggregation=tf.VariableAggregation.MEAN)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | training_util.py | 131 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
143 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variables.py | 123 elif isinstance(other, VariableAggregation):
130 class VariableAggregation(enum.Enum): class
146 VariableAggregation.__doc__ = (
155 aggregation = VariableAggregation.NONE
158 (VariableAggregation, VariableAggregationV2)):
196 aggregation=VariableAggregation.NONE,
205 aggregation = VariableAggregation.NONE
234 aggregation=VariableAggregation.NONE,
243 aggregation = VariableAggregation.NONE
380 aggregation=VariableAggregation.NONE,
[all …]
|
| D | variable_scope.py | 205 VariableAggregation = variables.VariableAggregation # pylint: disable=invalid-name variable
326 aggregation=VariableAggregation.NONE):
474 aggregation=VariableAggregation.NONE):
608 aggregation=VariableAggregation.NONE):
835 aggregation=VariableAggregation.NONE):
1283 aggregation=VariableAggregation.NONE):
1353 aggregation=VariableAggregation.NONE):
1576 aggregation=VariableAggregation.NONE):
1720 aggregation=VariableAggregation.NONE):
1763 aggregation=VariableAggregation.NONE):
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable-aggregation.pbtxt | 1 path: "tensorflow.VariableAggregation"
|
| D | tensorflow.keras.optimizers.-s-g-d.pbtxt | 37 …=[\'None\', \'zeros\', \'None\', \'VariableSynchronization.AUTO\', \'VariableAggregation.NONE\'], "
|
| D | tensorflow.optimizers.-adamax.pbtxt | 37 …=[\'None\', \'zeros\', \'None\', \'VariableSynchronization.AUTO\', \'VariableAggregation.NONE\'], "
|
| D | tensorflow.keras.optimizers.-adagrad.pbtxt | 37 …=[\'None\', \'zeros\', \'None\', \'VariableSynchronization.AUTO\', \'VariableAggregation.NONE\'], "
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variable.proto | 30 enum VariableAggregation { enum
72 VariableAggregation aggregation = 9;
|