Searched refs:AggregatingVariable (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | ps_values_test.py | 43 self.assertIsInstance(aggregating, ps_values.AggregatingVariable)
|
| D | ps_values.py | 36 class AggregatingVariable(variables_lib.Variable, core.Tensor): class
342 ops.register_tensor_conversion_function(AggregatingVariable,
|
| D | parameter_server_strategy.py | 449 wrapped = ps_values.AggregatingVariable(self._container_strategy(), v,
553 if isinstance(var, ps_values.AggregatingVariable):
583 not isinstance(val, ps_values.AggregatingVariable)):
|
| D | parameter_server_strategy_test.py | 853 self.assertIs(ps_values.AggregatingVariable, type(created_step))
854 self.assertIs(ps_values.AggregatingVariable, type(get_step))
885 self.assertIs(ps_values.AggregatingVariable, type(w))
|
| D | cross_device_ops.py | 73 (value_lib.DistributedValues, ops.Tensor, ps_values.AggregatingVariable,
|
| D | values_test.py | 546 if isinstance(v1, ps_values.AggregatingVariable):
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 517 not isinstance(variable, ps_distribute_values.AggregatingVariable)):
528 if issubclass(ps_distribute_values.AggregatingVariable,
|
| /external/tensorflow/tensorflow/python/module/ |
| D | module_test.py | 256 aggregating = ps_values.AggregatingVariable(
|