Searched refs:DistributedValues (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/contrib/distribute/python/ |
D | values_test.py | 48 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))
|
D | keras_utils_test.py | 179 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))
|
D | mirrored_strategy_multigpu_test.py | 652 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)
|
D | cross_device_ops_test.py | 47 elif isinstance(devices, value_lib.DistributedValues):
|
/external/tensorflow/tensorflow/python/distribute/ |
D | cross_device_ops.py | 57 (value_lib.DistributedValues, 74 if isinstance(value, value_lib.DistributedValues): 154 if isinstance(destinations, value_lib.DistributedValues): 165 if isinstance(destinations, (value_lib.DistributedValues,
|
D | collective_all_reduce_strategy.py | 436 if (isinstance(value, values.DistributedValues) and 442 if (not isinstance(value, values.DistributedValues) and
|
D | values.py | 232 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):
|
D | parameter_server_strategy.py | 382 if not isinstance(value, values.DistributedValues): 442 if isinstance(val, values.DistributedValues):
|
D | cross_device_utils.py | 680 elif isinstance(value, value_lib.DistributedValues): 689 assert isinstance(value, value_lib.DistributedValues)
|
D | tpu_strategy.py | 451 if not isinstance(value, values.DistributedValues): 504 if isinstance(val, values.DistributedValues):
|
D | mirrored_strategy.py | 699 if not isinstance(value, values.DistributedValues): 745 if isinstance(val, values.DistributedValues):
|
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
D | autocast_variable.py | 175 if not isinstance(variable, distribute_values.DistributedValues):
|