Home
last modified time | relevance | path

Searched refs:destinations (Results 1 – 25 of 137) sorted by relevance

123456

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py53 def check_destinations(destinations): argument
63 if isinstance(destinations,
65 return bool(destinations.device)
66 return bool(destinations)
69 def validate_destinations(destinations): argument
72 destinations,
75 )) and not resource_variable_ops.is_resource_variable(destinations):
79 if not check_destinations(destinations):
84 reduce_op, value, destinations, num_replicas_in_graph): argument
108 validate_destinations(destinations)
[all …]
Dparameter_server_strategy.py409 def _broadcast_to(self, tensor, destinations): argument
417 if not cross_device_ops_lib.check_destinations(destinations):
419 destinations = self._compute_devices
420 return self._cross_device_ops.broadcast(tensor, destinations)
492 def _verify_destinations_not_different_worker(self, destinations): argument
495 if destinations is None:
497 for d in cross_device_ops_lib.get_devices_from(destinations):
504 def _gather_to_implementation(self, value, destinations, axis, argument
506 self._verify_destinations_not_different_worker(destinations)
511 destinations=destinations,
[all …]
Dmirrored_strategy.py655 def _broadcast_to(self, tensor, destinations): argument
664 if not destinations:
666 destinations = self._devices
667 return self._get_cross_device_ops(tensor).broadcast(tensor, destinations)
709 def _gather_to_implementation(self, value, destinations, axis, options): argument
715 destinations=destinations,
719 def _reduce_to(self, reduce_op, value, destinations, options): argument
730 reduce_op, value, destinations, self._num_replicas_in_sync)
732 …(not cross_device_ops_lib._devices_match(value, destinations) or # pylint: disable=protected-acce…
734 for d in cross_device_ops_lib.get_devices_from(destinations)))):
[all …]
Ddistribute_lib.py2224 def reduce_to(self, reduce_op, value, destinations, options=None): argument
2301 assert not isinstance(destinations, (list, tuple))
2307 return self._reduce_to(reduce_op, value, destinations, options)
2309 def _reduce_to(self, reduce_op, value, destinations, options): argument
2386 self.reduce_to(reduce_op, t, destinations=v, options=options)
2426 def _gather_to(self, value, destinations, axis, options=None): argument
2453 assert not isinstance(destinations, (list, tuple))
2456 return self._gather_to_implementation(value, destinations, axis, options)
2458 def _gather_to_implementation(self, value, destinations, axis, options): argument
2466 self._gather_to(t, destinations=v, axis=axis, options=options)
[all …]
Done_device_strategy.py308 def _broadcast_to(self, tensor, destinations): argument
309 del destinations
396 def _reduce_to(self, reduce_op, value, destinations, options): argument
397 del reduce_op, destinations, options
400 def _gather_to_implementation(self, value, destinations, axis, options): argument
401 del destinations, axis, options
Dcollective_all_reduce_strategy.py742 def _gather_to_implementation(self, value, destinations, axis, options): argument
745 destinations=destinations,
749 def _reduce_to(self, reduce_op, value, destinations, options): argument
768 reduce_op, value, destinations, len(self.worker_devices))
772 destinations=destinations,
Dvalues_util.py325 def apply_aggregation(strategy, value, aggregation, destinations): argument
329 destinations=destinations)
331 return strategy.extended.reduce_to(reduce_op, value, destinations)
Dtpu_strategy.py1168 def _gather_to_implementation(self, value, destinations, axis, options): argument
1196 output = self._broadcast_output(destinations, output)
1199 def _broadcast_output(self, destinations, output): argument
1200 devices = cross_device_ops_lib.get_devices_from(destinations)
1211 output = cross_device_ops_lib.simple_broadcast(output, destinations)
1215 def _reduce_to(self, reduce_op, value, destinations, options): argument
1233 reduce_op, value, destinations, self._num_replicas_in_sync)
1259 output = self._broadcast_output(destinations, output)
1313 def _broadcast_to(self, tensor, destinations): argument
1314 del destinations
/external/tensorflow/tensorflow/python/debug/lib/
Dsource_remote.py98 def _send_call_tracebacks(destinations, argument
130 if not isinstance(destinations, list):
131 destinations = [destinations]
133 destinations = [
136 for dest in destinations]
165 for destination in destinations:
176 def send_graph_tracebacks(destinations, argument
196 destinations, origin_stack, is_eager_execution=False, call_key=run_key,
200 def send_eager_tracebacks(destinations, argument
213 destinations, origin_stack, is_eager_execution=True,
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py187 for destinations in all_destinations:
192 destinations=destinations), _fake_mirrored(mean, destinations),
198 destinations=destinations),
199 _fake_mirrored(mean_2, destinations), sess)
204 destinations=destinations),
205 _fake_mirrored(mean * len(devices), destinations), sess)
210 destinations=destinations),
211 _fake_mirrored(mean_2 * len(devices), destinations), sess)
231 for destinations in all_destinations:
233 cross_device_ops.broadcast(constant_op.constant(1.), destinations),
[all …]
/external/gemmlowp/meta/generators/
Dneon_emitter.py407 def EmitVLoadA(self, load_no, load_type, destinations, source): argument
409 '{%s}' % ', '.join(_ExpandQuads(destinations)), source)
414 destinations, argument
418 destinations = _ExpandQuads(destinations)
419 if len(destinations) * 64 < bits_to_load:
421 (len(destinations), bits_to_load))
425 self.EmitVLoadA(1, 32, destinations[:4],
428 destinations = destinations[4:]
430 self.EmitVLoadA(1, 32, destinations[:3],
433 destinations = destinations[3:]
[all …]
Dneon_emitter_64.py810 def EmitVLoadA(self, load_no, load_type, destinations, source): argument
813 [_LoadStoreSize(destination) for destination in destinations]) / 8
814 self.EmitVLoadAPostIncrement(load_no, load_type, destinations, source,
817 self.EmitVLoadAPostIncrement(load_no, load_type, destinations, source,
820 def EmitVLoadAPostIncrement(self, load_no, load_type, destinations, source, argument
823 if len(destinations) == 1 and destinations[0].lane is -1:
824 destination = '{%s}' % _AppendType(load_type, destinations[0])
831 destination_list = _RegisterList(load_type, destinations)
840 destinations, argument
845 min_bits = min([destination.register_bits for destination in destinations])
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-cross-device-ops.pbtxt19 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
23 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
27 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
31 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
Dtensorflow.distribute.-reduction-to-one-device.pbtxt20 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
24 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
28 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
32 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
Dtensorflow.distribute.-nccl-all-reduce.pbtxt21 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
25 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
Dtensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt21 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
25 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-cross-device-ops.pbtxt19 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
23 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
27 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
31 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
Dtensorflow.distribute.-reduction-to-one-device.pbtxt20 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
24 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
28 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
32 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
Dtensorflow.distribute.-nccl-all-reduce.pbtxt21 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
25 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
Dtensorflow.distribute.-hierarchical-copy-all-reduce.pbtxt21 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
25 …argspec: "args=[\'self\', \'tensor\', \'destinations\'], varargs=None, keywords=None, defaults=Non…
29 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
33 …argspec: "args=[\'self\', \'reduce_op\', \'per_replica_value\', \'destinations\', \'options\'], va…
/external/deqp-deps/glslang/glslang/MachineIndependent/
DLiveTraverser.h77 destinations.push_back(candidate); in pushFunction()
95 destinations.push_back(candidate); in pushGlobalReference()
103 TDestinationStack destinations; variable
/external/angle/third_party/vulkan-deps/glslang/src/glslang/MachineIndependent/
DLiveTraverser.h77 destinations.push_back(candidate); in pushFunction()
95 destinations.push_back(candidate); in pushGlobalReference()
103 TDestinationStack destinations; variable
/external/harfbuzz_ng/test/subset/
Dsubset_test_suite.py79 destinations = {
95 if line in destinations:
96 current_destination = destinations[line]
/external/walt/ios/WALT/
DMIDIEndpoint.m39 NSMutableArray<MIDIDestination *> *destinations =
48 [destinations addObject:destination];
54 return destinations;
/external/python/cpython3/Lib/test/
Dtest_clinic.py55 self.destinations = {}
59 d = self.destinations.get
71 d = self.destinations.get(name)
77 if name in self.destinations:
79 self.destinations[name] = clinic.Destination(name, type, self, *args)

123456