Home
last modified time | relevance | path

Searched refs:ReplicaDeviceMap (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dvalues_test.py47 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0"))
60 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0"))
87 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0"))
100 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0"))
119 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0"))
128 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0"))
185 device_map = values.ReplicaDeviceMap(devices)
205 device_map = values.ReplicaDeviceMap((_device_str(0), _device_str(1)))
237 device_map = values.ReplicaDeviceMap((_device_str(0), _device_str(1)))
276 device_map = values.ReplicaDeviceMap((_device_str(0), _device_str(1)))
[all …]
Dcross_device_ops_test.py68 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)
296 value_lib.ReplicaDeviceMap(devices), (t0, t1))
331 value_lib.ReplicaDeviceMap(devices), (t0, t1))
Dcross_device_utils_test.py106 device_map = value_lib.ReplicaDeviceMap(("/gpu:0", "/cpu:0"))
Dkeras_utils_test.py178 device_map = values.ReplicaDeviceMap(('/device:CPU:0', '/device:GPU:0'))
204 device_map = values.ReplicaDeviceMap(('/device:CPU:0', '/device:GPU:0'))
Dinput_lib_test.py40 device_map = values.ReplicaDeviceMap(devices)
Dmirrored_strategy_multigpu_test.py651 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0"))
1391 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0"))
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy.py161 self._device_map = values.ReplicaDeviceMap(compute_devices)
229 self._device_map = values.ReplicaDeviceMap(compute_devices)
Dmirrored_strategy.py471 self._device_map = values.ReplicaDeviceMap(devices)
506 self._device_map = values.ReplicaDeviceMap(devices)
Dtpu_strategy.py238 self._device_map = values.ReplicaDeviceMap(self._tpu_devices)
Dvalues.py160 class ReplicaDeviceMap(DeviceMap): class