| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 564 with self.distribute_strategy.scope():
725 def distribute_strategy(self): member in Model
837 outputs = model.distribute_strategy.run(run_step, args=(data,))
839 outputs, self.distribute_strategy, reduction='first')
1095 if self.distribute_strategy._should_use_with_coordinator: # pylint: disable=protected-access
1097 self.distribute_strategy)
1099 with self.distribute_strategy.scope(), \
1277 outputs = model.distribute_strategy.run(run_step, args=(data,))
1279 outputs, self.distribute_strategy, reduction='first')
1399 if self.distribute_strategy._should_use_with_coordinator: # pylint: disable=protected-access
[all …]
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | worker_training_state.py | 81 checkpoint_dir, self._model.distribute_strategy)
82 if self._model.distribute_strategy.extended.should_checkpoint:
99 self._model.distribute_strategy)
|
| D | parameter_server_training_test.py | 357 strategy = model.distribute_strategy
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values_util.py | 72 strategy = var.distribute_strategy
123 strategy = var.distribute_strategy
186 with ds_context.enter_or_assert_strategy(var.distribute_strategy):
198 with ds_context.enter_or_assert_strategy(var.distribute_strategy):
211 with ds_context.enter_or_assert_strategy(var.distribute_strategy):
|
| D | values.py | 68 assert strategy == var.distribute_strategy
654 def distribute_strategy(self): member in DistributedVariable
878 return self.distribute_strategy.extended.update(
919 with ds_context.enter_or_assert_strategy(self.distribute_strategy):
927 values_util.assert_replica_context(self.distribute_strategy)
1390 with ds_context.enter_or_assert_strategy(var.distribute_strategy):
1400 with ds_context.enter_or_assert_strategy(var.distribute_strategy):
1410 with ds_context.enter_or_assert_strategy(var.distribute_strategy):
1411 return var.distribute_strategy.reduce(
1427 with ds_context.enter_or_assert_strategy(var.distribute_strategy):
[all …]
|
| D | mirrored_variable_test.py | 95 self.assertIs(strategy, var.distribute_strategy)
590 self.assertIs(distribution, mirrored.distribute_strategy)
591 self.assertIs(distribution, sync_on_read.distribute_strategy)
|
| D | ps_values.py | 81 def distribute_strategy(self): member in AggregatingVariable
|
| D | packed_distributed_variable.py | 71 …distribute_strategy=distributed_variables[0]._distribute_strategy, # pylint: disable=protected-ac…
|
| D | values_test.py | 545 self.assertEqual(v1.distribute_strategy, v2.distribute_strategy)
|
| D | parameter_server_strategy_test.py | 855 self.assertIs(strategy, created_step.distribute_strategy)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 345 distribute_strategy=None, argument
418 self._distribute_strategy = distribute_strategy
505 distribute_strategy=self._distribute_strategy,
910 distribute_strategy=self._distribute_strategy), ()
1501 distribute_strategy=None, argument
1588 distribute_strategy=distribute_strategy)
1600 distribute_strategy=None, argument
1820 distribute_strategy=distribute_strategy,
1908 distribute_strategy=None, argument
1977 distribute_strategy=distribute_strategy,
|
| D | variable_scope.py | 2609 distribute_strategy = kwargs.get("distribute_strategy", None)
2621 distribute_strategy=distribute_strategy,
2655 distribute_strategy = kwargs.get("distribute_strategy", None)
2670 distribute_strategy=distribute_strategy,
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks.py | 1438 file_path, self.model.distribute_strategy)
1446 self._write_filepath, self.model.distribute_strategy)
1637 self.model.distribute_strategy, self._supported_strategies):
1642 type(self.model.distribute_strategy).__name__)
2209 self.model.distribute_strategy)
2214 self.model.distribute_strategy)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.-model.pbtxt | 20 name: "distribute_strategy"
|
| D | tensorflow.keras.models.-sequential.pbtxt | 22 name: "distribute_strategy"
|
| D | tensorflow.keras.models.-model.pbtxt | 20 name: "distribute_strategy"
|
| D | tensorflow.keras.experimental.-wide-deep-model.pbtxt | 21 name: "distribute_strategy"
|
| D | tensorflow.keras.experimental.-linear-model.pbtxt | 21 name: "distribute_strategy"
|
| D | tensorflow.keras.-sequential.pbtxt | 22 name: "distribute_strategy"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.models.-model.pbtxt | 20 name: "distribute_strategy"
|
| D | tensorflow.keras.models.-sequential.pbtxt | 22 name: "distribute_strategy"
|
| D | tensorflow.keras.experimental.-linear-model.pbtxt | 21 name: "distribute_strategy"
|
| D | tensorflow.keras.-model.pbtxt | 20 name: "distribute_strategy"
|
| D | tensorflow.keras.-sequential.pbtxt | 22 name: "distribute_strategy"
|
| D | tensorflow.keras.experimental.-wide-deep-model.pbtxt | 21 name: "distribute_strategy"
|