Home
last modified time | relevance | path

Searched refs:value_lib (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py29 from tensorflow.python.distribute import values as value_lib unknown
57 (value_lib.DistributedValues,
59 value_lib.AggregatingVariable,
61 value_lib.TPUMirroredVariable,
64 value_lib.LogicalDeviceSpec)):
74 if isinstance(value, value_lib.DistributedValues):
104 if isinstance(input_tensor, value_lib.PerReplica):
113 device_map = value_lib.SingleDeviceMap(device)
114 return value_lib.PerReplica(device_map, (input_tensor,))
145 if not all(isinstance(v[0], value_lib.PerReplica)
[all …]
Dcross_device_utils.py25 from tensorflow.python.distribute import values as value_lib unknown
680 elif isinstance(value, value_lib.DistributedValues):
689 assert isinstance(value, value_lib.DistributedValues)
/external/tensorflow/tensorflow/contrib/distribute/python/
Dcross_device_ops_test.py35 from tensorflow.python.distribute import values as value_lib unknown
47 elif isinstance(devices, value_lib.DistributedValues):
68 return value_lib.PerReplica(value_lib.ReplicaDeviceMap(devices), index)
79 return value_lib.Mirrored(value_lib.ReplicaDeviceMap(devices),
95 return value_lib.Mirrored(value_lib.ReplicaDeviceMap(devices), values)
295 per_replica = value_lib.PerReplica(
296 value_lib.ReplicaDeviceMap(devices), (t0, t1))
330 per_replica = value_lib.PerReplica(
331 value_lib.ReplicaDeviceMap(devices), (t0, t1))
Dcross_device_utils_test.py26 from tensorflow.python.distribute import values as value_lib unknown
106 device_map = value_lib.ReplicaDeviceMap(("/gpu:0", "/cpu:0"))
107 per_replica = value_lib.PerReplica(device_map, (t0, t1))