Home
last modified time | relevance | path

Searched refs:VariableAggregation (Results 1 – 25 of 503) sorted by relevance

12345678910>>...21

/external/tensorflow/tensorflow/python/distribute/
Dvars_test.py100 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 …]
Dmirrored_variable_test.py222 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 …]
Dreduce_util.py43 variable_scope.VariableAggregation.SUM: ReduceOp.SUM,
44 variable_scope.VariableAggregation.MEAN: ReduceOp.MEAN,
Dvalues.py48 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 …]
Ddistribute_utils.py271 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):
Dvalues_test.py101 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 …]
Dvalues_util.py122 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:
Dparameter_server_strategy_test.py153 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 …]
Dtpu_values.py260 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):
Dparameter_server_strategy.py429 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
Dmoving_averages_test.py203 aggregation=variables.VariableAggregation.MEAN)
228 aggregation=variables.VariableAggregation.MEAN)
247 aggregation=variables.VariableAggregation.MEAN)
Ddistribute_lib_test.py170 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)
Dmirrored_strategy_test.py699 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)
Dps_values_test.py51 1, aggregation=variables_lib.VariableAggregation.MEAN)
Dcheckpoint_utils_test.py85 aggregation=variable_scope.VariableAggregation.MEAN)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-variable-aggregation.pbtxt1 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/
Dsaved_model_test.py88 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/
Dtraining_util.py131 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
143 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
/external/tensorflow/tensorflow/python/ops/
Dvariables.py123 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 …]
Dvariable_scope.py205 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/
Dtensorflow.-variable-aggregation.pbtxt1 path: "tensorflow.VariableAggregation"
Dtensorflow.keras.optimizers.-s-g-d.pbtxt37 …=[\'None\', \'zeros\', \'None\', \'VariableSynchronization.AUTO\', \'VariableAggregation.NONE\'], "
Dtensorflow.optimizers.-adamax.pbtxt37 …=[\'None\', \'zeros\', \'None\', \'VariableSynchronization.AUTO\', \'VariableAggregation.NONE\'], "
Dtensorflow.keras.optimizers.-adagrad.pbtxt37 …=[\'None\', \'zeros\', \'None\', \'VariableSynchronization.AUTO\', \'VariableAggregation.NONE\'], "
/external/tensorflow/tensorflow/core/framework/
Dvariable.proto30 enum VariableAggregation { enum
72 VariableAggregation aggregation = 9;

12345678910>>...21