Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/python/distribute/
Dvars_test.py98 variables_lib.VariableAggregation.NONE,
99 variables_lib.VariableAggregation.SUM,
100 variables_lib.VariableAggregation.MEAN,
101 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,
110 variables_lib.VariableAggregation.SUM):
128 2., aggregation=variables_lib.VariableAggregation.MEAN)
130 -2., aggregation=variables_lib.VariableAggregation.MEAN)
144 variables_lib.VariableAggregation.NONE,
145 variables_lib.VariableAggregation.SUM,
146 variables_lib.VariableAggregation.MEAN,
[all …]
Dvalues.py46 if var.aggregation == vs.VariableAggregation.NONE:
53 if var.aggregation == vs.VariableAggregation.MEAN and (
83 if var.aggregation == vs.VariableAggregation.MEAN and (
108 if aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:
482 if (aggregation == variables_lib.VariableAggregation.MEAN and
1103 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and
1104 self._aggregation != vs.VariableAggregation.NONE):
1113 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and
1114 self._aggregation != vs.VariableAggregation.NONE):
1123 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and
[all …]
Dvalues_test.py466 variable_scope.VariableAggregation.SUM)
472 self.assertEqual(variable_scope.VariableAggregation.SUM,
490 aggregation=variables_lib.VariableAggregation.MEAN,
499 variable_scope.VariableAggregation.SUM, distribution)
518 variable_scope.VariableAggregation.ONLY_FIRST_REPLICA, distribution)
535 aggregation=variables_lib.VariableAggregation.SUM,
539 aggregation=variables_lib.VariableAggregation.SUM,
561 aggregation=variables_lib.VariableAggregation.NONE,
608 variable_scope.VariableAggregation.SUM, distribution)
632 variable_scope.VariableAggregation.MEAN, distribution)
[all …]
Dmirrored_variable_test.py223 aggregation=variable_scope.VariableAggregation.SUM)
228 aggregation=variable_scope.VariableAggregation.MEAN)
245 self.assertEqual(variable_scope.VariableAggregation.SUM, v2.aggregation)
248 self.assertEqual(variable_scope.VariableAggregation.MEAN, v3.aggregation)
261 aggregation=variable_scope.VariableAggregation.SUM)
265 aggregation=variable_scope.VariableAggregation.MEAN)
283 self.assertEqual(variable_scope.VariableAggregation.SUM, v2.aggregation)
286 self.assertEqual(variable_scope.VariableAggregation.MEAN,
292 aggregation = variable_scope.VariableAggregation.ONLY_FIRST_REPLICA
451 aggregation=variable_scope.VariableAggregation.SUM)
[all …]
Dreduce_util.py39 variable_scope.VariableAggregation.SUM: ReduceOp.SUM,
40 variable_scope.VariableAggregation.MEAN: ReduceOp.MEAN,
Ddistribute_utils.py278 aggregation = kwargs.get("aggregation", vs.VariableAggregation.NONE)
280 if aggregation not in (vs.VariableAggregation.NONE,
281 vs.VariableAggregation.SUM,
282 vs.VariableAggregation.MEAN,
283 vs.VariableAggregation.ONLY_FIRST_REPLICA):
Ddistributed_variable_test.py92 variables_lib.VariableAggregation.MEAN,
93 variables_lib.VariableAggregation.SUM,
94 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,
201 aggregation != variables_lib.VariableAggregation.SUM):
264 and aggregation == variables_lib.VariableAggregation.SUM):
276 aggregation != variables_lib.VariableAggregation.NONE):
360 if aggregation != variables_lib.VariableAggregation.MEAN:
478 if aggregation != variables_lib.VariableAggregation.MEAN:
548 (aggregation == variables_lib.VariableAggregation.SUM or
552 and aggregation == variables_lib.VariableAggregation.MEAN))):
Dvalues_util.py118 if aggregation == vs.VariableAggregation.SUM:
184 if var.aggregation == vs.VariableAggregation.SUM:
196 if var.aggregation == vs.VariableAggregation.SUM:
213 if var.aggregation == vs.VariableAggregation.SUM:
322 if aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:
Dparameter_server_strategy_test.py149 aggregation=variable_scope.VariableAggregation.SUM)
163 aggregation=variable_scope.VariableAggregation.SUM)
173 aggregation=variable_scope.VariableAggregation.SUM)
231 aggregation=variable_scope.VariableAggregation.SUM,
244 aggregation=variable_scope.VariableAggregation.SUM,
304 aggregation=variable_scope.VariableAggregation.SUM)
316 aggregation=variable_scope.VariableAggregation.SUM)
327 aggregation=variable_scope.VariableAggregation.SUM)
385 aggregation=variable_scope.VariableAggregation.SUM)
388 aggregation=variable_scope.VariableAggregation.SUM)
[all …]
Dtpu_values.py252 self.aggregation == variable_scope.VariableAggregation.NONE):
276 self.aggregation == variable_scope.VariableAggregation.NONE):
300 self.aggregation == variable_scope.VariableAggregation.NONE):
428 var.aggregation == variable_scope.VariableAggregation.NONE):
446 var.aggregation == variable_scope.VariableAggregation.NONE):
459 var.aggregation == variable_scope.VariableAggregation.NONE):
479 if self._aggregation != variable_scope.VariableAggregation.NONE:
Dparameter_server_strategy_v2.py605 aggregation = kwargs.pop("aggregation", vs.VariableAggregation.NONE)
617 if aggregation not in (vs.VariableAggregation.NONE,
618 vs.VariableAggregation.SUM,
619 vs.VariableAggregation.MEAN,
620 vs.VariableAggregation.ONLY_FIRST_REPLICA):
Dparameter_server_strategy.py426 aggregation = kwargs.pop("aggregation", vs.VariableAggregation.NONE)
428 vs.VariableAggregation.NONE,
429 vs.VariableAggregation.SUM,
430 vs.VariableAggregation.MEAN,
431 vs.VariableAggregation.ONLY_FIRST_REPLICA
Dmoving_averages_test.py194 aggregation=variables.VariableAggregation.MEAN)
219 aggregation=variables.VariableAggregation.MEAN)
238 aggregation=variables.VariableAggregation.MEAN)
Dvalues_v2_test.py88 variables_lib.VariableAggregation.MEAN,
89 variables_lib.VariableAggregation.SUM,
90 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,
91 variables_lib.VariableAggregation.NONE,
Dmirrored_strategy_test.py723 1.0, name="foo", aggregation=variable_scope.VariableAggregation.SUM)
761 1.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
782 1.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
842 1.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
863 1.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
905 5.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
926 5.0, name="foo", aggregation=variable_scope.VariableAggregation.MEAN)
973 aggregation=variable_scope.VariableAggregation.SUM)
999 aggregation=variable_scope.VariableAggregation.SUM)
1026 aggregation=variable_scope.VariableAggregation.MEAN)
Ddistribute_lib_test.py167 variable_scope.VariableAggregation.NONE)
187 variable_scope.VariableAggregation.NONE)
245 variable_scope.VariableAggregation.MEAN)
252 aggregation=variable_scope.VariableAggregation.MEAN))
267 variable_scope.VariableAggregation.NONE)
Dps_values_test.py47 1, aggregation=variables_lib.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/keras/legacy_tf_layers/
Dvariable_scope_shim.py105 aggregation = variables.VariableAggregation.NONE
108 (variables.VariableAggregation,
170 aggregation=vs.VariableAggregation.NONE):
312 aggregation=vs.VariableAggregation.NONE):
402 aggregation=vs.VariableAggregation.NONE):
/external/tensorflow/tensorflow/python/distribute/integration_test/
Dsaved_model_test.py82 aggregation=tf.VariableAggregation.SUM)
138 aggregation=tf.VariableAggregation.SUM)
195 aggregation=tf.VariableAggregation.SUM)
221 aggregation=tf.VariableAggregation.MEAN)
352 aggregation=tf.VariableAggregation.SUM)
405 aggregation=tf.VariableAggregation.SUM)
486 aggregation=tf.VariableAggregation.MEAN)
/external/tensorflow/tensorflow/python/ops/
Dvariables.py127 elif isinstance(other, VariableAggregation):
134 class VariableAggregation(enum.Enum): class
150 VariableAggregation.__doc__ = (
159 aggregation = VariableAggregation.NONE
162 (VariableAggregation, VariableAggregationV2)):
200 aggregation=VariableAggregation.NONE,
209 aggregation = VariableAggregation.NONE
238 aggregation=VariableAggregation.NONE,
247 aggregation = VariableAggregation.NONE
383 aggregation=VariableAggregation.NONE,
[all …]
Dvariable_scope.py194 VariableAggregation = variables.VariableAggregation # pylint: disable=invalid-name variable
332 aggregation=VariableAggregation.NONE):
480 aggregation=VariableAggregation.NONE):
614 aggregation=VariableAggregation.NONE):
841 aggregation=VariableAggregation.NONE):
1289 aggregation=VariableAggregation.NONE):
1359 aggregation=VariableAggregation.NONE):
1615 aggregation=VariableAggregation.NONE):
1792 aggregation=VariableAggregation.NONE):
1835 aggregation=VariableAggregation.NONE):
/external/tensorflow/tensorflow/python/training/
Dtraining_util.py239 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
251 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
/external/tensorflow/tensorflow/core/framework/
Dvariable.proto30 enum VariableAggregation { enum
72 VariableAggregation aggregation = 9;
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-variable-aggregation.pbtxt1 path: "tensorflow.VariableAggregation"

1234