Searched refs:ONLY_FIRST_REPLICA (Results 1 – 22 of 22) sorted by relevance
103 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,150 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,202 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,253 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,278 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,331 aggregation=variables_lib.VariableAggregation.ONLY_FIRST_REPLICA)406 variable_scope.VariableAggregation.ONLY_FIRST_REPLICA,550 aggregation=variables_lib.VariableAggregation.ONLY_FIRST_REPLICA)577 aggregation=variables_lib.VariableAggregation.ONLY_FIRST_REPLICA)604 aggregation=variables_lib.VariableAggregation.ONLY_FIRST_REPLICA)[all …]
1042 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and1051 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and1060 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and1238 if self._aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:1246 if self._aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:1393 if self._aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:1406 if self._aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:1561 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and1569 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and1577 if (self._aggregation != vs.VariableAggregation.ONLY_FIRST_REPLICA and
276 vs.VariableAggregation.ONLY_FIRST_REPLICA):
326 if aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:
434 vs.VariableAggregation.ONLY_FIRST_REPLICA
291 aggregation = variable_scope.VariableAggregation.ONLY_FIRST_REPLICA
416 variables_lib.VariableAggregation.ONLY_FIRST_REPLICA,1191 variable_scope.VariableAggregation.ONLY_FIRST_REPLICA, distribution)
418 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA)
306 1.0, aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA)
395 aggregation=variable_scope.VariableAggregation.ONLY_FIRST_REPLICA)
630 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA)
574 aggregation=variables.VariableAggregationV2.ONLY_FIRST_REPLICA)577 self.assertEqual(variables.VariableAggregation.ONLY_FIRST_REPLICA,585 self.assertEqual(variables.VariableAggregationV2.ONLY_FIRST_REPLICA,903 self.assertEqual(v1.ONLY_FIRST_REPLICA, v2.ONLY_FIRST_REPLICA)904 self.assertEqual(v1.ONLY_FIRST_TOWER, v2.ONLY_FIRST_REPLICA)909 self.assertEqual(v2.ONLY_FIRST_REPLICA, v1.ONLY_FIRST_REPLICA)910 self.assertEqual(v2.ONLY_FIRST_REPLICA, v1.ONLY_FIRST_TOWER)915 self.assertEqual(hash(v1.ONLY_FIRST_REPLICA), hash(v2.ONLY_FIRST_REPLICA))916 self.assertEqual(hash(v1.ONLY_FIRST_TOWER), hash(v2.ONLY_FIRST_REPLICA))
131 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,143 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
13 name: "ONLY_FIRST_REPLICA"
39 // `ONLY_FIRST_REPLICA`: This is for when every replica is performing the same
99 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
974 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)987 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
248 aggregation=variables.VariableAggregationV2.ONLY_FIRST_REPLICA)
332 agg = variables.VariableAggregationV2.ONLY_FIRST_REPLICA628 aggregation=variables.VariableAggregationV2.ONLY_FIRST_REPLICA)
115 ONLY_FIRST_REPLICA = 3 variable in VariableAggregationV2134 ONLY_FIRST_REPLICA = 3 variable in VariableAggregation
1772 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA)1775 self.assertEqual(variables.VariableAggregation.ONLY_FIRST_REPLICA,1783 self.assertEqual(variables.VariableAggregation.ONLY_FIRST_REPLICA,