Home
last modified time | relevance | path

Searched refs:replica (Results 1 – 25 of 209) sorted by relevance

123456789

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_placer.cc47 for (int replica = 0; replica < replica_count(); ++replica) { in Serialize() local
48 computation_device->add_replica_device_ids((*this)(replica, computation)); in Serialize()
70 for (int replica = 0; replica < proto.replica_count(); ++replica) { in Deserialize() local
71 (*assignment)(replica, computation) = in Deserialize()
72 computation_device.replica_device_ids(replica); in Deserialize()
83 for (int replica = 0; replica < replica_count(); ++replica) { in ToString() local
84 StrAppend(&output, operator()(replica, computation), " "); in ToString()
91 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId() argument
94 TF_RET_CHECK(replica < replica_count); in DeviceId()
97 return computation * replica_count + replica; in DeviceId()
[all …]
Dservice.cc245 for (int replica = 0; replica < options_.number_of_replicas(); ++replica) { in ResolveAndValidateArguments() local
246 const ShapedBuffer* shaped_buffer = replicated_buffers[replica]; in ResolveAndValidateArguments()
247 int replica_device_ordinal = stream_executors[replica]->device_ordinal(); in ResolveAndValidateArguments()
258 replicated_arguments[replica].push_back(shaped_buffer); in ResolveAndValidateArguments()
417 for (int64 replica = 0; replica < replicas.size(); ++replica) { in ExecuteParallelAndRegisterResult() local
418 device_assignment(replica, i) = replicas[replica]->device_ordinal(); in ExecuteParallelAndRegisterResult()
427 for (int64 replica = 0; replica < replicas.size(); ++replica) { in ExecuteParallelAndRegisterResult() local
429 backend->BorrowStream(replicas[replica])); in ExecuteParallelAndRegisterResult()
432 if (replica == 0 && profile != nullptr) { in ExecuteParallelAndRegisterResult()
441 if (replica == 0 && in ExecuteParallelAndRegisterResult()
[all …]
/external/tensorflow/tensorflow/python/framework/
Ddevice.py68 def __init__(self, job=None, replica=None, task=None, device_type=None, argument
81 self.replica = replica
111 def replica(self): member in DeviceSpec
114 @replica.setter
115 def replica(self, replica): argument
116 if replica is not None:
117 self._replica = int(replica)
158 self.replica = y[1]
187 if dev.replica is not None:
188 self.replica = dev.replica
[all …]
Ddevice_test.py38 d = device.DeviceSpec(job="j", replica=0, task=1,
41 self.assertEqual(0, d.replica)
60 d.replica = 12
74 d = device.DeviceSpec(job="foo", replica=12, task=3, device_type="GPU")
/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils.cc89 static string DeviceName(const string& job, int replica, int task, in DeviceName() argument
93 CHECK_LE(0, replica); in DeviceName()
97 return strings::StrCat("/job:", job, "/replica:", replica, "/task:", task, in DeviceName()
102 string DeviceNameUtils::FullName(const string& job, int replica, int task, in FullName() argument
104 return DeviceName(job, replica, task, "/device:", type, id); in FullName()
108 string LegacyName(const string& job, int replica, int task, const string& type, in LegacyName() argument
110 return DeviceName(job, replica, task, "/", str_util::Lowercase(type), id); in LegacyName()
130 if (p->has_replica && !ConsumeNumber(&fullname, &p->replica)) { in ParseFullName()
196 parsed_name->replica = parsed_basename.replica; in CompleteName()
252 if (pn.has_replica) strings::StrAppend(&buf, "/replica:", pn.replica); in ParsedNameToString()
[all …]
Ddevice_name_utils.h49 static string FullName(const string& job, int replica, int task,
60 replica = 0; in Clear()
68 (has_replica ? (other.has_replica && replica == other.replica)
80 int replica = 0; member
Ddevice_name_utils_test.cc97 EXPECT_EQ(p.replica, 1); in TEST()
113 EXPECT_EQ(p.replica, 1); in TEST()
129 EXPECT_EQ(p.replica, 1); in TEST()
142 EXPECT_EQ(p.replica, 4); in TEST()
154 EXPECT_EQ(p.replica, 4); in TEST()
166 EXPECT_EQ(p.replica, 4); in TEST()
178 EXPECT_EQ(p.replica, 4); in TEST()
200 EXPECT_EQ(p.replica, 4); in TEST()
212 EXPECT_EQ(p.replica, 4); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc117 device: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
127 s: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
133 s: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
158 device: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
168 s: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
174 s: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
201 device: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
211 s: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
219 device: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
235 s: "/job:a/replica:0/task:0/device:CPU:0" in TEST_F()
[all …]
Dremote_device.cc99 device_name_parsed.replica == worker_name_parsed.replica && in NewRemoteDevices()
106 worker_name_parsed.job, worker_name_parsed.replica, in NewRemoteDevices()
/external/tensorflow/tensorflow/python/tpu/
Ddevice_assignment.py35 for replica in xrange(core_assignment.shape[0]):
37 coordinates = core_assignment[replica, logical_core, :]
44 task_and_cores_to_replicas[task_id][logical_core].add(replica)
129 def coordinates(self, replica, logical_core): argument
131 return tuple(self.core_assignment[replica, logical_core, :])
153 def tpu_ordinal(self, replica=0, logical_core=0): argument
155 coordinates = self.coordinates(replica, logical_core)
158 def host_device(self, replica=0, logical_core=0, job=None): argument
160 coordinates = self.coordinates(replica, logical_core)
163 def tpu_device(self, replica=0, logical_core=0, job=None): argument
[all …]
Dtpu_context.py540 replica=replica_id, job=master)
556 return self.device_assignment.tpu_device(replica=i, job=master)
581 replica = self.device_assignment.lookup_replicas(host_id,
583 return self.device_assignment.tpu_ordinal(replica=replica)
689 replica=replica_id, job=master),
690 self.device_assignment.tpu_ordinal(replica=replica_id))
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AllToAll.pbtxt15 replica ids in the ith subgroup.
51 On each replica, the input is split into `split_count` blocks along
57 replica 0 receives input: `[[A, B]]`
58 replica 1 receives input: `[[C, D]]`
65 replica 0's output: `[[A], [C]]`
66 replica 1's output: `[[B], [D]]`
Dapi_def_TPUReplicate.pbtxt7 the inputs to 'computation', flattened, in replica-major order.
14 broadcast inputs are appended to the per-replica inputs when calling
47 the number of logical cores in each replica.
69 [replica, num_cores_per_replica, mesh_dimension] that maps the coordinates
70 of logical cores in each replica of a computation to physical coordinates in
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt31 …argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=No…
35 …argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, de…
43 …argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, de…
47 …argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=[\…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt31 …argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=No…
35 …argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, de…
43 …argspec: "args=[\'self\', \'replica\', \'logical_core\', \'job\'], varargs=None, keywords=None, de…
47 …argspec: "args=[\'self\', \'replica\', \'logical_core\'], varargs=None, keywords=None, defaults=[\…
Dtensorflow.-device-spec.pbtxt10 name: "replica"
19 …argspec: "args=[\'self\', \'job\', \'replica\', \'task\', \'device_type\', \'device_index\'], vara…
/external/guava/guava-tests/test/com/google/common/collect/
DMinMaxPriorityQueueTest.java194 SortedMap<Integer, AtomicInteger> replica = Maps.newTreeMap(); in testHeapIntact() local
199 insertIntoReplica(replica, randomInt); in testHeapIntact()
210 insertIntoReplica(replica, randomInt); in testHeapIntact()
215 removeMinFromReplica(replica, mmHeap.poll()); in testHeapIntact()
217 removeMaxFromReplica(replica, mmHeap.pollLast()); in testHeapIntact()
219 for (Integer v : replica.keySet()) { in testHeapIntact()
828 Map<Integer, AtomicInteger> replica, in insertIntoReplica() argument
830 if (replica.containsKey(newValue)) { in insertIntoReplica()
831 replica.get(newValue).incrementAndGet(); in insertIntoReplica()
833 replica.put(newValue, new AtomicInteger(1)); in insertIntoReplica()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer_test.cc763 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
780 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
802 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
826 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
838 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
845 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
866 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
887 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
907 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
919 device: "/job:localhost/replica:0/task:0/device:CPU:0" in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc282 auto execute = [this, &argument_handles, &results](int replica) { in ExecutePerReplica() argument
283 const int device_ordinal = device_assignment_(replica, 0); in ExecutePerReplica()
284 VLOG(3) << "Replica " << replica in ExecutePerReplica()
288 argument_buffers.reserve(argument_handles[replica].size()); in ExecutePerReplica()
289 for (auto& handle : argument_handles[replica]) { in ExecutePerReplica()
302 results[replica] = std::move(result_buffer_status); in ExecutePerReplica()
314 for (int replica = 0; replica < num_replicas() - 1; ++replica) { in ExecutePerReplica() local
315 pool.Schedule([&execute, replica] { execute(replica); }); in ExecutePerReplica()
321 for (int replica = 0; replica < num_replicas(); ++replica) { in ExecutePerReplica() local
322 auto& statusor = results[replica]; in ExecutePerReplica()
[all …]
/external/replicaisland/src/com/replica/replicaisland/
DHitReactionComponent.java17 package com.replica.replicaisland;
19 import com.replica.replicaisland.CollisionParameters.HitType;
20 import com.replica.replicaisland.GameObject.ActionType;
21 import com.replica.replicaisland.GameObject.Team;
22 import com.replica.replicaisland.GameObjectFactory.GameObjectType;
DLauncherComponent.java17 package com.replica.replicaisland;
19 import com.replica.replicaisland.GameObject.ActionType;
20 import com.replica.replicaisland.SoundSystem.Sound;
DCollisionVolume.java17 package com.replica.replicaisland;
19 import com.replica.replicaisland.CollisionParameters.HitType;
DButtonAnimationComponent.java17 package com.replica.replicaisland;
19 import com.replica.replicaisland.SoundSystem.Sound;
DSelectDialogComponent.java17 package com.replica.replicaisland;
19 import com.replica.replicaisland.CollisionParameters.HitType;
/external/tensorflow/tensorflow/core/profiler/g3doc/
Dprofile_time.md136 SoftmaxCrossEntropyWithLogits 1.37sec (100.00%, 36.44%), /job:worker/replica:0/task:0/cpu:0
137 …replica:0/task:0/cpu:0|/job:worker/replica:0/task:0/device:GPU:0|/job:worker/replica:0/task:0/devi…

123456789