Home
last modified time | relevance | path

Searched defs:replica (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_placer.cc47 for (int replica = 0; replica < replica_count(); ++replica) { in Serialize() local
70 for (int replica = 0; replica < proto.replica_count(); ++replica) { in Deserialize() local
83 for (int replica = 0; replica < replica_count(); ++replica) { in ToString() local
91 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
103 for (int replica = 0; replica < replica_count; ++replica) { in AssignDevices() local
Dservice.cc245 for (int replica = 0; replica < options_.number_of_replicas(); ++replica) { in ResolveAndValidateArguments() local
417 for (int64 replica = 0; replica < replicas.size(); ++replica) { in ExecuteParallelAndRegisterResult() local
427 for (int64 replica = 0; replica < replicas.size(); ++replica) { in ExecuteParallelAndRegisterResult() local
535 for (int64 replica = 0; replica < replicas.size(); ++replica) { in ExecuteAndRegisterResult() local
1164 for (int replica = 0; replica < options_.number_of_replicas(); ++replica) { in Replicas() local
/external/tensorflow/tensorflow/python/framework/
Ddevice.py68 def __init__(self, job=None, replica=None, task=None, device_type=None, argument
111 def replica(self): member in DeviceSpec
115 def replica(self, replica): argument
/external/tensorflow/tensorflow/python/tpu/
Ddevice_assignment.py129 def coordinates(self, replica, logical_core): argument
153 def tpu_ordinal(self, replica=0, logical_core=0): argument
158 def host_device(self, replica=0, logical_core=0, job=None): argument
163 def tpu_device(self, replica=0, logical_core=0, job=None): argument
/external/guava/guava-tests/test/com/google/common/collect/
DMinMaxPriorityQueueTest.java194 SortedMap<Integer, AtomicInteger> replica = Maps.newTreeMap(); in testHeapIntact() local
828 Map<Integer, AtomicInteger> replica, in insertIntoReplica()
838 SortedMap<Integer, AtomicInteger> replica, in removeMinFromReplica()
846 SortedMap<Integer, AtomicInteger> replica, in removeMaxFromReplica()
854 Map<Integer, AtomicInteger> replica, int value) { in removeFromReplica()
/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils.cc89 static string DeviceName(const string& job, int replica, int task, in DeviceName()
102 string DeviceNameUtils::FullName(const string& job, int replica, int task, in FullName()
108 string LegacyName(const string& job, int replica, int task, const string& type, in LegacyName()
Ddevice_name_utils.h80 int replica = 0; member
/external/tensorflow/tensorflow/python/distribute/
Ddistribution_strategy_context.py39 def __init__(self, dist, cross, replica): argument
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc282 auto execute = [this, &argument_handles, &results](int replica) { in ExecutePerReplica()
314 for (int replica = 0; replica < num_replicas() - 1; ++replica) { in ExecutePerReplica() local
321 for (int replica = 0; replica < num_replicas(); ++replica) { in ExecutePerReplica() local
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables.py769 replica=None): argument