Home
last modified time | relevance | path

Searched refs:DistributedValues (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dvalues_test.py48 v = values.DistributedValues(device_map, (one, two))
61 v = values.DistributedValues(device_map, (one, two))
69 v = values.DistributedValues(values.SingleDeviceMap(""), (42,))
71 v = values.DistributedValues(values.SingleDeviceMap("/device:CPU:0"), (42,))
73 v = values.DistributedValues(values.SingleDeviceMap("/cpu:0"), (42,))
75 v = values.DistributedValues(values.SingleDeviceMap("/CPU:0"), (42,))
78 v = values.DistributedValues(
88 v = values.DistributedValues(device_map, (one, two))
101 v = values.DistributedValues(device_map, (one, two))
Dkeras_utils_test.py179 x = values.DistributedValues(device_map, (a, b))
180 y = values.DistributedValues(device_map, (a, a))
205 x = values.DistributedValues(device_map, (a, b))
206 y = values.DistributedValues(device_map, (a, a))
Dmirrored_strategy_multigpu_test.py652 names = values.DistributedValues(device_map, ("foo", "bar"))
750 self.assertIsInstance(v, values.DistributedValues)
819 self.assertIsInstance(v, values.DistributedValues)
836 self.assertIsInstance(v, values.DistributedValues)
862 self.assertIsInstance(result_b, values.DistributedValues)
863 self.assertIsInstance(result_c, values.DistributedValues)
891 self.assertIsInstance(result_b, values.DistributedValues)
892 self.assertIsInstance(result_c, values.DistributedValues)
Dcross_device_ops_test.py47 elif isinstance(devices, value_lib.DistributedValues):
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py57 (value_lib.DistributedValues,
74 if isinstance(value, value_lib.DistributedValues):
154 if isinstance(destinations, value_lib.DistributedValues):
165 if isinstance(destinations, (value_lib.DistributedValues,
Dcollective_all_reduce_strategy.py436 if (isinstance(value, values.DistributedValues) and
442 if (not isinstance(value, values.DistributedValues) and
Dvalues.py232 class DistributedValues(object): class
311 class DistributedDelegate(DistributedValues):
383 class PerReplica(DistributedValues):
1388 return x.values[replica_id] if isinstance(x, DistributedValues) else x
1396 if isinstance(x, DistributedValues):
1417 if isinstance(u, DistributedValues):
Dparameter_server_strategy.py382 if not isinstance(value, values.DistributedValues):
442 if isinstance(val, values.DistributedValues):
Dcross_device_utils.py680 elif isinstance(value, value_lib.DistributedValues):
689 assert isinstance(value, value_lib.DistributedValues)
Dtpu_strategy.py451 if not isinstance(value, values.DistributedValues):
504 if isinstance(val, values.DistributedValues):
Dmirrored_strategy.py699 if not isinstance(value, values.DistributedValues):
745 if isinstance(val, values.DistributedValues):
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dautocast_variable.py175 if not isinstance(variable, distribute_values.DistributedValues):