/external/tensorflow/tensorflow/contrib/distribute/python/ |
D | values_test.py | 47 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0")) 48 v = values.DistributedValues(device_map, (one, two)) 60 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0")) 61 v = values.DistributedValues(device_map, (one, two)) 87 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0")) 88 v = values.DistributedValues(device_map, (one, two)) 100 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0")) 101 v = values.DistributedValues(device_map, (one, two)) 119 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0")) 120 v = values.DistributedDelegate(device_map, (Foo(7), Foo(8))) [all …]
|
D | keras_utils_test.py | 178 device_map = values.ReplicaDeviceMap(('/device:CPU:0', '/device:GPU:0')) 179 x = values.DistributedValues(device_map, (a, b)) 180 y = values.DistributedValues(device_map, (a, a)) 204 device_map = values.ReplicaDeviceMap(('/device:CPU:0', '/device:GPU:0')) 205 x = values.DistributedValues(device_map, (a, b)) 206 y = values.DistributedValues(device_map, (a, a))
|
D | cross_device_utils_test.py | 106 device_map = value_lib.ReplicaDeviceMap(("/gpu:0", "/cpu:0")) 107 per_replica = value_lib.PerReplica(device_map, (t0, t1))
|
D | input_lib_test.py | 40 device_map = values.ReplicaDeviceMap(devices) 41 input_workers = input_lib.InputWorkers(device_map, worker_device_pairs)
|
D | mirrored_strategy_multigpu_test.py | 651 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0")) 652 names = values.DistributedValues(device_map, ("foo", "bar")) 1391 device_map = values.ReplicaDeviceMap(("/device:CPU:0", "/device:GPU:0")) 1392 factors = values.PerReplica(device_map, (5.0, 3.0)) 1393 expected_result = values.PerReplica(device_map, (5.0 * 1.25, 3.0 * 1.25))
|
/external/tensorflow/tensorflow/python/distribute/ |
D | cross_device_ops.py | 72 def reduce_non_distributed_value(reduce_op, device_map, value, destinations): argument 92 if device_map.num_replicas_in_graph != 1: 113 device_map = value_lib.SingleDeviceMap(device) 114 return value_lib.PerReplica(device_map, (input_tensor,)) 157 return destinations.device_map.logical_to_actual_devices( 167 return destinations.device_map, destinations.logical_device 190 device_map, logical_device = get_device_map_from(destinations) 191 devices = device_map.logical_to_actual_devices(logical_device) 201 return value_lib.Mirrored(device_map, value_updates, logical_device) 452 device_map, logical_device = get_device_map_from(destinations) [all …]
|
D | mirrored_strategy.py | 86 def _call_for_each_replica(distribution, device_map, fn, args, kwargs): argument 116 for index in range(device_map.num_replicas_in_graph): 120 distribution, coord, index, device_map, variable_creator_fn, fn, 171 device_map, tuple(t.merge_args for t in threads)) 173 device_map, tuple(t.merge_kwargs for t in threads)) 195 return values.regroup(device_map, tuple(t.main_result for t in threads)) 198 def _create_mirrored_variable(strategy, device_map, logical_device, # pylint: disable=missing-docs… argument 248 devices = device_map.logical_to_actual_devices(logical_device) 253 strategy, device_map, value_list, aggregation, 257 strategy, device_map, value_list, aggregation, [all …]
|
D | values.py | 235 def __init__(self, device_map, values, logical_device=None): argument 236 assert isinstance(device_map, DeviceMap) 237 self._device_map = device_map 240 logical_device = device_map.logical_device_from_values(self._values) 272 def device_map(self): member in DistributedValues 444 def __init__(self, strategy, device_map, values, logical_device=None): argument 447 device_map, values, logical_device=logical_device) 652 self, strategy, device_map, values, aggregation, logical_device=None): argument 654 strategy, device_map, values, logical_device=logical_device) 774 self, strategy, device_map, values, aggregation, logical_device=None): argument [all …]
|
D | tpu_strategy.py | 71 strategy, device_map, logical_device, real_mirrored_creator, argument 102 devices = device_map.logical_to_actual_devices(logical_device) 105 strategy, device_map, value_list, aggregation, 188 device_map = self.extended._device_map # pylint: disable=protected-access 189 return values.regroup(device_map, replicate_outputs) 401 device_map = self._device_map 407 device_map = colocate_with.device_map 438 self._container_strategy(), device_map, logical_device,
|
D | input_lib.py | 44 def __init__(self, device_map, worker_device_pairs=None, logical_device=0): argument 54 self._device_map = device_map 59 device_map.logical_to_actual_devices(logical_device)),) 65 device_map.logical_to_actual_devices(logical_device)), ( 68 device_map.logical_to_actual_devices(logical_device))) 71 def device_map(self): member in InputWorkers 200 return values.regroup(self._input_workers.device_map, replicas)
|
D | collective_all_reduce_strategy.py | 240 device_map = self._device_map 246 device_map = colocate_with.device_map 328 self._container_strategy(), device_map, logical_device,
|
D | one_device_strategy.py | 57 device_map = values.SingleDeviceMap(device) 59 device_map, worker_device_pairs)
|
D | parameter_server_strategy.py | 294 device_map=self._device_map, logical_device=0)
|
/external/tensorflow/tensorflow/python/grappler/ |
D | tf_optimizer.i | 75 void DetectDevices(std::unordered_map<string, tensorflow::DeviceProperties>* device_map) { in DetectDevices() argument 84 tensorflow::DeviceProperties& prop = (*device_map)[device->name()]; in DetectDevices()
|
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
D | tpu_cluster_resolver.py | 126 device_map = collections.defaultdict(list) 133 device_map[host_id].append(core_id) 135 return DeviceDetails(device_map, num_cores) 496 device_details.device_map)}
|
D | tpu_cluster_resolver_test.py | 584 self.assertEqual(device_details.device_map,
|
/external/chromium-trace/catapult/devil/devil/android/ |
D | forwarder.py | 235 device_map = instance._device_to_host_port_map 237 for device_serial_and_port, host_port in device_map.items(): 240 del device_map[device_serial_and_port]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_sharding.cc | 120 std::map<int64, int64> device_map; in UsedDevices() local 125 device_map[*unique_device] += 1; in UsedDevices() 132 device_map[*unique_device] += 1; in UsedDevices() 138 return device_map; in UsedDevices()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | graph_rewrite_dataset.cc | 175 std::unordered_map<string, tensorflow::DeviceProperties> device_map; in ApplyOptimizations() local 176 tensorflow::grappler::VirtualCluster cluster(device_map); in ApplyOptimizations()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | step_stats_collector.h | 171 const std::unordered_map<string, const Graph*>& device_map);
|
D | step_stats_collector.cc | 263 const std::unordered_map<string, const Graph*>& device_map) { in BuildCostModel() argument 310 for (auto itr : device_map) { in BuildCostModel()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | context.h | 109 gtl::FlatMap<string, Device*, StringPieceHasher>* device_map() { in device_map() function
|
D | tensor_handle.cc | 323 const auto& map = *ctx->device_map(); in GetResourceDevice()
|