Home
last modified time | relevance | path

Searched refs:ONLY_FIRST_REPLICA (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dtraining_util.py132 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
143 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA,
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariables_test.py840 self.assertEqual(v1.ONLY_FIRST_REPLICA, v2.ONLY_FIRST_REPLICA)
841 self.assertEqual(v1.ONLY_FIRST_TOWER, v2.ONLY_FIRST_REPLICA)
846 self.assertEqual(v2.ONLY_FIRST_REPLICA, v1.ONLY_FIRST_REPLICA)
847 self.assertEqual(v2.ONLY_FIRST_REPLICA, v1.ONLY_FIRST_TOWER)
852 self.assertEqual(hash(v1.ONLY_FIRST_REPLICA), hash(v2.ONLY_FIRST_REPLICA))
853 self.assertEqual(hash(v1.ONLY_FIRST_TOWER), hash(v2.ONLY_FIRST_REPLICA))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-variable-aggregation.pbtxt13 name: "ONLY_FIRST_REPLICA"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-variable-aggregation.pbtxt13 name: "ONLY_FIRST_REPLICA"
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py577 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
589 aggregation=tf_variables.VariableAggregation.ONLY_FIRST_REPLICA)
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy.py309 vs.VariableAggregation.ONLY_FIRST_REPLICA
Dtpu_strategy.py90 vs.VariableAggregation.ONLY_FIRST_REPLICA,
Dvalues.py624 if aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:
1264 if self._aggregation == vs.VariableAggregation.ONLY_FIRST_REPLICA:
Dmirrored_strategy.py235 variable_scope.VariableAggregation.ONLY_FIRST_REPLICA
/external/tensorflow/tensorflow/python/ops/
Dvariables.py104 ONLY_FIRST_REPLICA = 3 variable in VariableAggregationV2
123 ONLY_FIRST_REPLICA = 3 variable in VariableAggregation
/external/tensorflow/tensorflow/contrib/distribute/python/
Dstrategy_test_lib.py339 aggregation=variables.VariableAggregation.ONLY_FIRST_REPLICA)
Dparameter_server_strategy_test.py429 aggregation=variable_scope.VariableAggregation.ONLY_FIRST_REPLICA)
Dmirrored_strategy_multigpu_test.py515 aggregation = variable_scope.VariableAggregation.ONLY_FIRST_REPLICA