Home
last modified time | relevance | path

Searched refs:device_map (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dvalues_test.py47 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 …]
Dkeras_utils_test.py178 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))
Dcross_device_utils_test.py106 device_map = value_lib.ReplicaDeviceMap(("/gpu:0", "/cpu:0"))
107 per_replica = value_lib.PerReplica(device_map, (t0, t1))
Dinput_lib_test.py40 device_map = values.ReplicaDeviceMap(devices)
41 input_workers = input_lib.InputWorkers(device_map, worker_device_pairs)
Dmirrored_strategy_multigpu_test.py651 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/
Dcross_device_ops.py72 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 …]
Dmirrored_strategy.py86 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 …]
Dvalues.py235 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 …]
Dtpu_strategy.py71 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,
Dinput_lib.py44 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)
Dcollective_all_reduce_strategy.py240 device_map = self._device_map
246 device_map = colocate_with.device_map
328 self._container_strategy(), device_map, logical_device,
Done_device_strategy.py57 device_map = values.SingleDeviceMap(device)
59 device_map, worker_device_pairs)
Dparameter_server_strategy.py294 device_map=self._device_map, logical_device=0)
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer.i75 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/
Dtpu_cluster_resolver.py126 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)}
Dtpu_cluster_resolver_test.py584 self.assertEqual(device_details.device_map,
/external/chromium-trace/catapult/devil/devil/android/
Dforwarder.py235 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/
Dhlo_sharding.cc120 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/
Dgraph_rewrite_dataset.cc175 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/
Dstep_stats_collector.h171 const std::unordered_map<string, const Graph*>& device_map);
Dstep_stats_collector.cc263 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/
Dcontext.h109 gtl::FlatMap<string, Device*, StringPieceHasher>* device_map() { in device_map() function
Dtensor_handle.cc323 const auto& map = *ctx->device_map(); in GetResourceDevice()