HomeSort by relevance Sort by last modified time
    Searched refs:replicas (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
service.cc 157 << "Requested more replicas than there are devices.";
514 // Stream executors for the replicas of the current computation.
515 TF_ASSIGN_OR_RETURN(auto replicas, Replicas(*backend, device_handles[i]));
516 for (int64 replica = 0; replica < replicas.size(); ++replica) {
518 backend->BorrowStream(replicas[replica]));
556 // All replicas share the same device address for the result allocation,
557 // so only one of the replicas need to register the result handle.
636 TF_ASSIGN_OR_RETURN(auto replicas,
637 Replicas(*backend, SingleComputationDeviceHandle()))
1068 std::vector<se::StreamExecutor*> replicas; local
1614 std::vector<perftools::gputools::StreamExecutor*> replicas; local
    [all...]
  /external/autotest/site_utils/
check_slave_db_delay.py 11 Readonly replicas of TKO database, passed in by option --replicas.
72 parser.add_argument('-r', '--replicas', nargs='+',
73 help='IP addresses of readonly replicas of TKO.')
98 for replica in options.replicas:
  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
device_assignment.py 107 """Computes a nested dict which maps task and logical core to replicas."""
127 for core, replicas in core_to_replicas.items():
128 core_to_sorted_replicas[core] = sorted(replicas)
159 """The number of replicas of the computation."""
193 A sorted list of the replicas that are attached to that task and
247 num_replicas: The number of computation replicas to run. The replicas will
259 ValueError: If computation's replicas cannot fit into the TPU topology.
311 "requested {} replicas but only {} replicas with shape {} and
    [all...]
tpu.py 194 `[[]]`), indexed by `[replica_num][input_num]`. All replicas must
203 replicas is equal to the number of cores in the TPU system.
208 ValueError: If all replicas do not have equal numbers of input tensors.
235 # No replicas? Nothing to do.
242 # Verifies that all replicas have matching numbers and types of inputs
247 raise ValueError("Replicas must have the same number of inputs. "
254 "Replicas must have matching input types. Replica 0 had "
281 replicas = [inputs[replica][i] for replica in xrange(num_replicas)]
283 tpu_ops.tpu_replicated_input(replicas, name="input{}".format(i)))
  /external/tensorflow/tensorflow/contrib/estimator/python/estimator/
replicate_model_fn.py 74 the CPU. Replicas of operations are placed on each individual GPU.
176 the CPU. Replicas of operations are placed on each individual GPU.
708 # `i` is `[1.. number_of_towers]`. We are going to add values from replicas
716 replicas = [
720 ] # `replicas` doesn't contain the first-tower variable.
723 math_ops.add_n(replicas))
726 for replica in replicas:

Completed in 1282 milliseconds