Searched refs:SingleDeviceMap (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | values_test.py | 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,))
79 values.SingleDeviceMap("/device:cpu:0"), (42,))
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 57 device_map = values.SingleDeviceMap(device)
|
| D | cross_device_ops.py | 113 device_map = value_lib.SingleDeviceMap(device)
172 return value_lib.SingleDeviceMap(device), 0
|
| D | values.py | 96 class SingleDeviceMap(DeviceMap): class
|